#687207 (C++17) No.1640 簡単な色塗り

提出ソース

結果

問題	No.1640 簡単な色塗り
ユーザー	Example0911
提出日時	2021-08-07 14:53:46
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	219 ms / 2,000 ms
コード長	2,331 bytes
コンパイル時間	2,633 ms
コンパイル使用メモリ	210,640 KB
実行使用メモリ	21,732 KB
最終ジャッジ日時	2023-09-12 03:50:33
合計ジャッジ時間	13,062 ms
ジャッジサーバーID （参考情報）	judge14 / judge15

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 5,840 KB
testcase_01	AC	3 ms 5,912 KB
testcase_02	AC	3 ms 5,776 KB
testcase_03	AC	3 ms 5,764 KB
testcase_04	AC	155 ms 12,588 KB
testcase_05	AC	170 ms 21,732 KB
testcase_06	AC	3 ms 5,748 KB
testcase_07	AC	3 ms 5,772 KB
testcase_08	AC	3 ms 5,912 KB
testcase_09	AC	3 ms 5,832 KB
testcase_10	AC	116 ms 12,548 KB
testcase_11	AC	93 ms 11,268 KB
testcase_12	AC	81 ms 10,632 KB
testcase_13	AC	182 ms 16,520 KB
testcase_14	AC	183 ms 16,572 KB
testcase_15	AC	61 ms 9,448 KB
testcase_16	AC	73 ms 10,208 KB
testcase_17	AC	149 ms 14,696 KB
testcase_18	AC	15 ms 6,556 KB
testcase_19	AC	82 ms 10,648 KB
testcase_20	AC	109 ms 12,228 KB
testcase_21	AC	90 ms 11,340 KB
testcase_22	AC	12 ms 6,312 KB
testcase_23	AC	120 ms 13,024 KB
testcase_24	AC	34 ms 7,652 KB
testcase_25	AC	81 ms 10,712 KB
testcase_26	AC	138 ms 13,932 KB
testcase_27	AC	56 ms 9,052 KB
testcase_28	AC	172 ms 15,924 KB
testcase_29	AC	132 ms 13,592 KB
testcase_30	AC	10 ms 7,464 KB
testcase_31	AC	67 ms 16,456 KB
testcase_32	AC	54 ms 15,132 KB
testcase_33	AC	33 ms 11,960 KB
testcase_34	AC	42 ms 13,612 KB
testcase_35	AC	34 ms 12,060 KB
testcase_36	AC	10 ms 7,436 KB
testcase_37	AC	12 ms 7,852 KB
testcase_38	AC	57 ms 15,612 KB
testcase_39	AC	23 ms 9,920 KB
testcase_40	AC	22 ms 10,124 KB
testcase_41	AC	43 ms 13,808 KB
testcase_42	AC	26 ms 10,420 KB
testcase_43	AC	27 ms 10,644 KB
testcase_44	AC	26 ms 10,440 KB
testcase_45	AC	21 ms 9,444 KB
testcase_46	AC	11 ms 7,332 KB
testcase_47	AC	7 ms 6,944 KB
testcase_48	AC	59 ms 15,776 KB
testcase_49	AC	5 ms 6,324 KB
testcase_50	AC	3 ms 5,760 KB
testcase_51	AC	3 ms 5,720 KB
testcase_52	AC	213 ms 20,244 KB
testcase_53	AC	219 ms 20,232 KB
07_evil_01.txt	RE	-
07_evil_02.txt	RE	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include "bits/stdc++.h"

#define int long long

using namespace std;
using ll = long long;
using P = pair<ll, ll>;
const ll INF = (1LL << 61);
ll mod = 1000000007;
struct UnionFind {
	vector<int>par, vsize, esize;
	UnionFind(int sz_) : par(sz_), vsize(sz_, 1), esize(sz_, 0) {
		for (int i = 0; i < sz_; i++)par[i] = i;
	}
	void init(int sz_) {
		par.resize(sz_);
		vsize.resize(sz_, 1);
		esize.resize(sz_, 0);
		for (int i = 0; i < sz_; i++)par[i] = i;
	}
	int root(int x) {
		if (par[x] == x)return x;
		else return par[x] = root(par[x]);
	}
	bool merge(int x, int y) {
		x = root(x), y = root(y);
		if (x == y) {
			esize[x]++;
			return false;
		}
		if (vsize[x] < vsize[y])swap(x, y);
		vsize[x] += vsize[y];
		esize[x] += esize[y] + 1;
		par[y] = x;
		return true;
	}
	bool issame(int x, int y) {
		return root(x) == root(y);
	}
	int vs(int x) {
		return vsize[root(x)];
	}
	int es(int x) {
		return esize[root(x)];
	}
};

int ans[100010];
vector<P>G[100010];
int now = -1;
void dfs(int v, int p) {
	for (auto nv : G[v]) {
		if (nv.first == p || nv.second == now)continue;
		ans[nv.second] = nv.first + 1;
		dfs(nv.first, v);
	}
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int N; cin >> N;
	UnionFind uf(N), uf2(N);
	vector<int>A(N), B(N);
	// 同連結成分か？ → 閉路を作り出す辺であることと同値
	vector<bool>ok(N);
	for (int i = 0; i < N; i++) {
		int a, b; cin >> a >> b; a--; b--;
		uf2.merge(a, b);
		if (uf.issame(a, b)) {
			ok[i] = true;
		}
		else {
			// 閉路のないグラフにする
			// このとき, 必ずグラフは木になっている
			uf.merge(a, b);
			G[a].push_back({ b, i });
			G[b].push_back({ a, i });
		}
		A[i] = a; B[i] = b;
	}
	int cnt = 0;
	for (int i = 0; i < N; i++) {
		if (uf2.root(i) == i)cnt += min(uf2.vs(i), uf2.es(i));
	}
	if (cnt != N) {
		cout << "No" << endl; return 0;
	}
	// ある連結成分について, すでに見終わっているか？
	vector<bool>seen(N);
	for (int i = 0; i < N; i++) {
		int root = uf.root(A[i]);
		// すでに見終わっているか, ほかの頂点から必ず見られる場合はcontinueする
		if (seen[root] || !ok[i])continue;
		seen[root] = true;
		now = i;
		dfs(A[i], -1);
		ans[i] = A[i] + 1;
	}

	cout << "Yes" << endl;
	for (int i = 0; i < N; i++)cout << ans[i] << endl;
	return 0;

}

yukicoder

結果

テストケース

ソースコード