#951272 (C++17) No.2630 Colorful Vertices and Cheapest Paths

• 問題: No.2630 Colorful Vertices and Cheapest Paths
• ユーザー: Carpenters-Cat
• 提出日時: 2024-02-16 22:01:32
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 238 ms / 2,500 ms
• コード長: 1,200 bytes
• コンパイル時間: 2,427 ms
• コンパイル使用メモリ: 209,960 KB
• 実行使用メモリ: 6,676 KB
• 最終ジャッジ日時: 2024-02-16 22:01:43
• 合計ジャッジ時間: 7,119 ms
• ジャッジサーバーID （参考情報）: judge13 / judge15

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	83 ms; 6,676 KB
testcase_01	AC	195 ms; 6,676 KB
testcase_02	AC	233 ms; 6,676 KB
testcase_03	AC	2 ms; 6,676 KB
testcase_04	AC	2 ms; 6,676 KB
testcase_05	AC	2 ms; 6,676 KB
testcase_06	AC	110 ms; 6,676 KB
testcase_07	AC	181 ms; 6,676 KB
testcase_08	AC	172 ms; 6,676 KB
testcase_09	AC	181 ms; 6,676 KB
testcase_10	AC	236 ms; 6,676 KB
testcase_11	AC	237 ms; 6,676 KB
testcase_12	AC	235 ms; 6,676 KB
testcase_13	AC	238 ms; 6,676 KB
testcase_14	AC	236 ms; 6,676 KB
testcase_15	AC	226 ms; 6,676 KB
testcase_16	AC	197 ms; 6,676 KB
testcase_17	AC	198 ms; 6,676 KB
testcase_18	AC	108 ms; 6,676 KB
testcase_19	AC	150 ms; 6,676 KB
testcase_20	AC	137 ms; 6,676 KB
testcase_21	AC	170 ms; 6,676 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
int par[100010], siz[100010];
void init(int N) {
	iota(par, par + N, 0);
	fill(siz, siz + N, 1);
}
int root(int v) {
	return par[v] = (v == par[v] ? v : root(par[v]));
}
void merge(int a, int b) {
	a = root(a); b = root(b);
	if (a == b) return;
	if (siz[a] < siz[b]) swap(a, b);
	par[b] = a;
	siz[a] += siz[b];
}
int main () {
	int N, M;
	cin >> N >> M;
	using P = pair<int, int>;
	std::vector<P> E(M);
	for (auto& [a, b] : E) {
		cin >> a >> b;
		a --; b --;
	}
	vector<int> C(N);
	for (auto& a : C) {
		cin >> a;
		a --;
	}
	using ll = long long;
	ll W[10];
	for (auto& a : W) cin >> a;
	int Q;
	cin >> Q;
	vector<ll> ans(Q, (ll)1e18);
	vector<P> que(Q);
	for (auto& [a, b] : que) {
		cin >> a >> b;
		a --; b --;
	}
	for (int s = 1; s < (1 << 10); s ++) {
		ll w = 0;
		for (int i = 0; i < 10; i ++) {
			if ((s >> i) & 1) {
				w += W[i];
			}
		}
		init(N);
		for (auto [a, b] : E) {
			if ((s >> C[a]) & (s >> C[b]) & 1) {
				merge(a, b);
			}
		}
		for (int i = 0; i < Q; i ++) {
			auto [a, b] = que[i];
			if (root(a) == root(b)) {
				ans[i] = min(ans[i], w);
			}
		}
	}
	for (auto a : ans) {
		cout<< (a >= (ll)1e17 ? -1ll : a) << endl;
	}
}