毫不容易写完了树形DP,竟然发现自己的rank那么落后,后来才得知这是一道贪心可做题。。。GG
贪心做法
对于每个入度为0的点能控制的点,如果它之前没有被支配,那么选它一定是最优的。为什么?…
先更新所有入度为0的点,则剩下的点一定能构成几个简单环。而对于每个有n个点的简单环,最多只能选n/2个点。计算出所有环统计答案。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int ans,n,cnt,m,a[1000001],d[1000001],c[1000001],l=1,r;
bool vis[1000001];
template<class T>inline void read(T &res){
static char ch;T flag=1;
while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
int main(){
read(n);
for(register int i=1;i<=n;i++)read(a[i]),d[a[i]]++;
for(register int i=1;i<=n;i++)if(!d[i])c[++r]=i;
while(l<=r){
if(!vis[c[l]]&&!vis[a[c[l]]]){
ans++;vis[a[c[l]]]=1;
if(--d[a[a[c[l]]]]==0)c[++r]=a[a[c[l]]];
}
vis[c[l++]]=1;
}
for(register int j,i=1;i<=n;i++){
if(!vis[i]){
cnt=0,j=i;
while(a[j]!=i)
vis[j]=1,cnt++,j=a[j];
vis[j]=1;
ans+=(cnt+1)>>1;
}
}
printf("%d",ans);
return 0;
}
树形DP做法
转化为最小支配集。比较麻烦,网上有题解。
就是讨论支配的情况。
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int N=1000010;
int n,m,ans,rt,tot;
int head[N],dp[N],f[N],fa[N],a[N],b[N];
int find(int x){
return fa[x]==x?x:fa[x]=find(fa[x]);
}
struct data{
int nxt,to;
}E[N<<1];
void dfs(int x){
int t=1<<30;
f[x]=0;
for(register int v,i=head[x];i;i=E[i].nxt){
if(v=E[i].to,v!=rt)dfs(v);
f[x]+=max(dp[v],f[v]);
t=min(t,max(dp[v],f[v])-f[v]);
}
dp[x]=f[x]+1-t;
}
void addedge(int x,int y){
E[++tot].nxt=head[x],head[x]=tot,E[tot].to=y;
}
template<class T>inline void read(T &res){
static char ch;T flag=1;
while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
int main(){
read(n);
for(register int i=1;i<=n;i++)fa[i]=i;
for(register int j,i=1;i<=n;i++){
read(j);
if(find(j)!=find(i))
addedge(j,i),fa[find(j)]=find(i);
else a[++m]=j,b[m]=i;
}
for(register int tmp,i=1;i<=m;i++){
dfs(a[i]),rt=a[i];
dfs(b[i]),tmp=dp[b[i]];
dp[a[i]]=f[a[i]]+1;
dfs(b[i]),ans+=max(tmp,f[b[i]]);
}
printf("%d",ans);
return 0;
}