bzoj 1040（树形DP）

1040: [ZJOI2008]骑士

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 3215   Solved: 1238
[ Submit][ Status][ Discuss]

Description

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

Input

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

Output

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

Sample Input

3
10 2
20 3
30 1

Sample Output

30

HINT

对于100%的测试数据，满足N ≤ 1 000 000，每名骑士的战斗力都是不大于   1 000 000的正整数。

解题思路：这道题有点恶心，分两种情况，多出来的一条边与之前重合，那么

就是树形DP，另一种会构成环，要先拆环，然后再分取和不取各做一边树形

DP

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
int len,zhi[1000005],tail,root,from;
long long ans;
int to[2000005],next[2000005];
int h[1000005];
long long f[2][1000005];
int q[1000005];
bool b[1000005];
int fg[1000005];
bool c[1000005];


inline int read()
 {
  char y; int x=0,f=1; y=getchar();
  while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;
 }
 
void insert(int x,int y)
 {
  ++len; to[len]=y; next[len]=h[x]; h[x]=len;
 } 


void find(int o,int fa)
 {
  fg[o]=fa; ++tail; q[tail]=o; b[o]=true; int u=h[o];
  while (u!=0)
  {
        if (fa==to[u]) {u=next[u]; continue;}
  if (b[to[u]]==true && fg[to[u]]!=o) 
{
  root=o; from=to[u];
}else find(to[u],o);
u=next[u];
}
 }
 
void solve(int o,int fa)
 {
  f[1][o]=zhi[o]; f[0][o]=0; c[o]=true;
  int u=h[o];
  while (u!=0)
  {
    if (c[to[u]]==false)
     {
       solve(to[u],o);
       f[1][o]+=f[0][to[u]];
       f[0][o]+=max(f[0][to[u]],f[1][to[u]]);
}
    u=next[u];
}
  } 
 
void dfs(int o,int fa)
 {
  c[o]=true;
  if (o!=root && o!=from)
  {
  f[0][o]=0; f[1][o]=zhi[o];
}
int u=h[o];
while (u!=0)
     {
      if (to[u]==fa && to[u]==root) {u=next[u]; continue;}
if (o==root && to[u]==from) {u=next[u]; continue;}
if (c[to[u]]==false)
{
dfs(to[u],o);
if (f[0][to[u]]>=0) f[1][o]+=f[0][to[u]];
if (f[1][to[u]]>=0 || f[0][to[u]]>=0) f[0][o]+=max(f[0][to[u]],f[1][to[u]]);
}
u=next[u];
}
 }
 
void solve1(int o)
 {
  f[1][o]=zhi[o]; 
  f[0][from]=0; 
  dfs(o,0); long long u=f[1][o];
  for (int i=1;i<=tail;++i) f[1][q[i]]=-0x7fffffff,f[0][q[i]]=-0x7fffffff,c[q[i]]=false;
  f[0][o]=0; f[0][from]=0; f[1][from]=zhi[from];
  dfs(o,0);
  u=max(u,f[0][o]);
  ans+=u;
 } 
 
int main()
{
n=read();
for (int i=1;i<=n;++i)
{
zhi[i]=read(); int x=read();
insert(i,x); insert(x,i);
}
memset(f,-0x7f,sizeof(f)); 
for (int i=1;i<=n;++i)
     if (b[i]==false)
{
  root=from=tail=0;
  find(i,0);
  if (root==0) solve(i,0),ans+=max(f[0][i],f[1][i]);else
   {
     solve1(root);
}
}
printf("%lld",ans);
}