题目描述
有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;
}