【BZOJ1040】骑士

题目：BZOJ1040

解析：

  尝试将每个人痛恨的人向他连一条有向边，容易发现最后整个图是由多个基环外向树组成。对于基环树，常用的方法是断开环中的一条边使其成为一棵树再进行操作。这里也可以将一条边断开，选择这条边的一个端点为根进行树形DP。但注意被删除的边的另一个端点和根不能同时选，所以若要选根进入骑士军团还需更新从该端点到根的路径，详见代码。

  所以BZOJ最终还是凉了，唉。

代码：

#include <bits/stdc++.h>
using namespace std;

const int Max=1000005;
int n,m,s,rt,fa[Max],first[Max],num[Max],cnt[Max],vis[Max];
long long f[Max][2],ans;
struct shu{int to,next;}edge[Max<<1];

inline int get_int()
{
	int x=0,f=1;char c;
	for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
	if(c=='-') f=-1,c=getchar();
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
	return x*f;
}

inline void build(int x,int y){edge[++s].next=first[x],first[x]=s,edge[s].to=y,fa[y]=x;}

inline int init()
{
	n=get_int();
	for(int i=1;i<=n;i++) num[i]=get_int(),build(get_int(),i);
}

inline void dfs(int p)
{
	f[p][1]=num[p],vis[p]=1;
	for(int u=first[p];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  if(vis[to]) continue;
	  dfs(to);
	  f[p][1]+=f[to][0];
	  f[p][0]+=max(f[to][1],f[to][0]);
	}
}

inline void dp(int x)
{
	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][1]=num[x],f[x][0]=0; 
	  for(int u=first[x];u;u=edge[u].next)
	  {
	  	int to=edge[u].to;
	  	f[x][1]+=f[to][0];
	  	f[x][0]+=max(f[to][1],f[to][0]);
	  }
	}
	f[rt][1]=num[rt];
	for(int u=first[rt];u;u=edge[u].next)
	{
	  int to=edge[u].to;
	  f[x][1]+=f[to][0];
	}
	ans+=max(f[rt][0],f[rt][1]);
}

inline void solve()
{
	for(int i=1;i<=n;i++) if(!vis[i]) dp(i);
	cout<<ans;
}

int main()
{
	freopen("lx.in","r",stdin);
	init();
	solve();
	return 0;
}