全排列的价值蓝桥杯python

Problem

【输入样例1】
3
【输出样例1】
9
【输入样例2】
2022
【输出样例2】
593300958

对于样例1，有如下解释：

1 2 3：0+1+2=3
1 3 2：0+1+1=2
2 1 3：0+0+2=2
2 3 1：0+1+0=1
3 1 2：0+0+1=1
3 2 1：0+0+0=0
总和是9
40 分数据 n < 20  ，70 分数据 n < 5000 ，100 分数据 n < 1 0e6 

这个题我用dp的思路想了好长时间，没看懂，突然看到了大佬的解法，很神奇，从每一位期望的角度统计贡献，很厉害，当然大佬用快速幂求逆元确实有些慢了，我来写一下用扩欧求逆元。

大佬的解法：

n=int(input())
mod=998244353
p,q=0,0
def exgcd(a,b):
    global p,q
    if (b==0):
        p,q=1,0
        return a
    ans=exgcd(b,a%b)
    mid=p
    p=q
    q=mid-int(a//b)*q
    return ans

ans=1
for i in range(2,n+1):
    ans*=i
    ans%=mod
exgcd(4,mod)
p=((p%mod)+mod)%mod
ans*=p
ans%=mod
ans*=n
ans%=mod
ans*=(n-1)
ans%=mod
print(ans)

通过情况：

 

 

原文链接：https://blog.csdn.net/qq_45351872/article/details/124478888