BZOJ1040 [ZJOI2008]骑士(基环树+树形dp)

【题解】

出现了基环森林,还是考虑断环为链,然后dp

对每个连通块,随意找到环上一边并断开,记该边的端点为x,y
由于x,y的关联断了,所以要限制x,y的情况:
1. x不取,y取上 
2. y不取,x取上 
然后就是树形dp:记 f[i][0]表示不取i时的最大战斗力; f[i][1]表示取i时的最大战斗力,以上两种情况取最优值即可 

注意:
1. 处理破环为链的问题,使用邻接表比较合适 
2. 对x,y作出限制时,若规定取x不取y,实际很难限制y取不取,那就改为规定不取x,y随意即可


【代码】

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef long long LL;
LL a[1000005],f[1000005][2];
int v[2000005],first[1000005],next[2000005],fan[2000005],vis[2000005];
int e=0,U,V,E;
LL max(LL a,LL b)
{
	if(a>b) return a;
	return b;
}
void tj(int x,int y)
{
	v[++e]=y;
	next[e]=first[x];
	first[x]=e;
}
void dfs(int x,int Efa)
{
	int i;
	vis[x]=1;
	for(i=first[x];i!=0;i=next[i])
		if(i!=Efa&&fan[i]!=Efa)
		{
			if(vis[v[i]]==1)
			{
				U=x;
				V=v[i];
				E=i;
				continue;
			}
			dfs(v[i],i);
		}
}
void solve(int x,int fa,int banE)
{
	int i;
	f[x][1]=a[x];
	for(i=first[x];i!=0;i=next[i])
		if(v[i]!=fa&&i!=banE&&fan[i]!=banE)
		{
			solve(v[i],x,banE);
			f[x][0]+=max(f[v[i]][0],f[v[i]][1]);
			f[x][1]+=f[v[i]][0];
		}
}
int main()
{
	LL ans=0,t=0;
	int n,i,x;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
	{
		scanf("%lld%d",&a[i],&x);
		tj(i,x);
		tj(x,i);
	}
	for(i=1;i<=e;i++)
	{
		if(i&1) fan[i]=i+1;
		else fan[i]=i-1;
	}
	for(i=1;i<=n;i++)
		if(vis[i]==0)
		{
			dfs(i,0);
			memset(f,0,sizeof(f));
			solve(U,0,E);
			t=f[U][0];
			memset(f,0,sizeof(f));
			solve(V,0,E);
			t=max(t,f[V][0]);
			ans+=t;
		}
	printf("%lld",ans);
	return 0;
}


你可能感兴趣的:(DFS,树形DP,基环树)