前言
我咋连表都没打啊。
too vegetable。
解析
题目给出的洗牌形式看着并不好看,合理猜测可以发现,这其实就等价于所有可能情况等概率出现。
然后就不会了
打表可以发现:当 tp=1 时,dp 数组是一个等差数列。当 tp=2 时,dp 数组的差分是一个等差数列。
进一步猜测:如果原来的dp数组是 i 次多项式,那么洗牌后还是 iii 次多项式。
然而我并不会证
有了这个结论后,不断维护这个不超过二次的多项式的系数就做完了。
代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define ok debug("line: %d\n",__LINE__)inline ll read(){ll x(0),f(1);char c=getchar();while(!isdigit(c)) {if(c=='-')f=-1;c=getchar();}while(isdigit(c)) {x=(x<<1)+(x<<3)+c-'0';c=getchar();}return x*f;
}
bool mem1;const int N=1e7+100;
const int inf=1e9+100;
const int mod=998244353;
const bool Flag=0;#define add(x,y) ((((x)+=(y))>=mod)&&((x)-=mod))
inline ll ksm(ll x,ll k){ll res(1);while(k){if(k&1) res=res*x%mod;x=x*x%mod;k>>=1;}return res;
}int n,m;ll a,b,c;
inline ll calc(ll x){return (x*x%mod*a+b*x+c)%mod;}bool mem2;
signed main(){
#ifndef ONLINE_JUDGEfreopen("a.in","r",stdin);freopen("a.out","w",stdout);
#endifdebug("mem=%lf\n",abs(&mem2-&mem1)/1024./1024);n=read();m=read();int tp=read();//n=1e7;m=5e5;int tp=2;a=tp==2;b=tp==1;c=0;ll niv1=ksm((1ll*n*n-1-3*(n-1))%mod,mod-2);ll nin=ksm(n,mod-2),nin1=ksm(n-1,mod-2);for(int i=1;i<=m;i++){int A=read();//int A=rand()%n+1;ll f1=calc(1),f2=calc(2),fa=calc(A),fa1=calc(A+1),fa2=calc(A+2),fn=calc(n);ll x=(A*f1+(n-A)%mod*fa1)%mod*nin%mod;ll y=(A*((A-1)*f2%mod+(n-A)*fa1%mod)+(n-A)*(A*f1%mod+(n-A-1)*fa2%mod))%mod*nin%mod*nin1%mod;ll z=(A*fa+(n-A)*fn)%mod*nin%mod;//printf("x=%lld y=%lld z=%lld\n",x,y,z);a=(z-x-(n-1)*(y+mod-x)%mod+mod+mod)%mod*niv1%mod;b=(y-x-3*a%mod+mod+mod)%mod;c=(x-a-b+mod+mod)%mod;}int ask=read();while(ask--){printf("%lld\n",calc(read()));}return 0;
}