#486768 (C++17) No.1059 素敵な集合

• 問題: No.1059 素敵な集合
• ユーザー: yuruhiya
• 提出日時: 2020-05-25 15:14:44
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 107 ms / 2,000 ms
• コード長: 18,248 bytes
• コンパイル時間: 2,357 ms
• コンパイル使用メモリ: 212,480 KB
• 実行使用メモリ: 70,300 KB
• 最終ジャッジ日時: 2024-07-23 09:38:54
• 合計ジャッジ時間: 3,600 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,812 KB
testcase_01	AC	107 ms; 69,076 KB
testcase_02	AC	11 ms; 8,656 KB
testcase_03	AC	2 ms; 6,944 KB
testcase_04	AC	2 ms; 6,944 KB
testcase_05	AC	2 ms; 6,940 KB
testcase_06	AC	10 ms; 7,888 KB
testcase_07	AC	10 ms; 8,528 KB
testcase_08	AC	12 ms; 8,140 KB
testcase_09	AC	4 ms; 6,940 KB
testcase_10	AC	18 ms; 11,772 KB
testcase_11	AC	11 ms; 8,784 KB
testcase_12	AC	5 ms; 6,940 KB
testcase_13	AC	18 ms; 12,596 KB
testcase_14	AC	2 ms; 6,940 KB
testcase_15	AC	35 ms; 21,312 KB
testcase_16	AC	11 ms; 9,164 KB
testcase_17	AC	10 ms; 7,508 KB
testcase_18	AC	7 ms; 6,940 KB
testcase_19	AC	97 ms; 69,672 KB
testcase_20	AC	96 ms; 70,300 KB
testcase_21	AC	34 ms; 21,324 KB

ソースコード

#define _USE_MATH_DEFINES
#define _CRT_SECURE_NO_WARNINGS
#include "bits/stdc++.h"
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define FOR(i, m, n) for (int i = (m); i < (n); ++i)
#define rrep(i, n) for (int i = (n) - 1; i >= 0; --i)
#define rfor(i, m, n) for (int i = (m); i >= (n); --i)
#define unless(c) if (!(c))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define range_it(a, l, r) (a).begin() + (l), (a).begin() + (r)

using namespace std;
using ll = long long;		using LD = long double;
using VB = vector<bool>;	using VVB = vector<VB>;
using VI = vector<int>;		using VVI = vector<VI>;
using VL = vector<ll>;		using VVL = vector<VL>;
using VS = vector<string>;	using VD = vector<LD>;
using PII = pair<int, int>;	using VP = vector<PII>;
using PLL = pair<ll, ll>;	using VPL = vector<PLL>;
template<class T>using PQ = priority_queue<T>;
template<class T>using PQS = priority_queue<T, vector<T>, greater<T>>;
constexpr int inf = (int)1e9;
constexpr ll inf_ll = (ll)1e18, MOD = 1000000007;
constexpr LD PI = M_PI, EPS = 1e-12;

