求解答,这个做法为什么过不了样例
查看原帖
求解答,这个做法为什么过不了样例
758200
_LDX_WWS_楼主2024/9/28 09:26

先找环,然后对每个基环树上的点做一遍树形dp, 处理出这个点取或不取的dp值,让后枚举相邻两个点的取或不取状态,当成链来做

#include<bits/stdc++.h>
#define test() cout<<*******<<endl;
#define etr() cout<<endl;
using namespace std;
const int N=1e6+10;
int head[N],tot,w[N],vis[N];
struct node{
	int to,nxt;
};
node E[N<<1];
void addedge(int u,int v){
	tot++;
	E[tot].nxt=head[u];
	E[tot].to=v;
	head[u]=tot;
}
int n,is[N],fa[N],fir,sec;
void dfs(int u,int f){
	vis[u]=1;
	fa[u]=f;
	for(int i=head[u];i;i=E[i].nxt){
		int v=E[i].to;
		if(v==f) continue;
		if(vis[v]==2) continue;
		if(vis[v]){
			is[v]=1;
			fir=u;
			sec=v;
			int k=u;
			while(k!=v&&k!=0){
				is[k]=1;
				k=fa[k];
			}
			return;
		}
		else dfs(v,u);
	}
	vis[u]=2;
}
int dp[N][2];
void work(int u,int f){
	dp[u][0]=0;dp[u][1]=w[u];
	for(int i=head[u];i;i=E[i].nxt){
		int v=E[i].to;
		if(v==f||is[v]) continue;
		work(v,u);
		dp[u][0]=max(max(dp[v][1],dp[v][0]),dp[u][0]);
		dp[u][1]=max(dp[v][0]+w[u],dp[u][1]);
	}
}
int ans,f[N][2],a[N][2];
void getans(int u,int ff){
	for(int i=head[u];i;i=E[i].nxt){
		int v=E[i].to;
		if(v==ff||!is[v]) continue;
		if(v==fir) continue;
		f[u][0]=max(max(f[v][1],f[v][0]),f[u][0]);
		f[u][1]=max(f[v][0]+max(a[u][1],a[u][0]),f[u][1]);
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		int x;cin>>w[i]>>x;
		addedge(i,x);
		addedge(x,i);
	}
	dfs(1,0);
	for(int i=1;i<=n;i++) cout<<is[i]<<" ";
	etr();
	for(int i=1;i<=n;i++){
		if(!dp[i][0]&&!dp[i][1]&&is[i]) work(i,0);
	}
	for(int i=1;i<=n;i++) cout<<dp[i][0]<<" "<<dp[i][1]<<endl;
	for(int i=1;i<=n;i++) a[i][0]=dp[i][0],a[i][1]=dp[i][1];
	a[sec][1]=-114514;
	a[fir][0]=-114514;
	getans(fir,0);
	ans=max(ans,f[fir][1]);
	a[sec][1]=dp[sec][1];
	a[fir][0]=dp[fir][0];
	a[fir][1]=-114514;
	a[sec][0]=-114514;
	getans(fir,0);
	ans=max(ans,f[fir][0]);
	a[sec][1]=-114514;
	a[fir][1]=-114514;
	a[fir][0]=dp[fir][0];
	a[sec][0]=dp[sec][0];
	getans(fir,0);
	ans=max(ans,f[fir][0]);
	cout<<ans<<endl;
	return 0;
} 
2024/9/28 09:26
加载中...