[基环外向树+树形DP]BZOJ 1040—— [ZJOI2008]骑士

题目描述

有n个骑士,每个骑士有一个不想一起组队的人。

每个骑士还有一个战斗力,求战斗力最大的队伍。

解题思路

把每个骑士和不想组队的骑士之间连边,会形成一堆基环外向树形成的森林。

假设每个联通块是树,边数=n-1,显然不符合。

假设每个联通块存在两个及以上的简单环,边数>n,也不符合。

所以图必然是基环外向树形成的森林。

先随意建树寻找返祖边,确定两个点和一条边。

分别以这两个点为树根并砍掉这条边进行树形DP, f[i][0/1] 分别表示这个节点选或不选的最大战斗力,转移状态显然。

这样做的好处是,DP最后的根节点都不选,累计两次的最大值好了。

#include<cstdio>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=1000005;
int lnk[maxn],son[2*maxn],nxt[2*maxn],tot;
int a[maxn],n,hx,hy,hc;
LL ans,f[maxn][2];
bool vis[maxn];
inline int _read(){
    int num=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void DFS(int x,int fa){
    vis[x]=1;
    for (int j=lnk[x];j;j=nxt[j]) if (!vis[son[j]]) DFS(son[j],x);
    else if (son[j]!=fa) hx=x,hy=son[j],hc=(j+1)/2;
}
void DP(int x,int fa){
    f[x][0]=0,f[x][1]=a[x];
    for (int j=lnk[x];j;j=nxt[j])
    if (son[j]!=fa&&((j+1)/2)!=hc){
        DP(son[j],x);
        f[x][0]+=max(f[son[j]][1],f[son[j]][0]);
        f[x][1]+=f[son[j]][0];
    }
}
LL work(int x){DP(x,0);return f[x][0];}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();
    for (int i=1;i<=n;i++){
        a[i]=_read();int x=_read();
        add(x,i);add(i,x);
    }
    for (int i=1;i<=n;i++)if (!vis[i]){
        DFS(i,0);
        ans+=max(work(hx),work(hy));
    }
    printf("%lld\n",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值