// --- input --- //
#ifdef _WIN32
#define getchar_unlocked _getchar_nolock
#define putchar_unlocked _putchar_nolock
#define fwrite_unlocked fwrite
#define fflush_unlocked fflush
#endif
class Input {
	static int gc() {
		return getchar_unlocked();
	}
	template<class T>static void i(T& v) {
		cin >> v;
	}
	static void i(char& v) {
		while (isspace(v = gc()));
	}
	static void i(bool& v) {
		v = in<char>() != '0';
	}
	static void i(string& v) {
		v.clear(); char c; for (i(c); !isspace(c); c = gc())v += c;
	}
	static void i(int& v) {
		bool neg = false; v = 0; char c; i(c);
		if (c == '-') { neg = true; c = gc(); }
		for (; isdigit(c); c = gc())v = v * 10 + (c - '0');
		if (neg)v = -v;
	}
	static void i(long long& v) {
		bool neg = false; v = 0; char c; i(c);
		if (c == '-') { neg = true; c = gc(); }
		for (; isdigit(c); c = gc())v = v * 10 + (c - '0');
		if (neg)v = -v;
	}
	static void i(double& v) {
		double dp = 1; bool neg = false, adp = false; v = 0; char c; i(c);
		if (c == '-') { neg = true; c = gc(); }
		for (; isdigit(c) || c == '.'; c = gc()) {
			if (c == '.')adp = true;
			else if (adp)v += (c - '0') * (dp *= 0.1);
			else v = v * 10 + (c - '0');
		}
		if (neg)v = -v;
	}
	static void i(long double& v) {
		long double dp = 1; bool neg = false, adp = false; v = 0; char c; i(c);
		if (c == '-') { neg = true; c = gc(); }
		for (; isdigit(c) || c == '.'; c = gc()) {
			if (c == '.')adp = true;
			else if (adp)v += (c - '0') * (dp *= 0.1);
			else v = v * 10 + (c - '0');
		}
		if (neg)v = -v;
	}
	template<class T, class U>static void i(pair<T, U>& v) {
		i(v.first); i(v.second);
	}
	template<class T>static void i(vector<T>& v) {
		for (auto& e : v)i(e);
	}
	struct InputV {
		int n, id;
		InputV(int _n) :n(_n), id(0) {}
		InputV(const pair<int, int>& nm) :n(nm.first), id(nm.second) {}
		template<class T>operator vector<T>() {
			vector<T> v(n); i(v); return v;
		}
		template<class T>operator vector<vector<T>>() {
			vector<vector<T>> v(n, vector<T>(id)); i(v); return v;
		}
	};
public:
	static string get_line() {
		string v; char c;
		for (i(c); c != '\n' && c != '\0'; c = gc())v += c;
		return v;
	}
	template<class T>static T in() {
		T v; i(v); return v;
	}
	template<class T>operator T()const {
		return in<T>();
	}
	int operator--(int)const {
		return in<int>() - 1;
	}
	InputV operator[](int n)const {
		return InputV(n);
	}
	InputV operator[](const pair<int, int>& n)const {
		return InputV(n);
	}
	void operator()()const {}
	template<class H, class...T>void operator()(H&& h, T&& ...t)const {
		i(h); operator()(forward<T>(t)...);
	}
#define input(T) Input::in<T>()
#define INT input(int)
#define LL input(ll)
#define STR input(string)
#define inputs(T, ...) T __VA_ARGS__; in(__VA_ARGS__)
#define ini(...) inputs(int, __VA_ARGS__)
#define inl(...) inputs(ll, __VA_ARGS__)
#define ins(...) inputs(string, __VA_ARGS__)
}in;

