QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#358956 | #6325. Peaceful Results | 275307894a | WA | 469ms | 207652kb | C++14 | 3.6kb | 2024-03-20 09:51:52 | 2024-03-20 09:51:52 |
Judging History
answer
#include<bits/stdc++.h>
#define Gc() getchar()
#define Me(x,y) memset(x,y,sizeof(x))
#define Mc(x,y) memcpy(x,y,sizeof(x))
#define d(x,y) ((m)*(x-1)+(y))
#define R(n) (rnd()%(n)+1)
#define Pc(x) putchar(x)
#define LB lower_bound
#define UB upper_bound
#define fi first
#define se second
#define eb emplace_back
#define all(x) x.begin(),x.end()
using namespace std;using ll=long long;using db=double;using lb=long db;using ui=unsigned;using ull=unsigned long long;using pii=pair<int,int>;
const int N=1.5e6+5,M=(1<<22)+5,K=1000+5,mod=998244353,Mod=mod-1;const db eps=1e-9;const int INF=1e9+7;mt19937 rnd(263082);
#define Tp template<typename T>
#define Ts template<typename T,typename... Ar>
namespace Debug{
Tp void _debug(char* f,T t){cerr<<f<<'='<<t<<endl;}
Ts void _debug(char* f,T x,Ar... y){while(*f!=',') cerr<<*f++;cerr<<'='<<x<<",";_debug(f+1,y...);}
#ifdef LOCAL
#define gdb(...) _debug((char*)#__VA_ARGS__,__VA_ARGS__)
#else
#define gdb(...) void()
#endif
}using namespace Debug;
int n,Ar,Ap,As,Br,Bp,Bs,Cr,Cp,Cs;
ll frc[N],inv[N],A[M],B[M],C[M];
int k;
ll mpow(ll x,int y=mod-2){ll ans=1;while(y) y&1&&(ans=ans*x%mod),y>>=1,x=x*x%mod;return ans;}
namespace poly{
int tr[M];ll pw[M];const int g=3,Ig=mpow(g);
void init(int n){
k=0;for(k=1;k<=n;k<<=1);
for(int i=0;i<k;i++) tr[i]=(tr[i>>1]>>1)|((i&1)?k/2:0);
ll w=mpow(g,(mod-1)/k);
pw[k/2]=1;for(int i=k/2+1;i<k;i++) pw[i]=pw[i-1]*w%mod;
for(int i=k/2-1;i;i--) pw[i]=pw[i<<1];
}
void NTT(ll *A,int k,int flag){
static ull a[M];
int i,j,h;for(int i=0;i<k;i++) a[tr[i]]=A[i];
for(int i=2;i<=k;i<<=1){
ll *e=pw+i/2;
for(int j=0;j<k;j+=i){
for(int h=j;h<j+i/2;h++) {
ll now=e[h-j]*a[h+i/2]%mod;
a[h+i/2]=a[h]+mod-now;
a[h]+=now;
}
}
if(i==(1<<18)){for(int j=0;j<k;j++) a[j]%=mod;}
}
for(int i=0;i<k;i++) A[i]=a[i]%mod;
if(flag) return;
ll iv=mpow(k);for(int i=0;i<k;i++) A[i]=A[i]*iv%mod;
reverse(A+1,A+k);
}
}
void convolute(ll *A,ll *B,ll *C){
poly::init(2*(n+1));
poly::NTT(A,k,1);poly::NTT(B,k,1);
for(int i=0;i<k;i++) C[i]=A[i]*B[i]%mod;
poly::NTT(C,k,0);
// for(int i=0;i<=n;i++) if(A[i]) for(int j=0;j<=n;j++) if(B[j]) C[i+j]=(C[i+j]+A[i]*B[j])%mod;
}
void Solve(){
int i,j;scanf("%d%d%d%d%d%d%d%d%d%d",&n,&Ar,&Ap,&As,&Br,&Bp,&Bs,&Cr,&Cp,&Cs);
inv[1]=1;for(i=2;i<=n;i++) inv[i]=(mod-inv[mod%i])*(mod/i)%mod;
for(inv[0]=frc[0]=i=1;i<=n;i++) inv[i]=inv[i-1]*inv[i]%mod,frc[i]=frc[i-1]*i%mod;
ll ans=0;
for(int x1=0;x1<=n;x1++){
if(x1>Ar||x1>Br||x1>Cr) continue;
int x4=Ar+Br+Cr-3*x1-Ap-Bp-Cp;
if(x4%3) continue;x4/=-3;
if(x4<0||x4>Ap||x4>Bp||x4>Cp) continue;
int x7=Ar+Br+Cr-3*x1-As-Bs-Cs;
if(x7%3) continue;x7/=-3;
if(x7<0||x7>As||x7>Bs||x7>Cs) continue;
A[x1]=inv[x1]*inv[x4]%mod*inv[x7]%mod;
}
for(int x2=0;x2<=n;x2++){
if(x2>Ar||x2>Bp||x2>Cs) continue;
int x5=Ar+Bp+Cs-3*x2-Ap-Bs-Cr;
if(x5%3) continue;x5/=-3;
if(x5<0||x5>Ap||x5>Bs||x5>Cr) continue;
int x8=Ar+Bp+Cs-3*x2-As-Br-Cp;
if(x8%3) continue;x8/=-3;
if(x8<0||x8>As||x8>Bp||x8>Cr) continue;
B[x2]=inv[x2]*inv[x5]%mod*inv[x8]%mod;
}
convolute(A,B,C);
for(int xs=0;xs<=2*n;xs++){
int x3=Ar-xs;
if(x3<0||x3>Ar||x3>Bs||x3>Cp) continue;
int x6=Ar+Bs+Cp-3*x3-Ap-Br-Cs;
if(x6%3) continue;x6/=-3;
if(x6<0||x6>Ap||x6>Br||x6>Cs) continue;
int x9=Ar+Bs+Cp-3*x3-As-Bp-Cr;
if(x9%3) continue;x9/=-3;
if(x9<0||x9>As||x9>Bp||x9>Cr) continue;
ans+=C[xs]*inv[x3]%mod*inv[x6]%mod*inv[x9]%mod;
}
printf("%lld\n",ans%mod*frc[n]%mod);
}
int main(){
int t=1;
// scanf("%d",&t);
while(t--) Solve();
cerr<<clock()*1.0/CLOCKS_PER_SEC<<'\n';
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 8148kb
input:
2 2 0 0 1 1 0 1 0 1
output:
2
result:
ok 1 number(s): "2"
Test #2:
score: 0
Accepted
time: 1ms
memory: 6060kb
input:
3 0 1 2 3 0 0 1 1 1
output:
0
result:
ok 1 number(s): "0"
Test #3:
score: 0
Accepted
time: 76ms
memory: 56160kb
input:
333333 111111 111111 111111 111111 111111 111111 111111 111111 111111
output:
383902959
result:
ok 1 number(s): "383902959"
Test #4:
score: 0
Accepted
time: 469ms
memory: 207652kb
input:
1500000 500000 500000 500000 500000 500000 500000 500000 500000 500000
output:
355543262
result:
ok 1 number(s): "355543262"
Test #5:
score: 0
Accepted
time: 447ms
memory: 207648kb
input:
1499999 499999 499999 500001 499999 499999 500001 499999 499999 500001
output:
934301164
result:
ok 1 number(s): "934301164"
Test #6:
score: -100
Wrong Answer
time: 463ms
memory: 207640kb
input:
1500000 1 0 1499999 1499999 1 0 0 1499999 1
output:
0
result:
wrong answer 1st numbers differ - expected: '1500000', found: '0'