codeforce 884C

分析

题意说输入数据有两个规则：
1 每个i只有一个pi表示i站的目的地 i可以等于pi
2 每个pi都指向唯一的一个站i
那么也就是说i指向的目的只有一个且唯一 每个目的由唯一的始发站出来 所以也就是说 图中的每一个点只有一条单向边 那么当我们求最大的数量的可达对时 也就是尽可能构造大圈 所以每一个点 要么形成自回路 要么在一个多点圈中 而连接任意两个圈的花费就是修改两站的目的地 所以 把两个圈构造成一个大圈的花费就是2
所以这道题可以任选两个最大的圈 连成一个 那么最多的点对 就是这个圈中的点数的平方

题意归纳出来就是尽可能构造大的圈，注意数据有可能会超int

code

#include<bits/stdc++.h>using namespace std;typedef long long ll; const int maxn = 100010;struct node{    int id,val;    node(int a,int b):id(a),val(b){}};bool cmp(node a,node b){    return a.val>b.val;}vector<node>ans;int k,nex[maxn];bool bok[maxn];void dfs(int s){    if(!bok[s]){        bok[s]=1;        k++;        dfs(nex[s]);    }}int main(){    int n;    scanf("%d",&n);    for(int i=1;i<=n;i++){        scanf("%d",&nex[i]);    }    for(int i=1;i<=n;i++){        if(!bok[i]){            k=0;            dfs(i);            ans.push_back(node(i,k));        }    }    sort(ans.begin(),ans.end(),cmp);    ll sum=0,add=0;    bool f=0;    if(ans.size()>=2){        for(int i=0;i<ans.size();i++){            if(i<2){                add+=ans[i].val;                }            else {                sum+=1LL*ans[i].val*ans[i].val;//防止溢出            }        }           sum+=1LL*add*add;//防止溢出    }    else{        sum = 1LL*n*n;    }    printf("%lld\n",sum);    return 0;}

WA在爆int…