这道也是一道基环树的题目,建议先做我的博客这道题再做
这道题与那道题很像,只用把无向边改为有向边,因为没有保证联通,所以还要用并查集判断一下
记得开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;
}