题目大意:有n个玩家,编号1~n,已知狼人不会投狼人,其他人可能投任何人,给你一局投票的情况,判断最多有多少狼人。
比赛时没有什么思路,题解说是基环外向树,第一次听到这个名词感觉十分高端,搜了一下才发现这个东西就是一颗普通的树再加上一条边,那么一定会在这个树里形成一个环,外向树是说其他链的方向指向外,内向树则是指向内。
对于n个玩家,每个人投一票,构成的关系便有n条边,这些点不一定是相互连通的,所以构成的其实是一个基环外向树的森林。
如果这是一个普通的树,我们可以很简单的进行dp,对于基环外向树,我们可以首先找出里面的环,然后任意断开一条边(即选某一个点为树的顶点,忽略fa[x]指向x的这条边),进行dp。
我们知道,当计算dp[x][0]的值时,因为x为平民,所以无论fa[x]是否为狼人,都是可以投它的,所以我们可以忽略这条边。
而计算dp[x][1]时,x为狼人,此时fa[x]有指向x的边,所以此时fa[x]必须为平民,此时再进行一次dp即可求出。
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <set>
using namespace std;
const double eps = 1e-6;
const double pi = acos(-1.0);
const int INF = 0x3f3f3f3f;
#define ll long long
#define CL(a) memset(a,0,sizeof(a))
const int maxn=5e5+10;
int n;
int vis[maxn],mark[maxn];
vector<int> g[maxn];
int dp[maxn][2];
int fa[maxn];
int root;
int dfs(int x)
{
if(vis[x]) return x;
else {
vis[x]=1;
return dfs(fa[x]);
}
}
void Dp1(int x,int pre)
{
vis[x]=true;
dp[x][0]=0;dp[x][1]=1;
// cout<<x<<' '<<fa<<endl;
for(int i=0;i<g[x].size();i++)
{
int v=g[x][i];
if(v==pre||(v==root&&x==fa[root])) continue;
Dp1(v,x);
dp[x][0]+=max(dp[v][0],dp[v][1]);
dp[x][1]+=dp[v][0];
}
}
void Dp2(int x,int pre)
{
vis[x]=true;
dp[x][0]=0;dp[x][1]=1;
for(int i=0;i<g[x].size();i++)
{
int v=g[x][i];
if(v==pre||v==root&&x==fa[root]) continue;
Dp2(v,x);
dp[x][0]+=max(dp[v][0],dp[v][1]);
dp[x][1]+=dp[v][0];
}
if(x==fa[root]) dp[x][1]=0;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%d",&fa[i]);
g[fa[i]].push_back(i);
}
int ans=0;
for(int i=1;i<=n;i++)
{
if(!vis[i])
{
root=dfs(i);
Dp1(root,-1);
int maxw=dp[root][0];
Dp2(root,-1);
maxw=max(maxw,dp[root][1]);
ans+=maxw;
}
}
cout<<ans<<endl;
return 0;
}