早上受到lixinhu同志的教导, 图论的题目要从题给条件出发寻找图的性质, 很有用.
题意: n个点, 保证每个点出入度均为一, 保持这一性质的前提下改变两条边的指向, 使得单连通点对的数目最多.
思路: 刚看懂挺害怕的,单连通? 没见过哦.
然后想图的性质,每个点出入度均为一, 有自环,那就是好多个独立的环组成的图咯? 每个环强连通,贡献环上点数的平方,(a+b)^2>=a^2+b^2, 改两条边刚好把两个最大的环合并.
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
int n;
vector<int> G[100010];
int vis[100010];
LL sz[100010];
int cnt = 0;
int dfs(int u){
int res = 1;
for(int i=0;i<G[u].size();i++){
if(!vis[G[u][i]]) {
vis[G[u][i]] = 1;
res+=dfs(G[u][i]);
}
}
return res;
}
bool cmp(const LL &a,const LL &b){
return a>b;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
int p;scanf("%d",&p);
G[i].push_back(p);
}
for(int i=1;i<=n;i++){
if(!vis[i]){
vis[i] = 1;
sz[cnt++] = dfs(i);
}
}
sort(sz,sz+cnt,cmp);
if(cnt == 1){
cout<<sz[0]*sz[0]<<endl;
return 0;
}
else{
LL ans = (sz[0]+sz[1])*(sz[0]+sz[1]);
for(int i=2;i<cnt;i++){
ans+=sz[i]*sz[i];
}
cout<<ans<<endl;
}
return 0;
}