BZOJ1040 || 洛谷P2607 [ZJOI2008]骑士【基环树+树形DP】

Time Limit: 10 Sec
Memory Limit: 162 MB

Description

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

Input

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

Output

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

HINT

N ≤ 1 000 000，每名骑士的战斗力都是不大于 1 000 000的正整数。

---

题目分析

算是基环树的入门题吧
由题意可知，若将每对有矛盾的骑士以无向边相连
那么得到的森林中每个联通快会至多有一个环

假如不考虑这个环，其实就是个简单的树上最大独立集，所以考虑转化成这种情况即可
对每个联通快dfs找到环并断开其中任意一条边
假设这条边连接的点为$u,v$，那么分别强制$u$和$v$不选各做一次DP取较大值

最后累加每个联通快的答案即可

---

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
typedef double dd;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int maxn=2000010;
int n;
struct node{int v,nxt;}E[maxn<<1];
int head[maxn],tot=1;
int vis[maxn];
lt wi[maxn],dp[maxn][2],ans;
int tx,ty,rem;

void add(int u,int v)
{
    E[++tot].nxt=head[u];
    E[tot].v=v;
    head[u]=tot;
}

void dfs(int u,int pa)
{
    vis[u]=1;
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(v==pa) continue;
        if(vis[v]){ 
            tx=u; ty=v; rem=i;
            continue;
            }
        dfs(v,u);
    }
}

void DP(int u,int pa)
{
    dp[u][0]=0; dp[u][1]=wi[u];
    for(int i=head[u];i;i=E[i].nxt)
    {
        int v=E[i].v;
        if(v==pa) continue;
        if(i==rem||(i^1)==rem) continue;
        DP(v,u);
        dp[u][0]+=max(dp[v][0],dp[v][1]);
        dp[u][1]+=dp[v][0];
    }
}

int main()
{
    int n=read();
    for(int i=1;i<=n;++i)
    {
        wi[i]=read(); int x=read();
        add(i,x); add(x,i);
    }
    
    for(int i=1;i<=n;++i)
    {
        if(vis[i]) continue;
        dfs(i,0);
        
        DP(tx,0);
        lt tt=dp[tx][0];
        
        DP(ty,0);
        ans+=max(tt,dp[ty][0]);
    }
    
    printf("%lld",ans);
    return 0;
}