我们把每一个骑士看做一个点,和他厌恶的骑士之间连一条双向边。
那么问题变为求基环树森林上的最大权独立集。
我们先考虑一颗基环树上的情况。
下图是一棵基环树,蓝边为环上的边。
发现每一块连在一起的非环边(红边)构成一棵树,而且明显每一棵树中有且仅有一个环上的点。
那么我们把每棵树中环上的点作为根,设\(f_{i,0/1}\)表示\(i\)及其子树中,点\(i\)选/没选的最大权独立集的权值。
\(f\)的求得就是一个普通的树形动规问题。
于是我们考虑环上的点。
将\(f_{i,1}\)看作选择点\(i\)的权值,\(f_{i,0}\)看作不选点\(i\)的权值。
于是我们要求环上的最大权独立集。
一个简单的环形动规问题。
任选环上一点\(s\)开始dp。
设\(dp_{i,0/1,0/1}\)表示考虑\(s\)到\(i\)的路径上,\(s\)选/不选,点\(i\)选/不选,得到的最大权值。
那么
\(dp_{s,0,0}=f_{s,0}\)
\(dp_{s,1,1}=f_{s,1}\)
\(dp_{s,0,1}=dp_{s,1,0}=-\infty\)
设\(i\)的上一个点为\(j\),那么
\(dp_{i,0,0}=max(dp_{j,0,1},dp_{j,0,0})+f_{i,0}\)
\(dp_{i,1,0}=max(dp_{j,1,1},dp_{j,1,0})+f_{i,0}\)
\(dp_{i,0,1}=dp_{j,0,0}+f_{i,1}\)
\(dp_{i,1,1}=dp_{j,1,0}+f_{i,1}\)
设环的终点为\(t\),则答案=\(max(dp_{t,1,0},dp_{t,0,1},dp_{t,0,0})\)
所有基环树的答案总和就是基环树森林的答案。
code:
#include<bits/stdc++.h>
#define getchar()(p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
using namespace std;
const long long FINF=-1e16;
struct edge{
int t,nxt;
}e[2000010];
int n,w[1000010],u,cnt,be[1000010],d[1000010],tg[1000010],vis[1000010];
long long f[1000010][2],dp[1000010][2][2],ans;
char buf[1<<21],*p1=buf,*p2=buf;
queue<int>q;
void add(int x,int y){
e[++cnt].t=y,e[cnt].nxt=be[x],be[x]=cnt,++d[y];
}
void bfs(){
for(int i=1;i<=n;++i)d[i]==1?q.push(i),tg[i]=1:0;
while(!q.empty()){
u=q.front(),q.pop();
for(int i=be[u];i;i=e[i].nxt)--d[e[i].t]==1?q.push(e[i].t),tg[e[i].t]=1:0;
}
}
void Merge(int x,int y){
f[x][0]+=max(f[y][0],f[y][1]);
f[x][1]+=f[y][0];
}
void dfs(int x){
f[x][1]=w[x];
for(int i=be[x];i;i=e[i].nxt)!f[e[i].t][1]&&tg[e[i].t]?dfs(e[i].t),Merge(x,e[i].t),0:0;
}
void Upd(int x,int y){
dp[x][1][0]=max(dp[y][1][1],dp[y][1][0])+f[x][0];
dp[x][0][0]=max(dp[y][0][1],dp[y][0][0])+f[x][0];
dp[x][1][1]=dp[y][1][0]+f[x][1];
dp[x][0][1]=dp[y][0][0]+f[x][1];
}
void DP(int x){
int tag=1;
vis[x]=1;
for(int i=be[x];i&&tag;i=e[i].nxt)!tg[e[i].t]&&!vis[e[i].t]?Upd(e[i].t,x),DP(e[i].t),tag=0:0;
ans+=tag*max(dp[x][1][0],max(dp[x][0][1],dp[x][0][0]));
}
void scan(int &x){
x=0;
char c=getchar();
while(!isdigit(c))c=getchar();
while(isdigit(c))x=x*10+c-'0',c=getchar();
}
int main(){
scan(n);
for(int i=1;i<=n;++i)scan(w[i]),scan(u),add(i,u),add(u,i);
bfs();
for(int i=1;i<=n;++i)!tg[i]?dfs(i),0:0;
for(int i=1;i<=n;++i)!tg[i]&&!vis[i]?dp[i][1][1]=f[i][1],dp[i][0][0]=f[i][0],dp[i][1][0]=dp[i][0][1]=FINF,DP(i),0:0;
printf("%lld",ans);
return 0;
}