結果

問題 No.1640 簡単な色塗り
コンテスト
ユーザー platinum
提出日時 2021-06-18 13:53:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 2,376 bytes
コンパイル時間 2,257 ms
コンパイル使用メモリ 204,824 KB
最終ジャッジ日時 2025-01-22 08:47:22
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28 RE * 2 TLE * 19 MLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (int)(n); i++)

using namespace std;
using LL = long long;
using P = pair<int,int>;
using vv = vector<vector<int>>;
const int INF = (int)1e9;
const LL LINF = (LL)1e18;
const int Max_N = (int)1e5;

class UnionFind {
public:
	vector <long long> par;
	vector <long long> siz;

	UnionFind(long long sz_): par(sz_), siz(sz_, (long long)1) {
		for (long long i = 0; i < sz_; ++i) par[i] = i;
	}
	void init(long long sz_) {
		par.resize(sz_);
		siz.assign(sz_, (long long)1);
		for (long long i = 0; i < sz_; ++i) par[i] = i;
	}

	long long root(long long x) {
		while (par[x] != x) {
			x = par[x] = par[par[x]];
		}
		return x;
	}

	bool unite(long long x, long long y) {
		x = root(x);
		y = root(y);
		if (x == y) return false;
		if (siz[x] < siz[y]) swap(x, y);
		siz[x] += siz[y];
		par[y] = x;
		return true;
	}

	bool same(long long x, long long y) {
		return root(x) == root(y);
	}

	long long size(long long x) {
		return siz[root(x)];
	}
};


struct edge{
	int to, num;
	edge(int to, int num) : to(to), num(num) {}
};

vector<vector<edge>> G;
int visited[100010], ans[100010];

void dfs(int n, int par = -1){
	for(auto e : G[n]){
		if(e.to == par) continue;
		ans[e.num] = e.to + 1;
		dfs(e.to, n);
	}
}
int st = -1, erase_num = -1;
void cycle(int n, int edge_prev = -1){
	visited[n] = 1;
	for(int j = 0; j < G[n].size(); j++){
		edge e = G[n][j];
		if(e.num == edge_prev) continue;
		if(visited[e.to]){
			st = e.to;
			ans[e.num] = e.to + 1;
			erase_num = e.num;
			continue;
		}
		cycle(e.to, e.num);
	}
}

int main(){
	int N;
	cin >> N;
	G.resize(N);
	assert(1 <= N and N <= Max_N);
	vector<int> A(N);
	UnionFind uf(N);
	rep(i,N){
		int a, b;
		cin >> a >> b;
		assert(1 <= a and a <= N);
		assert(1 <= b and b <= N);
		a--; b--;
		uf.unite(a, b);
		A[i] = a;
		G[a].emplace_back(b, i); G[b].emplace_back(a, i);
	}
	vector<int> edge_num(N);
	rep(i,N){
		int p = uf.root(A[i]);
		edge_num[p]++;
	}
	rep(i,N){
		if(uf.root(i) != i) continue;
		int siz = uf.size(i);
		if(siz != edge_num[i]){
			cout << "No" << endl;
			return 0;
		}
	}
	rep(i,N){
		if(!visited[i]){
			cycle(i);
			int k = 0;
			for(int j = 0; j < G[st].size(); j++){
				if(G[st][j].num == erase_num) k = j;
			}
			G[st].erase(G[st].begin() + k);
			dfs(st);
		}
	}
	cout << "Yes" << endl;
	rep(i,N) cout << ans[i] << endl;

	return 0;
}
0