bzoj1040 基环树上dp

Description

Z国的骑士团是一个很有势力的组织，帮会中汇聚了来自各地的精英。他们劫富济贫，惩恶扬善，受到社会各界的赞扬。最近发生了一件可怕的事情，邪恶的Y国发动了一场针对Z国的侵略战争。战火绵延五百里，在和平环境中安逸了数百年的Z国又怎能抵挡的住Y国的军队。于是人们把所有的希望都寄托在了骑士团的身上，就像期待有一个真龙天子的降生，带领正义打败邪恶。骑士团是肯定具有打败邪恶势力的能力的，但是骑士们互相之间往往有一些矛盾。每个骑士都有且仅有一个自己最厌恶的骑士（当然不是他自己），他是绝对不会与自己最厌恶的人一同出征的。战火绵延，人民生灵涂炭，组织起一个骑士军团加入战斗刻不容缓！国王交给了你一个艰巨的任务，从所有的骑士中选出一个骑士军团，使得军团内没有矛盾的两人（不存在一个骑士与他最痛恨的人一同被选入骑士军团的情况），并且，使得这支骑士军团最具有战斗力。为了描述战斗力，我们将骑士按照1至N编号，给每名骑士一个战斗力的估计，一个军团的战斗力为所有骑士的战斗力总和。

Input

第一行包含一个正整数N，描述骑士团的人数。接下来N行，每行两个正整数，按顺序描述每一名骑士的战斗力和他最痛恨的骑士。

Output

应包含一行，包含一个整数，表示你所选出的骑士军团的战斗力。

Sample Input

3
10 2
20 3
30 1

Sample Output

30
【数据规模】
对于30%的测试数据，满足N ≤ 10；
对于60%的测试数据，满足N ≤ 100；
对于80%的测试数据，满足N ≤ 10 000。
对于100%的测试数据，满足N ≤ 1 000 000，每名骑士的战斗力都是不大于 1 000 000的正整数。

代码

太难了。。orz wuala

by wulala

这道题题目中貌似是有向边，实际上想一想就知道是无向的。

因为一个骑士觉得另一个骑士丑他们俩就走不到一起。

所以整个图实际上是一个无向环套树森林。

对于每一棵环套树，先dfs找环，找到环以后断环为链并将断开的两个点强制其中一个点为根且不选做一次树形DP，对另一个点做同样操作。

取两次结果最大值加入ans

  1 #include<cstring>
  2 #include<cmath>
  3 #include<iostream>
  4 #include<algorithm>
  5 #include<cstdio>
  6 
  7 #define N 1000007
  8 #define ll long long 
  9 using namespace std;
 10 
 11 int n,cnt;
 12 ll ans,answer;
 13 int v[N],fa[N],head[N],to[N],next[N],ez[N],bh[N];
 14 ll dp[N][2],f[N][4];
 15 bool mark[N];
 16 
 17 void Ins(int a,int b)
 18 {
 19     cnt++;
 20     to[cnt]=b;
 21     next[cnt]=head[a];
 22     head[a]=cnt;
 23 }
 24 void init()
 25 {
 26     int i,k;
 27     scanf("%d",&n);
 28     for(i=1;i<=n;i++)
 29     {scanf("%d%d",&v[i],&k);Ins(k,i);fa[i]=k;}
 30 }
 31 void treedp(int x)
 32 {
 33     dp[x][1]=v[x];dp[x][0]=0;
 34     mark[x]=1;
 35     int i=head[x];
 36     while(i>0)
 37     {
 38         treedp(to[i]);
 39         dp[x][0]+=max(dp[to[i]][0],dp[to[i]][1]);
 40         dp[x][1]+=dp[to[i]][0];
 41         i=next[i];
 42     }
 43 }
 44 void finaldp()
 45 {
 46     int i,k;
 47     ans=0;
 48     f[1][1]=0;f[1][2]=0;
 49     f[1][0]=dp[bh[1]][1];
 50     f[1][3]=dp[bh[1]][0];
 51     for(i=2;i<=bh[0];i++)
 52     {
 53         k=bh[i];
 54         f[i][0]=f[i-1][1]+dp[k][1];
 55         f[i][1]=max(f[i-1][0],f[i-1][1])+dp[k][0];
 56         f[i][2]=f[i-1][3]+dp[k][1];
 57         f[i][3]=max(f[i-1][2],f[i-1][3])+dp[k][0];
 58     }
 59     ans=max(f[bh[0]][1],max(f[bh[0]][2],f[bh[0]][3]));
 60 }
 61 void solve()
 62 {
 63     int i,k,j,now;
 64     for(i=1;i<=n;i++)
 65     {
 66     if(mark[i])continue;
 67     bh[0]=0;
 68     k=i;
 69     while(!mark[k])
 70     {
 71         mark[k]=1;
 72         k=fa[k];
 73         ez[fa[k]]=k;
 74     }
 75     now=k;
 76     while(1)
 77     {
 78         i=head[k];
 79         dp[k][1]=v[k];
 80         while(i>0)
 81         {
 82             if(to[i]!=ez[k])
 83             {treedp(to[i]);
 84             dp[k][0]+=max(dp[to[i]][0],dp[to[i]][1]);
 85             dp[k][1]+=dp[to[i]][0];}
 86              i=next[i];
 87         }
 88         bh[0]++;bh[bh[0]]=k;
 89         k=fa[k];
 90         if(k==now)
 91         break;
 92     }
 93     finaldp();
 94     answer+=ans;
 95     }
 96     printf("%lld",answer);
 97 }
 98 int main()
 99 {
100     init();
101     solve();
102     return 0;
103 }