結果

問題 No.3222 Let the World Forget Me
コンテスト
ユーザー mono
提出日時 2025-08-01 22:18:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 171 ms / 2,000 ms
コード長 2,264 bytes
コンパイル時間 3,952 ms
コンパイル使用メモリ 232,128 KB
実行使用メモリ 13,668 KB
最終ジャッジ日時 2025-08-01 22:19:01
合計ジャッジ時間 8,896 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:78:14: warning: structured bindings only available with ‘-std=c++17’ or ‘-std=gnu++17’ [-Wc++17-extensions]
   78 |         auto [x, i] = pq.top(); pq.pop();
      |              ^

ソースコード

diff # 

#include <bits/stdc++.h>
#include <stdlib.h>
#include <atcoder/all>
using namespace atcoder;
using namespace std;
#define rep(i, a, n) for(ll i = a; i < n; i++)
#define rrep(i, a, n) for(ll i = a; i >= n; i--)
#define inr(l, x, r) (l <= x && x < r)
#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define all(x) (x).begin(), (x).end()
//constexpr ll MOD = 1000000007;
constexpr ll MOD = 998244353;
constexpr int IINF = 1001001001;
constexpr ll INF = 1LL<<60;
template<class t,class u> void chmax(t&a,u b){if(a<b)a=b;}
template<class t,class u> void chmin(t&a,u b){if(b<a)a=b;}

using mint = modint998244353;

ll gcd(ll a, ll b){
    if(a%b == 0){
      return b;
    }else{
      return gcd(b, a%b);
    }
}

ll lcm(ll a, ll b){
    return a*b / gcd(a, b);
}

ll powMod(ll x, ll n, ll mod) {
    if (n == 0) return 1 % mod;
    ll val = powMod(x, n / 2, mod);
    val *= val;
    val %= mod;
    if (n % 2 == 1) val *= x;
    return val % mod;
}

int main() {
    ll n, m; cin >> n >> m;
    vector<ll> p(n), c(m);
    rep(i,0,n) cin >> p[i];
    vector<vector<ll>> g(n);
    rep(i,0,n-1){
        ll a, b; cin >> a >> b;
        a--;b--;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    rep(i,0,m) cin >> c[i], c[i]--;
    priority_queue<pll> pq;
    vector<ll> d(n,-1);
    queue<ll> q;
    for(auto x: c){
        d[x] = 0;
        q.push(x);
    }
    while(!q.empty()){
        ll v = q.front(); q.pop();
        for(auto nv: g[v]){
            if(d[nv] == -1){
                d[nv] = d[v]+1;
                q.push(nv);
            }
        }
    }
    vector<ll> deg(n),check(n,0);
    rep(i,0,n) deg[i] = g[i].size();
    rep(i,0,n){
        if(d[i] != 0 && g[i].size() == 1) pq.push({p[i], i});
    }
    ll ans = 0, cnt = 0;
    while(!pq.empty()){
        auto [x, i] = pq.top(); pq.pop();
        // cout << x << " " << i << " " << cnt <<  " " << d[i] << endl;
        if(d[i] > cnt && !check[i]){
            check[i] = 1;
            cnt++;
            ans += x;
            for(auto nv: g[i]){
                deg[nv]--;
                if(deg[nv] == 1 && d[nv] != 0){
                    pq.push({p[nv], nv});
                }
            }
        }
       
    }
    cout << ans << endl;
    return 0;
}
0