// --- output --- //
struct BoolStr {
	const char* t, * f; BoolStr(const char* _t, const char* _f) :t(_t), f(_f) {}
}Yes("Yes", "No"), yes("yes", "no"), YES("YES", "NO"), Int("1", "0");
struct DivStr {
	const char* d, * l; DivStr(const char* _d, const char* _l) :d(_d), l(_l) {}
}spc(" ", "\n"), no_spc("", "\n"), end_line("\n", "\n"), comma(",", "\n"), no_endl(" ", "");
class Output {
	BoolStr B{ Yes }; DivStr D{ spc };
	void p(int v)const {
		if (v < 0)putchar_unlocked('-'), v = -v;
		char b[10]; int i = 0;
		while (v)b[i++] = '0' + v % 10, v /= 10;
		if (!i)b[i++] = '0';
		while (i--)putchar_unlocked(b[i]);
	}
	void p(long long v)const {
		if (v < 0)putchar_unlocked('-'), v = -v;
		char b[20]; int i = 0;
		while (v)b[i++] = '0' + v % 10, v /= 10;
		if (!i)b[i++] = '0';
		while (i--)putchar_unlocked(b[i]);
	}
	void p(bool v)const {
		p(v ? B.t : B.f);
	}
	void p(char v)const {
		putchar_unlocked(v);
	}
	void p(const char* v)const {
		fwrite_unlocked(v, 1, strlen(v), stdout);
	}
	void p(double v)const {
		printf("%.20f", v);
	}
	void p(long double v)const {
		printf("%.20Lf", v);
	}
	template<class T> void p(const T& v)const {
		cout << v;
	}
	template<class T, class U>void p(const pair<T, U>& v)const {
		p(v.first); p(D.d); p(v.second);
	}
	template<class T>void p(const vector<T>& v)const {
		rep(i, sz(v)) { if (i)p(D.d); p(v[i]); }
	}
	template<class T>void p(const vector<vector<T>>& v)const {
		rep(i, sz(v)) { if (i)p(D.l); p(v[i]); }
	}
public:
	Output& operator()() {
		p(D.l); return *this;
	}
	template<class H>Output& operator()(H&& h) {
		p(h); p(D.l);
		return *this;
	}
	template<class H, class...T>Output& operator()(H&& h, T&& ...t) {
		p(h); p(D.d);
		return operator()(forward<T>(t)...);
	}
	template<class It>Output& range(const It& l, const It& r) {
		for (It i = l; i != r; i++) { if (i != l)p(D.d); p(*i); } p(D.l);
		return *this;
	}
	template<class T>Output& range(const T& a) {
		range(a.begin(), a.end());
		return *this;
	}
	template<class...T>void exit(T&& ...t) {
		operator()(forward<T>(t)...);
		std::exit(EXIT_SUCCESS);
	}
	Output& flush() {
		fflush_unlocked(stdout);
		return *this;
	}
	Output& set(const BoolStr& b) {
		B = b; return *this;
	}
	Output& set(const DivStr& d) {
		D = d; return *this;
	}
	Output& set(const char* t, const char* f) {
		B = BoolStr(t, f); return *this;
	}
}out;

// --- step --- //
template<class T>struct Step {
	class It {
		T a, b, c;
	public:
		constexpr It() : a(T()), b(T()), c(T()) {}
		constexpr It(T _b, T _c, T _s) : a(_b), b(_c), c(_s) {}
		constexpr It& operator++() {
			--b; a += c; return *this;
		}
		constexpr It operator++(int) {
			It tmp = *this; --b; a += c; return tmp;
		}
		constexpr const T& operator*()const {
			return a;
		}
		constexpr const T* operator->()const {
			return &a;
		}
		constexpr bool operator==(const It& i)const {
			return b == i.b;
		}
		constexpr bool operator!=(const It& i)const {
			return !(b == i.b);
		}
		constexpr T start()const {
			return a;
		}
		constexpr T count()const {
			return b;
		}
		constexpr T step()const {
			return c;
		}
	};
	constexpr Step(T b, T c, T s) : be(b, c, s) {}
	constexpr It begin()const {
		return be;
	}
	constexpr It end()const {
		return en;
	}
	constexpr T start()const {
		return be.start();
	}
	constexpr T count()const {
		return be.count();
	}
	constexpr T step()const {
		return be.step();
	}
	constexpr T sum()const {
		return start() * count() + step() * (count() * (count() - 1) / 2);
	}
	operator vector<T>()const {
		return as_vector();
	}
	template<class F>void each(const F& f)const {
		for (T i : *this)f(i);
	}
	auto as_vector()const {
		vector<T> res; res.reserve(count()); each([&](T i) {res.push_back(i); }); return res;
	}
	template<class F, class U = decay_t<result_of_t<F(T)>>>auto map(const F& f)const {
		vector<U> res; res.reserve(count()); each([&](T i) {res.push_back(f(i)); }); return res;
	}
	template<class F>auto select(const F& f)const {
		vector<T> res; each([&](T i) {if (f(i))res.push_back(i); }); return res;
	}
	template<class F>int count_if(const F& f)const {
		int res = 0; each([&](T i) {if (f(i))++res; }); return res;
	}
	template<class F>T find_if(const F& f)const {
		for (T i : *this)if (f(i))return i; return 0;
	}
	template<class F>auto max_by(const F& f)const {
		auto v = map(f); return *max_element(v.begin(), v.end());
	}
	template<class F>auto min_by(const F& f)const {
		auto v = map(f); return *min_element(v.begin(), v.end());
	}
	template<class F>bool all_of(const F& f)const {
		for (T i : *this)if (!f(i))return false; return true;
	}
	template<class F>bool any_of(const F& f)const {
		for (T i : *this)if (f(i))return true; return false;
	}
	template<class F, class U = decay_t<result_of_t<F(T)>>>auto sum(const F& f)const {
		U res = 0; each([&](T i) {res += static_cast<U>(f(i)); }); return res;
	}
	using value_type = T;
	using iterator = It;
private:
	It be, en;
};
template<class T>inline constexpr auto step(T a) {
	return Step<T>(0, a, 1);
}
template<class T>inline constexpr auto step(T a, T b) {
	return Step<T>(a, b - a, 1);
}
template<class T>inline constexpr auto step(T a, T b, T c) {
	return Step<T>(a, a < b ? (b - a - 1) / c + 1 : 0, c);
}

