結果

問題 No.1154 シュークリームゲーム（Hard）
コンテスト
ユーザー Super-Mini-PEKKA
提出日時 2026-01-11 13:07:29
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 14 ms / 2,000 ms
コード長 1,458 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,182 ms
コンパイル使用メモリ 222,976 KB
実行使用メモリ 14,788 KB
最終ジャッジ日時 2026-01-11 13:07:51
合計ジャッジ時間 3,952 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
template<typename T>
ll SZ(const T &a){
	return a.size();
}
constexpr ll N=200010;
constexpr ll inf=1e18;
ll n,a[N],fa[N];
vector<ll> T[N];
void dfs(ll u){
	for(ll v:T[u]){
		fa[v]=u;
		T[v].erase(find(T[v].begin(),T[v].end(),u));
		dfs(v);
	}
}
int main(){
	ios::sync_with_stdio(0),cin.tie(0);
	cin>>n;
	for(ll i=1;i<=n;i++){
		cin>>a[i];
	}
	T[0].push_back(1);
	for(ll i=1,u,v;i<n;i++){
		cin>>u>>v;
		T[u].push_back(v);
		T[v].push_back(u);
	}
	dfs(1);
	ll s=0;
	a[0]=-inf;
	while(1){
		ll rt=0;
		for(ll u=1;u<=n;u++){
			if(SZ(T[u])){
				ll ok=1;
				for(ll v:T[u]){
					if(SZ(T[v])){
						ok=0;
						break;
					}
				}
				if(ok){
					rt=u;
					break;
				}
			}
		}
		if(!rt){
			break;
		}
		ll son=rt;
		for(ll u:T[rt]){
			if(a[u]>a[son]){
				son=u;
			}
		}
		if(son==rt){
			for(ll u:T[rt]){
				fa[u]=fa[rt];
				T[fa[u]].push_back(u);
			}
			T[rt].clear();
			continue;
		}
		a[rt]-=a[son];
		T[rt].erase(find(T[rt].begin(),T[rt].end(),son));
		if(!SZ(T[rt])){
			s+=a[rt];
			T[fa[rt]].erase(find(T[fa[rt]].begin(),T[fa[rt]].end(),rt));
			continue;
		}
		son=0;
		for(ll u:T[rt]){
			if(a[u]>a[son]){
				son=u;
			}
		}
		a[rt]+=a[son];
		T[rt].erase(find(T[rt].begin(),T[rt].end(),son));
	}
	sort(T[0].begin(),T[0].end(),[&](ll u,ll v)->ll{
		return a[u]>a[v];
	});
	ll o=1,ans=0;
	for(ll u:T[0]){
		ans+=o*a[u];
		o=-o;
	}
	ans+=o*s;
	cout<<ans<<"\n";
	return 0;
}
0