C++——树形DP模板——骑士【浙江省选2008】

题目背景

ZJOI2008 DAY2 T4

题目描述

Z 国的骑士团是一个很有势力的组织，帮会中汇聚了来自各地的精英。他们劫富济贫，惩恶扬善，受到社会各界的赞扬。

最近发生了一件可怕的事情，邪恶的Y国发动了一场针对 Z 国的侵略战争。战火绵延五百里，在和平环境中安逸了数百年的Z国又怎能抵挡的住 Y 国的军队。于是人们把所有的希望都寄托在了骑士团的身上，就像期待有一个真龙天子的降生，带领正义打败邪恶。

骑士团是肯定具有打败邪恶势力的能力的，但是骑士们互相之间往往有一些矛盾。每个骑士都有且仅有一个自己最厌恶的骑士（当然不是他自己），他是绝对不会与自己最厌恶的人一同出征的。

战火绵延，人民生灵涂炭，组织起一个骑士军团加入战斗刻不容缓！国王交给了你一个艰巨的任务，从所有的骑士中选出一个骑士军团，使得军团内没有矛盾的两人（不存在一个骑士与他最痛恨的人一同被选入骑士军团的情况），并且，使得这支骑士军团最具有战斗力。

为了描述战斗力，我们将骑士按照 1 至 N 编号，给每名骑士一个战斗力的估计，一个军团的战斗力为所有骑士的战斗力总和。

输入格式

输入第一行包含一个正整数 N ，描述骑士团的人数。
接下来 N 行，每行两个正整数，按顺序描述每一名骑士的战斗力和他最痛恨的骑士。

输出格式

输出应包含一行，包含一个整数，表示你所选出的骑士军团的战斗力。

样例数据 1

输入　 [复制]

3
10 2
20 3
30 1

输出

30

备注

【数据范围】
对于 30% 的测试数据，满足：N≤10；
对于 60% 的测试数据，满足：N≤100；
对于 80% 的测试数据，满足：N≤10000。

对于 100% 的测试数据，满足：N≤1000000，每名骑士的战斗力都是不大于 1000000 的正整数。

#include
#include
#include
#include
#include
#include
using namespace std;
const int N=1e6+3;
int n,tot;
char ch;
int val[N],fa[N],cnt[N],info[N],nxt[N],go[N];
bool vis[N];
long long ans,f[N][2];
inline void sete(const int &y)
{
	int x;
	cin>>x;
	nxt[++tot]=info[x];
	info[x]=tot;
	go[tot]=y;
	fa[y]=x;
}
inline void dfs(const int &x)
{
	int y;
	vis[x]=true;
	f[x][1]=val[x];
	for(int k=info[x];y=go[k],k;k=nxt[k])
		if(!vis[y])
		{
			dfs(y);
			f[x][0]+=max(f[y][0],f[y][1]);
			f[x][1]+=f[y][0];
		}
}
inline void dp(int x)
{
	int rt,y;
	for(rt=x;cnt[rt]!=x;rt=fa[rt])
		cnt[rt]=x;
	dfs(rt);
	x=fa[rt];
	f[x][1]=f[x][0];
	for(x=fa[x];x!=rt;x=fa[x])
	{
		f[x][0]=0;
		f[x][1]=val[x];
		for(int k=info[x];y=go[k],k;k=nxt[k])
		{
			f[x][0]+=max(f[y][0],f[y][1]);
			f[x][1]+=f[y][0];
		}
	}
	f[rt][1]=val[rt];
	for(int k=info[rt];y=go[k],k;k=nxt[k])
		f[rt][1]+=f[y][0];
	ans+=max(f[rt][0],f[rt][1]);
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>val[i];
		sete(i);
	}
	for(int i=1;i<=n;++i)
		if(!vis[i]) dp(i);
	cout<