Educational Codeforces Round 31 C.Bertown Subway（图论）

题目链接：Bertown Subway
题意：简单地说，就是给一个n个地铁站的线路图，每个地铁站i有一趟地铁从i站出发，到达目的站pi，pi可以等于i 且 满足条件： 对于每个i站，只存在一个j站使得pj=i。定义有序对pair(a,b)表示从a站到b站，现在给你一个机会在满足条件下可以改变不超过两个地铁站的pi，使得(a,b)的个数最多，问最多个数是多少？
题解：题目先输入一个n，在输入pi，而且每个pi是不同的，再加上题目要求的条件，这道题只会存在多个环（多个强联通分量），而且每个环之间必然没有有向边。可能一个点成环，也可能多个点成环，那么依据路径把同环的点分组，并统计该组有多少个点，点数从大到小排个序。根据有向图的强联通分量，合并两个强联通分量在每个强联通分量改一条边指向另一个强联通分量即可合成一个强联通分量。要想pair(a,b)的个数最多，自然就是把点数最多的两个强联通分量合并嘛，因为按照定义，每个强联通分量的pair(a,b)个数计算都是该强联通分量的点数点数。所以把大的两个合并后，遍历每个强联通分量，累加所有强联通分量的点数点数即可。这道题我想到强联通分量，开始还想用Tarjan做，但发现根本不用，根据路径用dfs把点分下组就好了。
时间复杂度O(n)

#include <iostream>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set>
#define Max(a,b) a>b?a:b
#define Min(a,b) a>b?b:a
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
int dir[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};
const double eps = 1e-6;
const double Pi = acos(-1.0);
const int INF=0x3f3f3f3f;
const int maxn = 1e5+10;
int n,cnt;
int p[maxn];
struct PointSet{
    int start;
    int step;
    PointSet(int sta = 0, int ste = 0) : start(sta), step(ste) {}
};
PointSet sett[maxn];
bool vis[maxn];

void dfs(int i){
    while(!vis[i]){
        vis[i] = true;
        sett[cnt].step++;
        i = p[i];
    }
}

bool cmp(const PointSet &a, const PointSet &b){
    return a.step > b.step;
}

int main(){
    ll ans = 0;
    scanf("%d",&n);
    for(int i = 1 ; i <= n; i++){
        scanf("%d",&p[i]);
    }
    memset(vis,false,sizeof(vis));
    cnt = 0;

    for(int i = 1; i <= n; i++){
        if(!vis[i]){
            sett[cnt] = PointSet(i,0);
            dfs(i);
            cnt++;
        }
    }

    sort(sett,sett+cnt,cmp);
    if(cnt < 2){
        printf("%lld",(ll)sett[0].step*sett[0].step);
    }else{
        ans += (ll)(sett[0].step+sett[1].step)*(sett[0].step+sett[1].step);
        for(int i = 2; i < cnt; i++){
            ans += (ll)sett[i].step*sett[i].step;
        }
        printf("%lld",ans);
    }



    return 0;
}