这场A写的有点慢,主要原因是写太着急,一堆错误,写完都几千人过了。B思路来的快直接rush了。
C本来想写复杂度很低的算法(精确计算),看了一眼a和b很小,那么直接对lcm取模暴力就好了,看了一眼时限3.5s,然后写了个
a
∗
b
∗
q
∗
t
a*b*q*t
a∗b∗q∗t(2e9)的算法,t了…,后面对取模后的答案记录一下就过了。
D读了半天,读懂后,自然而然想到了二分,稍微想了一下正确性,码了几分钟就过了。其实ans就是所有后缀/ci的最大值,
O
(
n
)
O(n)
O(n)就可以求解。
E
场上分析出来了k<n,以及大致的一些结论,然后主要的组合就开始想假算法了。。。然后50分钟就过去了。。。
看了这个博主的思路:这里
正确思路:先选择n-k列,然后对于每一行可以在这n-k列随便放棋子,这样会出现空列没有棋子,所以需要减去有空列的情况,这部分用容斥考虑,(1列或1列以上有空 - 2列或2列以上有空 + 3列或3列以上有空…)
好像是第二类斯特林数
/* Author : Rshs */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define LDB long double
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const LDB pai = acos(-1.0L);
const LDB eps = 1e-10;
const LL mod = 998244353;
const int MX = 1e6+5;
LL infa[MX],inv[MX],fac[MX],suf[MX];
void CCinit(int n,LL p){
fac[0]=1;
infa[1]=inv[1]=fac[1]=1;
for(LL i=2;i<=n;i++) fac[i]=fac[i-1]*i%p; //阶乘
for(LL i=2;i<=n;i++) inv[i]=(p-p/i)*inv[p%i]%p; //逆元
for(LL i=2;i<=n;i++) infa[i]=infa[i-1]*inv[i]%p; //逆元前缀乘
}
LL CC(int a,int b,LL p){ //c(b/a) 勿忘特判
if(a<0||b<0||a<b) return 0;
if(b==0||a==b) return 1;
return fac[a]*infa[b]%p*infa[a-b]%p;
}
LL quickPow(LL x,LL y,LL MMM){
LL re=1,jin=x%MMM;
while(y){
if(y%2==1) re=re*jin%MMM;
y=y>>1;jin=jin*jin%MMM;
}
return re;
}
int main(){
CCinit(2e5,998244353);
// cout<<CC(4,2,mod);
LL n,k;cin>>n>>k;
if(k>=n) return puts("0"),0;
LL ans=CC(n,n-k,mod); //选 n-k列
LL a2=quickPow(n-k,n,mod); //n行个数随便放n-k列
LL kong=0;//存在空行
for(LL i=1,z=1;i<=n-k;i++,z=-z){ //容斥空行情况
kong=(kong+z*CC(n-k,i,mod)*quickPow(n-k-i,n,mod)%mod)%mod;
}
ans=ans*(a2-kong)%mod;
if(k!=0) ans=ans*2LL%mod;
cout<<(ans%mod+mod)%mod<<'\n';
return 0;
}