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