// --- functions --- //
inline namespace {
	template<class T>inline void Sort(T& a) {
		sort(all(a));
	}
	template<class T>inline void RSort(T& a) {
		sort(rall(a));
	}
	template<class T, class F>inline void Sort(T& a, const F& f) {
		sort(all(a), f);
	}
	template<class T, class F>inline void RSort(T& a, const F& f) {
		sort(rall(a), f);
	}
	template<class T>inline T Sorted(T a) {
		Sort(a); return a;
	}
	template<class T>inline T RSorted(T a) {
		RSort(a); return a;
	}
	template<class T, class F>inline T Sorted(T& a, const F& f) {
		Sort(a, f); return a;
	}
	template<class T, class F>inline T RSorted(T& a, const F& f) {
		RSort(a, f); return a;
	}
	template<class T, class F>inline void SortBy(T& a, const F& f) {
		sort(all(a), [&](const auto& x, const auto& y) {return f(x) < f(y); });
	}
	template<class T, class F>inline void RSortBy(T& a, const F& f) {
		sort(rall(a), [&](const auto& x, const auto& y) {return f(x) < f(y); });
	}
	template<class T>inline void Reverse(T& a) {
		reverse(all(a));
	}
	template<class T>inline void Unique(T& a) {
		a.erase(unique(all(a)), a.end());
	}
	template<class T>inline void Uniq(T& a) {
		Sort(a); Unique(a);
	}
	template<class T>inline void Rotate(T& a, int left) {
		rotate(a.begin(), a.begin() + left, a.end());
	}
	template<class T>inline T Reversed(T a) {
		Reverse(a); return a;
	}
	template<class T>inline T Uniqued(T a) {
		Unique(a); return a;
	}
	template<class T>inline T Uniqed(T a) {
		Uniq(a); return a;
	}
	template<class T>inline T Rotated(T a, int left) {
		Rotate(a, left); return a;
	}
	template<class T>inline auto Max(const T& a) {
		return *max_element(all(a));
	}
	template<class T>inline auto Min(const T& a) {
		return *min_element(all(a));
	}
	template<class T>inline int MaxPos(const T& a) {
		return max_element(all(a)) - a.begin();
	}
	template<class T>inline int MinPos(const T& a) {
		return min_element(all(a)) - a.begin();
	}
	template<class T, class F>inline auto MaxBy(const T& a, const F& f) {
		return *max_element(all(a), [&](const auto& x, const auto& y) {return f(x) < f(y); });
	}
	template<class T, class F>inline auto MinBy(const T& a, const F& f) {
		return *min_element(all(a), [&](const auto& x, const auto& y) {return f(x) < f(y); });
	}
	template<class T, class U>inline int Count(const T& a, const U& v) {
		return count(all(a), v);
	}
	template<class T, class F>inline int CountIf(const T& a, const F& f) {
		return count_if(all(a), f);
	}
	template<class T, class U>inline int Find(const T& a, const U& v) {
		return find(all(a), v) - a.begin();
	}
	template<class T, class F>inline int FindIf(const T& a, const F& f) {
		return find_if(all(a), f) - a.begin();
	}
	template<class T, class U = typename T::value_type>inline U Sum(const T& a) {
		return accumulate(all(a), U());
	}
	template<class T, class U>inline bool Includes(const T& a, const U& v) {
		return find(all(a), v) != a.end();
	}
	template<class T, class F>inline auto Sum(const T& v, const F& f) {
		return accumulate(next(v.begin()), v.end(), f(*v.begin()), [&](auto a, auto b) {return a + f(b); });
	}
	template<class T, class U>inline int Lower(const T& a, const U& v) {
		return lower_bound(all(a), v) - a.begin();
	}
	template<class T, class U>inline int Upper(const T& a, const U& v) {
		return upper_bound(all(a), v) - a.begin();
	}
	template<class T, class F>inline void RemoveIf(T& a, const F& f) {
		a.erase(remove_if(all(a), f), a.end());
	}
	template<class F>inline auto Vector(size_t size, const F& f) {
		vector<decay_t<result_of_t<F(size_t)>>> res(size);
		for (size_t i = 0; i < size; ++i)res[i] = f(i);
		return res;
	}
	template<class T>inline auto Grid(size_t h, size_t w, const T& v = T()) {
		return vector<vector<T>>(h, vector<T>(w, v));
	}
	template<class T>inline auto Slice(const T& v, size_t i, size_t len) {
		return i < v.size() ? T(v.begin() + i, v.begin() + min(i + len, v.size())) : T();
	}
	template<class T, class F>inline auto Each(T& v, F&& f) {
		for (auto& i : v)f(i);
	}
	template<class T, class F>inline auto Each(const T& v, F&& f) {
		for (const auto& i : v)f(i);
	}
	template<class T, class F>inline auto Select(const T& v, const F& f) {
		T res;
		for (const auto& e : v)if (f(e))res.push_back(e);
		return res;
	}
	template<class T, class F>inline auto Map(const T& v, F&& f) {
		vector<decay_t<result_of_t<F(typename T::value_type)>>> res(v.size());
		size_t i = 0; for (const auto& e : v)res[i++] = f(e);
		return res;
	}
	template<class T, class F>inline auto MapIndex(const T& v, const F& f) {
		vector<decay_t<result_of_t<F(size_t, typename T::value_type)>>> res(v.size());
		size_t i = 0; for (auto it = v.begin(); it != v.end(); ++it, ++i)res[i] = f(i, *it);
		return res;
	}
	template<class T, class F>inline auto TrueIndex(const T& v, const F& f) {
		vector<size_t> res;
		for (size_t i = 0; i < v.size(); ++i)if (f(v[i]))res.push_back(i);
		return res;
	}
	template<class T, class U = typename T::value_type>inline auto Indexed(const T& v) {
		vector<pair<U, int>> res(v.size());
		for (int i = 0; i < (int)v.size(); ++i)res[i] = make_pair(static_cast<U>(v[i]), i);
		return res;
	}
	inline auto operator*(string s, size_t n) {
		string res;
		for (size_t i = 0; i < n; ++i)res += s;
		return res;
	}
	template<class T>inline auto& operator<<(vector<T>& v, const vector<T>& v2) {
		v.insert(v.end(), all(v2)); return v;
	}
	template<class T>inline T Ceil(T n, T id) {
		return (n + id - 1) / id;
	}
	template<class T>inline T Ceil2(T n, T id) {
		return Ceil(n, id) * id;
	}
	template<class T>inline T Tri(T n) {
		return (n & 1) ? (n + 1) / 2 * n : n / 2 * (n + 1);
	}
	template<class T>inline T nC2(T n) {
		return (n & 1) ? (n - 1) / 2 * n : n / 2 * (n - 1);
	}
	template<class T>inline T Mid(const T& l, const T& r) {
		return l + (r - l) / 2;
	}
	inline int pop_count(int n) {
		return bitset<32>(n).count();
	}
	inline int pop_count(ll n) {
		return bitset<64>(n).count();
	}
	template<class T>inline bool chmax(T& a, const T& b) {
		if (a < b) { a = b; return true; } return false;
	}
	template<class T>inline bool chmin(T& a, const T& b) {
		if (a > b) { a = b; return true; } return false;
	}
	template<class T>inline bool inRange(const T& v, const T& min, const T& max) {
		return min <= v && v < max;
	}
	template<class T>inline bool isSquere(T n) {
		T s = sqrt(n); return s * s == n || (s + 1) * (s + 1) == n;
	}
	template<class T = ll>inline T BIT(int b) {
		return T(1) << b;
	}
	template<class T>inline T Gcd(T n, T id) {
		return id ? Gcd(id, n % id) : n;
	}
	template<class T>inline T Lcm(T n, T id) {
		return n / Gcd(n, id) * id;
	}
	template<class T, class U = typename T::value_type>inline U Gcdv(const T& v) {
		return accumulate(next(v.begin()), v.end(), U(*v.begin()), Gcd<U>);
	}
	template<class T, class U = typename T::value_type>inline U Lcmv(const T& v) {
		return accumulate(next(v.begin()), v.end(), U(*v.begin()), Lcm<U>);
	}
	template<class T>inline T Pow(T a, T n) {
		T r = 1; while (n > 0) { if (n & 1)r *= a; a *= a; n /= 2; } return r;
	}
	template<class T>inline T Powmod(T a, T n, T id = MOD) {
		T r = 1; while (n > 0) { if (n & 1)r = r * a % id, n--; else a = a * a % id, n /= 2; }return r;
	}
}

