内向树dp~
就是先找环，任取环上有边相连两点，u和v，以u为根，断开u和v之间的边，做两次树形dp，dp[i][0]表示i不选，dp[i][1]表示i选
①强制u不选，v随意
②u随意，v不选
两种情况取最大值即可~
View Code
1 #include <iostream> 2 #include <cstring> 3 #include <cstdio> 4 #include <cstdlib> 5 #include <algorithm> 6 7 #define N 2000000 8 #define M 4000000 9 10 using namespace std; 11 12 int head[N],next[M],to[M],bh[N]; 13 int n,ban,rp,cnt,root; 14 long long val[N],ans,res,dp[N][2]; 15 bool vis[N],v[N]; 16 17 inline void add(int u,int v) 18 { 19 to[cnt]=v; next[cnt]=head[u]; head[u]=cnt++; 20 } 21 22 inline void read() 23 { 24 memset(head,-1,sizeof head); cnt=0; 25 scanf("%d",&n); 26 for(int i=1,a;i<=n;i++) 27 { 28 scanf("%lld%d",&val[i],&a); 29 add(i,a); add(a,i); 30 } 31 } 32 33 inline void circle(int u,int fa) 34 { 35 for(int i=head[u];~i;i=next[i]) 36 { 37 if(!bh[to[i]]) 38 { 39 bh[to[i]]=true; 40 circle(to[i],u); 41 } 42 else if(to[i]!=fa) rp=u,root=to[i],ban=i; 43 } 44 } 45 46 inline void dfscir(int u) 47 { 48 dp[u][0]=0; dp[u][1]=val[u]; 49 for(int i=head[u];~i;i=next[i]) 50 if(i!=ban&&(i^1)!=ban&&!vis[to[i]]) 51 { 52 vis[to[i]]=true; 53 dfscir(to[i]); 54 dp[u][1]+=dp[to[i]][0]; 55 dp[u][0]=max(dp[to[i]][0],dp[to[i]][1])+dp[u][0]; 56 } 57 } 58 59 inline void dfsban(int u) 60 { 61 dp[u][0]=0; dp[u][1]=val[u]; 62 for(int i=head[u];~i;i=next[i]) 63 if(i!=ban&&(i^1)!=ban&&!v[to[i]]) 64 { 65 v[to[i]]=true; 66 dfsban(to[i]); 67 dp[u][1]+=dp[to[i]][0]; 68 if(to[i]==rp) dp[u][0]+=dp[to[i]][0]; 69 else dp[u][0]+=max(dp[to[i]][0],dp[to[i]][1]); 70 } 71 } 72 73 inline void go() 74 { 75 for(int i=1;i<=n;i++) 76 if(!bh[i]) 77 { 78 root=-1;bh[i]=true; 79 circle(i,-1); 80 vis[root]=true; dfscir(root); res=dp[root][0]; 81 v[root]=true; dfsban(root); res=max(res,max(dp[root][0],dp[root][1])); 82 ans+=res; 83 } 84 printf("%lld\n",ans); 85 } 86 87 int main() 88 { 89 read(); 90 go(); 91 return 0; 92 }