結果

問題 No.2618 除霊
ユーザー TairitsuMeowTairitsuMeow
提出日時 2024-01-26 21:48:12
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,598 bytes
コンパイル時間 1,637 ms
コンパイル使用メモリ 171,852 KB
実行使用メモリ 56,700 KB
最終ジャッジ日時 2024-09-28 08:00:31
合計ジャッジ時間 9,167 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 7 ms
24,604 KB
testcase_01 AC 7 ms
25,272 KB
testcase_02 AC 9 ms
24,468 KB
testcase_03 WA -
testcase_04 AC 8 ms
25,036 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 154 ms
45,980 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 AC 86 ms
41,752 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 AC 141 ms
41,720 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// Problem: No.2618 除霊
// Contest: yukicoder
// URL: https://yukicoder.me/problems/no/2618
// Memory Limit: 512 MB
// Time Limit: 2000 ms

#include<bits/stdc++.h>
#define debug(x) cerr<<(#x)<<" "<<(x)<<endl
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define pii pair<ll,ll>
#define rep(i,a,b) for(ll i=(a);i<=(b);++i)
#define per(i,a,b) for(ll i=(a);i>=(b);--i)
using namespace std;
bool Mbe; 
ll read(){
	ll x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void write(ll x){
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
}
const ll N=5e5+9;
ll n,m,a[N],po[N],sum[N],tr[N],tag[N],ent[N],fa[N];
vector<ll>to[N];
void dfs(ll x,ll f){
	fa[x]=f;
	for(ll y:to[x]){
		if(y==f)continue;
		dfs(y,x);
		sum[x]+=tr[y];
		tr[x]+=tr[y];
		if(!ent[y]&&!tag[y]&&tag[x])tr[x]++;
		ent[x]+=tag[y];
	}
	tr[x]+=(ent[x]||tag[x]);
}
bool Med;
int main(){
	cerr<<fabs(&Med-&Mbe)/1048576.0<<"MB\n";
	n=read();
	rep(i,2,n){
		ll x=read(),y=read();
		to[x].push_back(y);
		to[y].push_back(x);
	}
	m=read();
	rep(i,1,m)a[i]=read(),tag[a[i]]=1;
	rep(i,1,m){
		po[a[i]]++;
		for(ll j:to[a[i]])po[j]++;
	}
	dfs(1,0);
	rep(i,1,n){
		ll res=tr[1]-(i==1?tr[1]:tr[fa[i]]);
		if(fa[fa[i]]){
			ll c=po[fa[fa[i]]]-tag[fa[i]];
			if(!c&&po[fa[fa[i]]])res--;
		}
		for(ll j:to[i]){
			if(j==fa[i])continue;
			res+=sum[j]+(ent[j]?1:0);
		}
		ll j=fa[i];
		if(j){
			res+=sum[j]-tr[i];
			if(tag[fa[j]]||ent[j]-tag[i])res++;
		}
		write(res),putchar('\n');
	}
	return 0;
}
0