CodeForces-884C

题意抽象：1.每一个点都只有一进一出，也就是他一定自己成环或者和别人一起成环。

                   2.舒适度计算——当x能到达y的时候，舒适度加一。经过计算，在一个环里面，每一个点的舒适度都为环的总点数N的平方，即n*n

                   3.题目通过更改两条路的终点，使舒适度MAX。实际上就是找到最大的两个环，改动两条使两个环变成一个环（因为设环1的点数为N，环2的点数为M，所以增加的舒适度为（N+M）( N+M )-N*N-M*M==2NM，所以环越大，最后增加的越大，所以找最大的两个环）。

所以AC代码：

#include <iostream>#include <algorithm>#include <string>#include <math.h>#include <queue>#include <cstdio>#define pie (long double)acos(-1)#define lld long doubleusing namespace std;int n;const int maxn=100000+10;int a[maxn]={0};int visit[maxn]={0};priority_queue<long long >q;int temp=0;int dfs(int i){    if(visit[i]==1) return temp;    else    {        temp++;        visit[i]=1;        dfs(a[i]);      //  cout<<i<<" "<<temp<<endl;        return temp;    }}int main(){   scanf("%d",&n);   for(int i=1;i<=n;i++)   {       scanf("%d",&a[i]);   }   for(int i=1;i<=n;i++)   {       if(visit[i]==0)       {           temp=0;           q.push(dfs(i));       }   }   if(q.size()>=2)   {        long long x=q.top();        q.pop();       // cout<<x<<endl;        long long y=q.top();        q.pop();       // cout<<y<<endl;         long long ans=(x+y)*(x+y);        while(!q.empty())       {          ans=ans+q.top()*q.top();          q.pop();      }      printf("%I64d\n",ans);   }   else if(q.size()==1)   {       long long x=q.top();       printf("%I64d\n",x*x);   }   return 0;}

错了两次，一个是因为当情况为“只有一个环”的时候，那么第二次q.push()会失败的~，应该要分类讨论

第二次是因为他最后要算的是总的舒适度，我只算了两个环变成一个环的舒适度。