結果

問題 No.3222 Let the World Forget Me
コンテスト
ユーザー mono
提出日時 2025-08-01 22:04:28
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,960 bytes
コンパイル時間 3,824 ms
コンパイル使用メモリ 232,368 KB
実行使用メモリ 12,208 KB
最終ジャッジ日時 2025-08-01 22:04:36
合計ジャッジ時間 8,563 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 25 WA * 6
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:76:14: warning: structured bindings only available with ‘-std=c++17’ or ‘-std=gnu++17’ [-Wc++17-extensions]
   76 |     for(auto [x,i]: vec){
      |              ^

ソースコード

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]--;
    vector<pll> vec;
    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);
            }
        }
    }
    rep(i,0,n){
        if(d[i] != 0 && g[i].size() == 1) vec.push_back({p[i], i});
    }
    sort(all(vec), greater<pll>());
    ll ans = 0, cnt = 0;
    for(auto [x,i]: vec){
        // cout << x << " " << i << " " << cnt <<  " " << d[i] << endl;
        if(d[i] > cnt){
            cnt++;
            ans += x;
        }
    }
    cout << ans << endl;
    return 0;
}
0