結果

問題 No.1154 シュークリームゲーム（Hard）
ユーザー fastmath
提出日時 2020-08-12 02:44:01
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,590 bytes
コンパイル時間 5,134 ms
コンパイル使用メモリ 163,608 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-09 12:27:34
合計ジャッジ時間 6,280 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define ii pair <int, int>
#define app push_back
#define all(a) a.begin(), a.end()
#define bp __builtin_popcountll
#define ll long long
#define mp make_pair
#define f first
#define s second
#define Time (double)clock()/CLOCKS_PER_SEC
#define debug(x) std::cout << #x << ": " << x << '\n';

const int N = 1e3+7;
vector <int> g[N];
int n, a[N], cnt[N], par[N];
bool del[N];
void dfs(int u, int p) {
    par[u] = p;
    for (int v : g[u]) {
        if (v != p) {
            dfs(v, u);
        }   
    }   
}   

signed main() {
    #ifdef HOME
    freopen("input.txt", "r", stdin);
    #else
    #define endl '\n'
    ios_base::sync_with_stdio(0); cin.tie(0);
    #endif
    cin >> n;
    for (int i = 1; i <= n; ++i)
        cin >> a[i];
    for (int i = 0; i < n - 1; ++i) {
        int u, v;
        cin >> u >> v;
        g[u].app(v); g[v].app(u);
    }       
    dfs(1, 1);
    const int INF = 1e18;
    for (int i = 1; i <= n; ++i)
        cnt[i] = 1;
    for (int t = 0; t < n - 1; ++t) {
        ii mx = mp(-INF, -INF);
        for (int u = 2; u <= n; ++u) {
            if (!del[u]) {
                mx = max(mx, mp(a[u], u));
            }   
        }   
        int u = mx.s;
        for (int i = 2; i <= n; ++i) {
            if (par[i] == u) {
                par[i] = par[u];
            }   
        }   
        del[u] = 1;
        if (cnt[par[u]]&1)
            a[par[u]] -= a[u];
        else            
            a[par[u]] += a[u];
        cnt[par[u]] += cnt[u];
    }   
    cout << a[1] << endl;
}
0