提交时间:2025-03-30 14:45:46
运行 ID: 37426
#include<bits/stdc++.h> using namespace std; const int N = 1e5+10; int T,n,a[N],f[N][21]; map<int,int> mp; int main(){ cin>>T; while(T--){ cin>>n; for(int i=1;i<=n;i++){ for(int j=1;j<=20;j++){ f[i][j]=0; } } for(int i=1;i<=n;i++){ cin>>a[i]; f[i][0]=a[i]; } for(int i=1;i<=20;i++){ for(int j=1;j<=n;j++){ f[j][i]=f[f[j][i-1]][i-1]; } } for(int i=1;i<=n;i++){ mp[f[i][20]]++; } long long ans=0; for(int i=1;i<=n;i++){ mp[f[i][20]]--; ans=ans+(n-i)-(long long)mp[f[i][20]]; } cout<<ans<<endl; } return 0; }