BZOJ 1040: [ZJOI2008]骑士

看题面戳我

这道也是一道基环树的题目，建议先做我的博客这道题再做

这道题与那道题很像，只用把无向边改为有向边，因为没有保证联通，所以还要用并查集判断一下

记得开long long

#include<cstdio>
#include<iostream>
#define ll long long
using namespace std;

const int N=1e6+5;
int n,st[N],top,num,b[N],a[N],ff[N];
ll ans,t,f[N][2],g[N][2];
int to[N],nxt[N],cnt,he[N];
bool fl[N];

inline void add(int u,int v)
{
	to[++cnt]=v,nxt[cnt]=he[u],he[u]=cnt;
}

void dfs1(int u)
{
	fl[u]=1; st[++top]=u;
	for(int e=he[u];e;e=nxt[e])
	{
		int v=to[e];
		if(fl[v])
		{
			b[++num]=v;
			while(top&&st[top]!=v) b[++num]=st[top--];	
		}else dfs1(v);
		if(num) return;
	}
	top--;
}

void dfs(int u,int jj1,int jj2)
{
	f[u][1]=a[u];
	for(int e=he[u];e;e=nxt[e])
	{
		int v=to[e];
		if(v!=jj1&&v!=jj2) 
		{
			dfs(v,jj1,jj2);
			f[u][0]+=max(f[v][1],f[v][0]);
			f[u][1]+=f[v][0];
		}
	}
}

int find(int u)
{
	return ff[u]==u?u:ff[u]=find(ff[u]);
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) ff[i]=i;
	for(int i=1;i<=n;i++)
	{
		int v; scanf("%d%d",&a[i],&v);
		add(v,i); 
		int f1=find(i),f2=find(v);
		if(f1!=f2) ff[f1]=f2;
	}
	for(int j=1;j<=n;j++)
		if(find(j)==j)
		{
			num=0;
			dfs1(j);
			for(int i=1;i<=num;i++)
				dfs(b[i],(i==1)?b[num]:b[i-1],(i==num)?b[1]:b[i+1]);
			for(int i=1;i<=num;i++) g[i][0]=g[i][1]=0;
			g[1][0]=f[b[1]][0],g[1][1]=0;
			for(int i=2;i<=num;i++)
			{
				g[i][0]=max(g[i-1][0],g[i-1][1])+f[b[i]][0];
				g[i][1]=g[i-1][0]+f[b[i]][1];
			}
			t=max(g[num][0],g[num][1]);
			for(int i=1;i<=num;i++) g[i][0]=g[i][1]=0;
			g[1][0]=0,g[1][1]=f[b[1]][1];
			for(int i=2;i<=num;i++)
			{
				g[i][0]=max(g[i-1][0],g[i-1][1])+f[b[i]][0];
				g[i][1]=g[i-1][0]+f[b[i]][1];
			}
			t=max(t,g[num][0]);
			ans+=t;
		}
	printf("%lld",ans);
	return 0;
}