[树] cf 698B D Fix a Tree

题意

题目分析

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<map>
#include<algorithm>
using namespace std;
const int maxn=2e5+5;
int a[maxn];
int root=0;
int dep[maxn]={0};
int cnt=0;
int ans;
void work(int x)
{
dep[x]=++cnt;
while(dep[a[x]]==0)
x=a[x],dep[x]=cnt;
if(dep[a[x]]==cnt)//祖先是当前批次，则有可能是环、不连通
{
if(root==0) root=x;
if(a[x]!=root)//祖先是环或不连通
a[x]=root,ans++;
}
}
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
if(a[i]==i) root=i;
for(int i=1;i<=n;i++)
if(dep[i]==0) work(i);
cout<<ans<<endl;
for(int i=1;i<n;i++) printf("%d ",a[i]);cout<<a
<<endl;
return 0;
}