結果

問題 No.2618 除霊
ユーザー TairitsuMeowTairitsuMeow
提出日時 2024-01-26 21:57:23
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 250 ms / 2,000 ms
コード長 1,653 bytes
コンパイル時間 1,805 ms
コンパイル使用メモリ 171,008 KB
実行使用メモリ 45,440 KB
最終ジャッジ日時 2024-09-28 08:07:32
合計ジャッジ時間 13,256 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 11 ms
15,488 KB
testcase_01 AC 11 ms
15,488 KB
testcase_02 AC 11 ms
15,488 KB
testcase_03 AC 11 ms
15,488 KB
testcase_04 AC 11 ms
15,488 KB
testcase_05 AC 12 ms
15,488 KB
testcase_06 AC 204 ms
32,128 KB
testcase_07 AC 202 ms
31,888 KB
testcase_08 AC 203 ms
32,848 KB
testcase_09 AC 209 ms
32,512 KB
testcase_10 AC 224 ms
33,536 KB
testcase_11 AC 165 ms
31,700 KB
testcase_12 AC 250 ms
45,440 KB
testcase_13 AC 108 ms
28,188 KB
testcase_14 AC 130 ms
30,584 KB
testcase_15 AC 125 ms
31,988 KB
testcase_16 AC 128 ms
31,820 KB
testcase_17 AC 135 ms
33,072 KB
testcase_18 AC 135 ms
33,096 KB
testcase_19 AC 129 ms
32,324 KB
testcase_20 AC 133 ms
37,012 KB
testcase_21 AC 138 ms
32,272 KB
testcase_22 AC 137 ms
34,064 KB
testcase_23 AC 132 ms
32,548 KB
testcase_24 AC 132 ms
32,044 KB
testcase_25 AC 140 ms
33,180 KB
testcase_26 AC 134 ms
33,028 KB
testcase_27 AC 128 ms
31,992 KB
testcase_28 AC 135 ms
32,488 KB
testcase_29 AC 144 ms
33,084 KB
testcase_30 AC 147 ms
33,844 KB
testcase_31 AC 250 ms
44,544 KB
testcase_32 AC 202 ms
32,128 KB
testcase_33 AC 208 ms
32,384 KB
testcase_34 AC 198 ms
31,744 KB
testcase_35 AC 207 ms
32,256 KB
testcase_36 AC 209 ms
32,768 KB
testcase_37 AC 206 ms
32,512 KB
testcase_38 AC 233 ms
33,152 KB
testcase_39 AC 226 ms
33,280 KB
testcase_40 AC 222 ms
33,024 KB
testcase_41 AC 226 ms
33,536 KB
testcase_42 AC 226 ms
33,536 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(po[fa[i]]&&ent[fa[i]]+tag[fa[i]]==0)res--;
		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]+(po[j]-tag[i]-tag[j]?1:0);
		}
		ll j=fa[i];
		if(j){
			res+=sum[j]-tr[i];
			if(po[j]-tag[i]-tag[j])res++;
		}
		write(res),putchar('\n');
	}
	return 0;
}
0