結果
問題 | No.2630 Colorful Vertices and Cheapest Paths |
ユーザー |
|
提出日時 | 2024-02-16 21:52:15 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 374 ms / 2,500 ms |
コード長 | 1,302 bytes |
コンパイル時間 | 2,006 ms |
コンパイル使用メモリ | 202,676 KB |
最終ジャッジ日時 | 2025-02-19 13:50:57 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; #include <atcoder/dsu> using namespace atcoder; int main() { int N, M; cin >> N >> M; vector<vector<int>> G(N); for (int i = 0; i< M; i++) { int a, b; cin >> a >> b; a--, b--; G[a].push_back(b); G[b].push_back(a); } vector<int> C(N); for (int i = 0; i < N; i++) { cin >> C[i]; C[i]--; } vector<ll> W(10); for (int i = 0; i < 10; i++) { cin >> W[i]; } int Q; cin >> Q; vector<int> u(Q), v(Q); for (int i = 0; i < Q; i++) { cin >> u[i] >> v[i]; u[i]--, v[i]--; } const ll inf = 1e18; vector<ll> ans(Q, inf); for (int s = 0; s < (1 << 10); s++) { dsu Union(N); for (int i = 0; i < N; i++) { if (s & (1 << C[i])) { for (int j = 0; j < (int)G[i].size(); j++) { if (s & (1 << C[G[i][j]])) { Union.merge(i, G[i][j]); } } } } ll cost = 0; for (int i = 0; i < 10; i++) { if (s & (1 << i)) { cost += W[i]; } } for (int i = 0; i < Q; i++) { if (Union.same(u[i], v[i])) { ans[i] = min(ans[i], cost); } } } for (int i = 0; i < Q; i++) { if (ans[i] == inf) { ans[i] = -1; } cout << ans[i] << endl; } return 0; }