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