結果

問題 No.3222 Let the World Forget Me
ユーザー ゼリトキ
提出日時 2025-08-01 22:43:07
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,627 bytes
コンパイル時間 3,616 ms
コンパイル使用メモリ 291,732 KB
実行使用メモリ 15,576 KB
最終ジャッジ日時 2025-08-01 22:43:14
合計ジャッジ時間 5,908 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (ll i = 0; i < (ll)(n); i++)
#define ll long long
const long long mod=998244353;
const long long hmod=46216567629137;
int main(){
    cin.tie(0)->sync_with_stdio(0);
    cout.tie(0);
    int N,M;
    cin>>N>>M;
    int P[N+1];
    int A[N],B[N];
    int C[M+1];
    for(int i=1;i<=N;i++) cin>>P[i];
    vector<int>G[N+1];
    for(int i=1;i<=N-1;i++){
        cin>>A[i]>>B[i];
        G[A[i]].push_back(B[i]);
        G[B[i]].push_back(A[i]);
    }
    for(int i=1;i<=M;i++) cin>>C[i];
    int dist[N+1];
    bool already[N+1];
    for(int i=1;i<=N;i++){
        dist[i]=1e9;
        already[i]=0;
    }
    queue<int>Q;
    for(int i=1;i<=M;i++){
        Q.push(C[i]);
        dist[C[i]]=0;
    }
    while(!Q.empty()){
        int pos=Q.front();
        Q.pop();
        if(already[pos]) continue;
        already[pos]=1;
        for(int i=0;i<G[pos].size();i++){
            int nex=G[pos][i];
            if(already[nex]) continue;
            if(dist[nex]>dist[pos]+1){
                dist[nex]=dist[pos]+1;
                Q.push(nex);
            }
        }
    }
    vector<pair<int,int>>polute[N+1];
    for(int i=1;i<=N;i++){
        polute[dist[i]].push_back({-P[i],i});
    }
    set<pair<int,int>>leaf;
    for(int i=1;i<=N;i++){
        if(G[i].size()==1) leaf.insert({-P[i],i});
    }
    ll ans=0;
    for(int i=1;i<=N;i++){
        if(leaf.size()==0) break;
        ans-=(*leaf.begin()).first;
        leaf.erase(*leaf.begin());
        rep(j,polute[i].size()){
            leaf.erase(polute[i][j]);
        }
    }
    cout<<ans<<"\n";
}
0