結果
問題 | No.2630 Colorful Vertices and Cheapest Paths |
ユーザー |
![]() |
提出日時 | 2024-02-29 11:12:55 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 292 ms / 2,500 ms |
コード長 | 2,036 bytes |
コンパイル時間 | 1,930 ms |
コンパイル使用メモリ | 201,764 KB |
最終ジャッジ日時 | 2025-02-19 21:58:28 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; struct UnionFind { int ngroup, N; vector<int> par, siz; UnionFind(int _N) : N(_N), par(_N), siz(_N){ ngroup = _N; for(int i = 0; i < N; i++) par[i] = i, siz[i] = 1; } int root(int x) { if (par[x] == x) return x; return par[x] = root(par[x]); } int unite(int x, int y) { int rx = root(x), ry = root(y); if (rx == ry) return rx; ngroup--; if (siz[rx] > siz[ry]) swap(rx, ry); par[rx] = ry; siz[ry] += siz[rx]; return ry; } bool same(int x, int y) { int rx = root(x), ry = root(y); return rx == ry; } int size(int x) {return siz[root(x)];} int group_count() {return ngroup;} vector<vector<int>> groups(){ vector<int> rev(N); int nrt=0; for (int i=0; i<N; i++) if (root(i) == i) rev[i] = nrt, nrt++; vector<vector<int>> res(nrt); for (int i=0; i<N; i++) res[rev[root(i)]].push_back(i); return res; } }; int main(){ cin.tie(nullptr); ios_base::sync_with_stdio(false); ll N, M, x, y, Q, z=1<<10, sm; cin >> N >> M; vector<ll> A(M+1), B(M+1), C(N+1), W(10); for (int i=1; i<=M; i++) cin >> A[i] >> B[i]; for (int i=1; i<=N; i++){ cin >> C[i]; C[i]--; } for (int i=0; i<10; i++) cin >> W[i]; cin >> Q; vector<ll> u(Q+1), v(Q+1), ans(Q+1, 1e18); for (int i=1; i<=Q; i++) cin >> u[i] >> v[i]; for (int i=0; i<z; i++){ sm = 0; for (int j=0; j<10; j++){ if (i>>j&1) sm += W[j]; } UnionFind uf(N+1); for (int j=1; j<=M; j++){ if (i>>C[A[j]]&1 && i>>C[B[j]]&1) uf.unite(A[j], B[j]); } for (int j=1; j<=Q; j++){ if (uf.same(u[j], v[j])) ans[j] = min(ans[j], sm); } } for (int i=1; i<=Q; i++){ if (ans[i] == 1e18) cout << -1 << endl; else cout << ans[i] << endl; } return 0; }