結果

問題 No.416 旅行会社
ユーザー srup٩(๑`н´๑)۶srup٩(๑`н´๑)۶
提出日時 2016-09-29 15:51:47
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 468 ms / 4,000 ms
コード長 2,619 bytes
コンパイル時間 690 ms
コンパイル使用メモリ 79,680 KB
実行使用メモリ 21,180 KB
最終ジャッジ日時 2023-08-21 10:06:43
合計ジャッジ時間 6,284 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 154 ms
14,016 KB
testcase_01 AC 3 ms
8,832 KB
testcase_02 AC 4 ms
8,828 KB
testcase_03 AC 3 ms
8,948 KB
testcase_04 AC 4 ms
8,840 KB
testcase_05 AC 3 ms
8,848 KB
testcase_06 AC 4 ms
8,852 KB
testcase_07 AC 5 ms
8,988 KB
testcase_08 AC 7 ms
9,000 KB
testcase_09 AC 21 ms
9,544 KB
testcase_10 AC 182 ms
15,156 KB
testcase_11 AC 178 ms
15,152 KB
testcase_12 AC 181 ms
15,100 KB
testcase_13 AC 151 ms
14,148 KB
testcase_14 AC 467 ms
20,748 KB
testcase_15 AC 468 ms
21,180 KB
testcase_16 AC 468 ms
20,612 KB
testcase_17 AC 467 ms
20,728 KB
testcase_18 AC 463 ms
20,880 KB
testcase_19 AC 291 ms
17,704 KB
testcase_20 AC 291 ms
17,900 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;
typedef long long ll;
#define rep(i,n) for(int i=0;i<(n);i++)

//Quick Find Weighted
const int MAX_N = 200010;
int i2g[MAX_N];//i2g[i] := 頂点iが所属するグループ
vector<int> g2i[MAX_N];//g2i[g]:= グループgに所属する頂点番号

// アイテムiaとアイテムibは同じグループに所属しているか？
bool issame(int ia, int ib){
	return i2g[ia] == i2g[ib];
}

// アイテムiaの所属するグループとアイテムibの所属するグループを1つにする(異なるグループに属するものに)
void merge(int ia, int ib){
	if(issame(ia, ib)) return;
	//iaの所属するグループがibの所属するグループより小さくならないようにする(一般的マージテク)
	if(g2i[i2g[ia]].size() < g2i[i2g[ib]].size()){
		swap(ia, ib);
	}
	int ga = i2g[ia], gb = i2g[ib];//グループgbの方が要素数が少ない
	for(auto u : g2i[gb]){//uには頂点の番号はいる
		i2g[u] = ga;//グループの番号を更新
	}
	g2i[ga].insert(g2i[ga].end(), g2i[gb].begin(), g2i[gb].end());//つなげる
	g2i[gb].clear();
}

void init(int n){
	for (int i = 0; i < n; ++i){
		i2g[i] = i;
		g2i[i].push_back(i);
	}
}

int ans[MAX_N];

int main(void){
	int n, m, q; cin >> n >> m >> q;
	vector<pair<int, int> > cd;
	set<pair<int, int> > s;//最後まで残っている橋を入れる

	for (int i = 0; i < m; ++i){
		int a, b; cin >> a >> b;
		a--; b--;
		s.insert(make_pair(a, b));
	}
	for (int i = 0; i < q; ++i){
		int c, d; cin >> c >> d;
		c--; d--;
		cd.push_back(make_pair(c, d));
		s.erase(make_pair(c, d));
	}

	init(n);//初期化
	for(auto u : s){//最後まで残っている橋をつなげる
		merge(u.first, u.second);
	}

	rep(i, n) ans[i] = 0;
	int g = i2g[0];//最後まで0と同じグループにあるものは、-1
	for(int v : g2i[g]){
		ans[v] = -1;
	}

    for (int i = q - 1; i >= 0 ; --i){
 		int a = cd[i].first, b = cd[i].second;
 		// printf("%d %d\n", a, b);
 		if(!issame(a, b)){
 			if(!issame(0, a) && !issame(0, b)){//0につながることはない
 				merge(a, b);
 				continue;
 			}

 			if(issame(0, a)){//0とaが同じグループなら、bも0と同じグループになる
 				for(auto u : g2i[i2g[b]]){
 					if(ans[u] == 0) ans[u] = i + 1;
 				}
 			}else{//0とbが同じグループなら、aも0と同じグループになる
 				for(auto u : g2i[i2g[a]]){
 					if(ans[u] == 0) ans[u] = i + 1;
 				}
 			}
 			merge(a, b);
 		}
    }
    for (int i = 1; i < n; ++i){
    	printf("%d\n", ans[i]);
    }
	return 0;
}
0