P2607 [ZJOI2008]骑士

本题本质上就是树dp，和没有上司的舞会差不多，只不过多了一个对基环树的处理。

#include<iostream>

#include<cstring>

#include<string>

#include<cstdio>

#define ll long long

using namespace std;

const int maxn=2e6;

int n,cnt;

ll val[maxn];

int root;

int head[maxn];

int vis[maxn];

ll fa[maxn];

ll f[maxn][2];

ll ans;

int rot;

ll max1(ll x,ll y){

	if(x>=y){

		return x;

	}

	return y;

}

inline ll read(){

	ll ret=0,f=1;

	char ch=getchar();

	while(ch<'0'||ch>'9'){

		if(ch=='-'){

			f=-f;

		}

		ch=getchar();

	}

	while(ch<='9'&&ch>='0'){

		ret=ret*10+(ch^'0');

		ch=getchar();

	}

	return f*ret;

}

struct node{

	int nex,to;

}e[maxn];

void add(int u,int v){

	cnt++;

	e[cnt].nex=head[u];

	e[cnt].to=v;

	head[u]=cnt;

}

void dp(int x){

	vis[x]=1;//打标记，用处是表明联通块，把该联通块进行标记

	//以便遍历整连接关系，除此之外并无用处

	f[x][1]=val[x];

	f[x][0]=0;

	for(int i=head[x];i;i=e[i].nex){

		int to=e[i].to;

		if(to!=root){

			dp(to);

			f[x][0]+=max1(f[to][0],f[to][1]);

			f[x][1]+=f[to][0];

		}

		else{

			f[to][1]=-maxn;//将环删边，定有一点不能选，故定此点不选

		}

	}

}

void find(int x){

	vis[x]=1;

	root=x;

	while(!vis[fa[root]]){//我们的目的是为了找环，而且该连通块先前一定未遍历过

		//故该方法可以找出环来

		root=fa[root];

		vis[root]=1;

	}

	/*

	该题中图的特殊性

	导致找环必须要从下到上

	若从上到下

	那么如果起点为树边就无法找到环

	该图的根为环

	*/

	dp(root);

	ll t;

	t=max(f[root][1],f[root][0]);//删一条边不选边上一点的情况下，该联通块的最大值

	root=fa[root];

	dp(root);

	ans+=max1(t,max1(f[root][1],f[root][0])); //将两种情况比较；

}

int main(){

	n=read();

	int x;

	for(int i=1;i<=n;i++){

		val[i]=read();

		x=read();

		add(x,i);

		fa[i]=x;

	}

	for(int i=1;i<=n;i++){

		if(!vis[i]){

			find(i);

		}

	}

	cout<<ans;

	return 0;

}

快去ac吧