結果

問題 No.1640 簡単な色塗り
ユーザー Example0911
提出日時 2021-08-07 14:53:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 248 ms / 2,000 ms
コード長 2,331 bytes
コンパイル時間 2,332 ms
コンパイル使用メモリ 205,756 KB
最終ジャッジ日時 2025-01-23 16:40:09
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 51 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;

}
0