結果
問題 |
No.3222 Let the World Forget Me
|
ユーザー |
|
提出日時 | 2025-08-02 10:59:02 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 227 ms / 2,000 ms |
コード長 | 1,817 bytes |
コンパイル時間 | 2,468 ms |
コンパイル使用メモリ | 213,960 KB |
実行使用メモリ | 20,232 KB |
最終ジャッジ日時 | 2025-08-02 10:59:11 |
合計ジャッジ時間 | 8,601 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 31 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,n) for(ll i=0;i<n;i++) #define repl(i,l,r) for(ll i=(l);i<(r);i++) #define per(i,n) for(ll i=(n)-1;i>=0;i--) #define perl(i,r,l) for(ll i=r-1;i>=l;i--) #define all(x) (x).begin(),(x).end() #define CST(x) cout<<fixed<<setprecision(x) #define vtpl(x,y,z) vector<tuple<x,y,z>> #define rev(x) reverse(x); using ll=long long; using vl=vector<ll>; using vvl=vector<vector<ll>>; using pl=pair<ll,ll>; using vpl=vector<pl>; using vvpl=vector<vpl>; const ll MOD=1000000007; const ll MOD9=998244353; const int inf=1e9+10; const ll INF=4e18; const ll dy[9]={1,0,-1,0,1,1,-1,-1,0}; const ll dx[9]={0,1,0,-1,1,-1,1,-1,0}; template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; } template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; } int main(){ ll n,m;cin >> n >> m; vl p(n);rep(i,n)cin >> p[i]; vector<set<ll>> g(n); rep(i,n-1){ ll a,b;cin >> a >> b;a--;b--; g[a].insert(b); g[b].insert(a); } vl time(n,INF); { queue<ll> que; rep(i,m){ ll c;cin >> c;c--; que.push(c);time[c]=0; } while(que.size()){ auto v=que.front();que.pop(); for(auto p:g[v]){ if(time[p]!=INF)continue; time[p]=time[v]+1; que.push(p); } } } priority_queue<pl> que; ll ans=0; rep(i,n){ if(g[i].size()==1)que.push({p[i],i}); } rep(i,n){ while(que.size()&&time[que.top().second]<=i){ que.pop(); } if(que.size()){ auto [val,v]=que.top();que.pop(); ans+=val; ll nxt=*g[v].begin(); g[nxt].erase(v); if(g[nxt].size()==1){ que.push({p[nxt],nxt}); } } } cout << ans << endl; }