codeforce 884C - Bertown Subway DFS+图论

分析

题意说输入数据有两个规则：
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…