// --- dump --- //
#if __has_include("dump.hpp")
#include "dump.hpp"
#else
#define dump(...) ((void)0)
#endif

// ---------------------------------------------------------------- //

using Weight = long long;
constexpr Weight INF = numeric_limits<Weight>::max();
struct Edge {
	int to; Weight cost;
	Edge() :to(-1), cost(-1) {}
	Edge(int _to, Weight _cost = 1) :to(_to), cost(_cost) {}
	friend bool operator>(const Edge& e1, const Edge& e2) {
		return e1.cost > e2.cost;
	}
	friend ostream& operator<<(ostream& os, const Edge& e) {
		return os << "->" << e.to << '(' << e.cost << ')';
	}
};
using Graph = vector<vector<Edge>>;
struct Edge2 {
	int from, to; Weight cost;
	Edge2() :from(-1), to(-1), cost(0) {}
	Edge2(int _from, int _to, Weight _cost) :from(_from), to(_to), cost(_cost) {}
	friend ostream& operator<<(ostream& os, const Edge2& e) {
		return os << e.from << "->" << e.to << '(' << e.cost << ')';
	}
};
using Edges = vector<Edge2>;
using Matrix = vector<vector<Weight>>;

class UnionFind {
	vector<int> d;
public:
	UnionFind(int n = 0) { init(n); }
	void init(int n) { d.assign(n, -1); }
	int root(int x) { return d[x] < 0 ? x : d[x] = root(d[x]); }
	bool unite(int x, int y) {
		x = root(x); y = root(y);
		if (x == y)return false;
		if (d[x] > d[y])swap(x, y);
		d[x] += d[y]; d[y] = x;
		return true;
	}
	bool same(int x, int y) { return root(x) == root(y); }
	int size(int x) { return -d[root(x)]; }
};

Weight Kruskal(int V, Edges& graph) {
	sort(graph.begin(), graph.end(),
		[](const Edge2& e1, const Edge2& e2) {return e1.cost < e2.cost; });
	Weight res = 0;
	UnionFind uf(V);
	for (const auto& e : graph) {
		if (uf.unite(e.from, e.to)) {
			res += e.cost;
		}
	}
	return res;
}

int main() {
	int l = in, r = in;
	Edges edges;
	FOR(i, l, r) {
		edges.emplace_back(i, i + 1, 1);
	}
	FOR(i, l, r) {
		for (int j : step(i * 2, r + 1, i)) {
			edges.emplace_back(i, j, 0);
		}
	}
	out(Kruskal(r + 1, edges));
}