CF884C：贪心

CF884C：

题解：给你一张有向图，修改两条边，使相互到达的点数量最多。

题解：

• p的值都不相同，说明图中只存在环或者点。
• 找出最大的两个环，可以通过修改两条边进行合并。
• 每个环的对数为cnt*cnt
• 利用并查集进行维护。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int const N = 100000 + 10;
int n,fa[N];
ll cnt[N],ans;
int find(int x){
	return x == fa[x] ? x : (fa[x] = find(fa[x]));
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)	fa[i] = i,	cnt[i] = 0;
	for(int i=1;i<=n;i++){
		int tmp;	scanf("%d",&tmp);
		fa[find(i)] = find(tmp);
	}
	for(int i=1;i<=n;i++)
		cnt[find(i)]++;
	sort(cnt+1,cnt+1+n);
	ans += (cnt[n] + cnt[n-1]) * (cnt[n] + cnt[n-1]);
	for(int i=1;i<=n-2;i++){
		ans += cnt[i] * cnt[i];
	}
	cout<<ans<<endl;
	return 0;
}