QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#423270 | #2834. Nonsense | TianTian2008 | TL | 0ms | 0kb | C++14 | 1.5kb | 2024-05-27 21:59:40 | 2024-05-27 21:59:41 |
answer
#include <iostream>
#include <cstdio>
#define mod 998244353
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pr;
ll n,m,x,y,q,fac[10010],inv[10010],px[10010],py[10010];
pr a[200001];
ll ksm(ll x,ll y) {
ll res=1;
while(y) {
if(y&1) res=res*x%mod;
x=x*x%mod;
y>>=1;
}
return res;
}
void init() {
fac[0]=1;
for(int i=1;i<=m;++i) fac[i]=fac[i-1]*(n-i+2)%mod;
ll pro=1;
for(int i=1;i<=m;++i) pro=pro*i%mod;
inv[m]=ksm(pro,mod-2);
for(int i=m;i>=1;--i) inv[i-1]=inv[i]*i%mod;
px[m]=ksm(x,n+1-m);
for(int i=m;i>=1;--i) px[i-1]=px[i]*x%mod;
py[m]=ksm(y,n+1-m);
for(int i=m;i>=1;--i) py[i-1]=py[i]*y%mod;
}
inline ll binom(ll x) {return fac[x]*inv[x]%mod;}
namespace sol1 {
ll f[5010][5010];
void work() {
ll div=ksm(x-y,mod-2);
for(int i=0;i<=m;++i)
for(int j=0;j<=m;++j) {
if(i==0) f[i][j]=-binom(j)*py[j]%mod;
else f[i][j]=-f[i-1][j];
if(j==0) f[i][j]+=binom(i)*px[i]%mod;
else f[i][j]+=f[i][j-1];
f[i][j]=f[i][j]*div%mod;
}
for(int i=1;i<=q;++i) printf("%lld\n",(f[a[i].first][a[i].second]+mod)%mod);
}
}
namespace sol2 {
void work() {
for(int i=1;i<=q;++i)
printf("%lld\n",px[a[i].first+a[i].second+1]*binom(a[i].first+a[i].second+1));
}
}
int main() {
while(scanf("%lld%lld%lld%lld",&n,&x,&y,&q)!=EOF) {
m=0;
for(int i=1;i<=q;++i) {
scanf("%lld%lld",&a[i].first,&a[i].second);
m=max(m,max(a[i].first,a[i].second));
}
m=m<<1|1;
init();
if(x!=y) sol1::work();
else sol2::work();
}
return 0;
}
详细
Test #1:
score: 0
Time Limit Exceeded
input:
3 1 2 2 1 1 1 2 100 2 3 1 1 1