骑士(树形dp)

时间:2023-03-08 19:45:48
骑士(树形dp)

题意:给你一个基环树森林,每个点有一个权值,一条边上的两个节点不能同时选择。选取任意个节点,求最大权值和

对于每颗基环树:找环→断边→树形dp(没有上司的舞会)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define MAXN 1000005
#define LL long long
using namespace std; inline int read()
{
int f=,x=;
char ch=getchar();
while(ch<'' || ch>'') {if(ch=='-') f=-; ch=getchar();}
while(ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
return x*f;
} int n,cnt=-;
int val[MAXN];
int v[MAXN<<],head[MAXN],nxt[MAXN<<];
int k1,k2,e;
bool book[MAXN];
LL f[MAXN][],nans,ans; void add(int x,int y)
{
v[++cnt]=y;
nxt[cnt]=head[x];
head[x]=cnt;
} void find_circle(int x,int fa)
{
book[x]=;
for(int i=head[x];i!=-;i=nxt[i])
{
int t=v[i];
if(t==fa) continue;
if(book[t])
{
k1=x;
k2=t;
e=i;
continue;
}
find_circle(t,x);
}
} void dfs(int x,int fa)
{
f[x][]=val[x];
f[x][]=;
for(int i=head[x];i!=-;i=nxt[i])
{
int t=v[i];
if(i==e || (i^)==e) continue;
if(t==fa) continue;
dfs(t,x);
f[x][]+=max(f[t][],f[t][]);
f[x][]+=f[t][];
}
} int main()
{
memset(head,-,sizeof(head));
int i;
int x;
n=read();
for(i=;i<=n;i++)
{
val[i]=read();
x=read();
add(i,x);
add(x,i);
}
for(i=;i<=n;i++)
{
if(!book[i])
{
find_circle(i,-);
dfs(k1,-);
nans=f[k1][];
dfs(k2,-);
nans=max(nans,f[k2][]);
ans+=nans;
}
}
printf("%lld",ans);
return ;
}