Codeforces - Maximum White Subtree

10736 단어 Codeforces트리
제목 링크: Codeforces - Maximum White Subtree
비교적 누드와 뿌리를 바꾸는 것은 dp[x]가 자수 중 가장 큰 값을 표시하면 된다.
AC 코드:
#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include
//#define int long long
using namespace std;
const int N=2e5+10;
int n,a[N],dp[N],f[N],res[N],vis[N],ans;
vector<int> g[N];
void dfs1(int x,int fa){
	dp[x]=(a[x]==1?1:-1);	int tmp=0;
	for(auto to:g[x]){
		if(to==fa)	continue;	dfs1(to,x);
		tmp+=max(0,dp[to]);
	}
	dp[x]+=tmp;
}
void dfs2(int x,int fa){
	for(auto to:g[x]){
		if(to==fa)	continue;
		res[to]=max(dp[to]+(res[x]-max(0,dp[to])),dp[to]);
		dfs2(to,x);
	}
}
signed main(){
	cin>>n;
	for(int i=1;i<=n;i++)	scanf("%d",&a[i]);
	for(int i=1,a,b;i<n;i++)	
		scanf("%d %d",&a,&b),g[a].push_back(b),g[b].push_back(a);
	dfs1(1,1);	res[1]=dp[1];	dfs2(1,1);
	for(int i=1;i<=n;i++)	printf("%d ",res[i]);
	return 0;
}
마인드 블로잉 VS 코드 트릭 🤫
CCF CSP 201509 - 4 고속도로 [Kosaraju]

좋은 웹페이지 즐겨찾기

개발자 우수 사이트 수집

개발자가 알아야 할 필수 사이트 100선 추천 우리는 당신을 위해 100개의 자주 사용하는 개발자 학습 사이트를 정리했습니다
best100-homepage