【想法】codeforces.com/contest/884/problem/C Bertown Subway

早上受到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;
}