Atcoder beginner contest 284 G - Only Once题解

题目描述

对于一个长度为n的数组A，其中1 <= ai <= n,对于1<= i <= n,定义一个长度为10^(100)的数组Bi:

---

·B(i,1) = i

·B(i,j+1) = A(B(i,j)) (1 <= j <10^(100)

令Si 表示 Bi 数组中仅出现一次的数的个数

在所有n^(n)种情况的数组A中，求所有Si的和

---

代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int MAXN=2e5+1;
int f[MAXN];
inline int ksm(int a,int b,int m) {
    int ret=1;
    while(b) {
        if(b&1) ret=ret*a%m;
        a=a*a%m;
        b=b>>1;
    }
    return ret;
}
signed main() {
    int n,mod,ans=0;
    scanf("%lld%lld",&n,&mod);
    f[n]=1;
    for(int i=n;i>=1;--i) f[i-1]=f[i]*i%mod;
    for(int k=1;k<=n;++k) ans=(ans+k*(k-1)/2%mod*ksm(n,n-k,mod)%mod*f[n-k]%mod)%mod;
    printf("%lld\n",ans);
}