結果
問題 |
No.3222 Let the World Forget Me
|
ユーザー |
![]() |
提出日時 | 2025-08-15 20:12:30 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 134 ms / 2,000 ms |
コード長 | 1,663 bytes |
コンパイル時間 | 4,183 ms |
コンパイル使用メモリ | 260,056 KB |
実行使用メモリ | 11,316 KB |
最終ジャッジ日時 | 2025-08-15 20:12:39 |
合計ジャッジ時間 | 7,892 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; #define rep(i, n) for(int i=0;i<(n);++i) #define rep1(i, n) for(int i=1;i<=(n);i++) #define ll long long using mint = modint998244353; using P = pair<ll,ll>; using lb = long double; using T = tuple<ll, ll, ll>; #ifdef LOCAL # include <debug_print.hpp> # define dbg(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__) #else # define dbg(...) (static_cast<void>(0)) #endif int main() { int n, m; cin >> n >> m; vector<int> p(n); rep(i,n) cin >> p[i]; vector<vector<int>> g(n); vector<int> d(n); rep(i,n-1) { int a, b; cin >> a >> b; --a;--b; g[a].push_back(b); g[b].push_back(a); d[a]++; d[b]++; } vector<int> c(m); rep(i,m) cin >> c[i]; rep(i,m) --c[i]; priority_queue<P> pq; rep(i,n) { if(d[i]==1) pq.emplace(p[i],i); } vector<int> used(n); queue<int> q; rep(i,m) { used[c[i]]=1; q.push(c[i]); } ll ans = 0; while(!pq.empty()) { dbg(pq,used); auto [val, id] = pq.top();pq.pop(); if(used[id]) continue; ans += val; for(int v : g[id]) { d[v]--; if(d[v]==1) pq.emplace(p[v], v); } vector<int> vs; while(!q.empty()) { int u = q.front();q.pop(); for(int v : g[u]) { if(used[v]) continue; vs.push_back(v); used[v] = 1; } } for(int v : vs) q.push(v); // dbg(pq,q); } cout << ans << endl; return 0; }