【Luogu】P2607骑士（基环树DP）

　　题目链接

　　这题……好吧我比着题解打的

　　题解连接

#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cctype>
#include<algorithm>
#define maxn 1000200
using std::max;
inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}
 
struct Edge{
    int next,to;
}edge[maxn*];
int head[maxn],num;
inline void add(int from,int to){
    edge[++num]=(Edge){head[from],to};
    head[from]=num;
}
 
int fa[maxn];
long long f[maxn][];
bool vis[maxn];
int q[maxn];
int ban;
long long ans;
void dfs(int x){
    vis[x]=;
    f[x][]=;
    f[x][]=q[x];
    for(int i=head[x];i;i=edge[i].next){
        int to=edge[i].to;
        if(to==ban){
            f[to][]=-;
            continue;
        }
        dfs(to);
        f[x][]+=max(f[to][],f[to][]);
        f[x][]+=f[to][];
    }
    return;
}
 
int main(){
    int n=read();
    for(int i=;i<=n;++i){
        q[i]=read();
        fa[i]=read();
        add(fa[i],i);
    }
    for(int i=;i<=n;++i){
        if(!vis[i]){
            int p=i;
            while(!vis[p]){
                vis[p]=;
                p=fa[p];
            }
            ban=p;
            dfs(p);
            long long now=max(f[p][],f[p][]);
            p=fa[p];ban=p;
            dfs(p);
            ans+=max(now,max(f[p][],f[p][]));
        }
    }
    printf("%lld",ans);
    return ;
}