QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#656245 | #5169. 夹娃娃 | QBF | 3 | 2771ms | 492268kb | C++14 | 8.5kb | 2024-10-19 11:57:50 | 2024-10-19 11:57:50 |
Judging History
answer
#include<bits/stdc++.h>
#define ci const int
#define ll long long
#define fo(i,l,r) for(int i=(l);i<=(r);++i)
#define fd(i,l,r) for(int i=(l);i>=(r);--i)
#define fu(i,l,r) for(int i=(l);i<(r);++i)
#define ms(x) memset(x,0,sizeof(x))
using namespace std;
char buf[1<<20],*p1=buf,*p2=buf;
#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<20,stdin),p1==p2)?EOF:*p1++)
int read(){int res(0);char ch(getchar());while(ch<48||ch>57)ch=getchar();while(ch>=48&&ch<=57)res=(res<<1)+(res<<3)+(ch^48),ch=getchar();return res;}
void out(ci x){
if(x>9)out(x/10);
putchar(x%10+48);
}
// namespace NTT{
// ci N=525,mod=998244353,g=3,ig=(mod+1)/3;
// inline void add(int &x,ci v){
// x+=v,x-=x<mod?0:mod;
// }
// inline void sub(int &x,ci v){
// x-=v,x+=x<0?mod:0;
// }
// ll qk(ll x,int y=mod-2){
// ll ans=1;
// while(y){
// if(y&1)ans=ans*x%mod;
// x=x*x%mod,y>>=1;
// }
// return ans;
// }
// int R[N<<2];
// void NTT(int *a,ci n,ci tp){
// for(int i=0;i<n;++i)R[i]=(R[i>>1]>>1)|(i&1?(n>>1):0);
// for(int i=0;i<n;++i)if(i<R[i])swap(a[i],a[R[i]]);
// for(int len=2;len<=n;len<<=1){
// ci gn=qk(tp==1?g:ig,(mod-1)/len);
// for(int l=0;l<n;l+=len)
// for(int k=l,al,ar,cur=1;k<l+(len>>1);++k)
// al=a[k],ar=a[k+(len>>1)],
// a[k]=(al+(ll)ar*cur)%mod,
// a[k+(len>>1)]=(al-(ll)ar*cur%mod+mod)%mod,
// cur=(ll)cur*gn%mod;
// }
// if(tp==-1){
// ci inv=qk(n);
// for(int i=0;i<n;++i)a[i]=(ll)a[i]*inv%mod;
// }
// }
// int a[N<<2],b[N<<2],c[N];
// void Mul(int *A,int *B,int *C,ci n){
// int lim=1;
// while(lim<=min(n,510)*2)lim<<=1;
// for(int i=0;i<lim;++i)a[i]=b[i]=0;
// for(int i=0;i<=min(n,510);++i)a[i]=A[i],b[i]=B[i];
// NTT(a,lim,1),NTT(b,lim,1);
// for(int i=0;i<lim;++i)a[i]=(ll)a[i]*b[i]%mod;
// NTT(a,lim,-1);
// for(int i=0;i<=n;++i)c[i]=a[i];
// for(int i=511;i<=n;++i)
// for(int j=0;i+j<=n;++j)
// add(c[i+j],(ll)A[i]*B[j]%mod);
// for(int j=511;j<=n;++j)
// for(int i=0;i+j<=n;++i)
// add(c[i+j],(ll)A[i]*B[j]%mod);
// for(int i=0;i<=n;++i)C[i]=c[i];
// }
// }
namespace NTT{
const int N=1024,mo=998244353;
inline void add(int &x,ci v){
x+=v,x-=x<mo?0:mo;
}
int n,m,a[N],b[N],G[N],Ginv[N],rev[N];
ll ksm(ll x,int y){ll s=1;for(;y;y>>=1,x=x*x%mo)if(y&1)s=s*x%mo;return s;}
void init(int n)
{
for(int i=1;i<n;i<<=1)
{
int g0=ksm(3,(mo-1)/(i<<1)),g0i=ksm(g0,mo-2),g=1,gi=1;
fu(j,i,i+i) G[j]=g,Ginv[j]=gi,g=1ull*g*g0%mo,gi=1ull*gi*g0i%mo;
}
}
void ntt(int n,int *a,int pd)
{
fu(i,0,n) if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int i=1,x;i<n;i<<=1)
for(int *j=a;j<a+n;j+=(i<<1))
{
for(int *k=j,*buf=(pd?G:Ginv)+i;k<j+i;++k,++buf)
{
x=1ull*k[i]*(*buf)%mo;
if((k[i]=*k+mo-x)>=mo) k[i]-=mo;
if((*k+=x)>=mo) *k-=mo;
}
}
if(!pd)
{
int invn=ksm(n,mo-2);
fu(i,0,n) a[i]=1ull*a[i]*invn%mo;
}
}
int f[N],g[N];
void Init(){
int n=1,s=0;while(n<1024) n<<=1,s++;
init(n);
fu(i,0,n) rev[i]=(rev[i>>1]>>1)|((i&1)<<(s-1));
}
void Get(int *aa,int *bb,int *A,int *B,int *C,int m=520){
ci n=1024,s=10;
fu(i,0,n) g[i]=1ull*aa[i]*bb[i]%mo;
ntt(n,g,0);
for(int i=512;i<=min(520,m);++i)
for(int j=0;i+j<=min(520,m);++j)
add(g[i+j],(ll)A[i]*B[j]%mo);
for(int j=512;j<=min(520,m);++j)
for(int i=0;i+j<=min(520,m);++i)
add(g[i+j],(ll)A[i]*B[j]%mo);
for(int i=0;i<=min(520,m);++i)C[i]=g[i];
}
void Mul(int *A,int *B,int *C,int m=520)
{
// int n=1,s=0;
// while(n<1024) n<<=1,s++;
ci n=1024,s=10;
for(int i=0;i<n;++i)f[i]=g[i]=0;
for(int i=0;i<=511;++i)f[i]=A[i],g[i]=B[i];
ntt(n,f,1);ntt(n,g,1);
fu(i,0,n) g[i]=1ull*g[i]*f[i]%mo;
ntt(n,g,0);
for(int i=512;i<=min(520,m);++i)
for(int j=0;i+j<=min(520,m);++j)
add(g[i+j],(ll)A[i]*B[j]%mo);
for(int j=512;j<=min(520,m);++j)
for(int i=0;i+j<=min(520,m);++i)
add(g[i+j],(ll)A[i]*B[j]%mo);
for(int i=0;i<=min(520,m);++i)C[i]=g[i];
}
}
ci N=525,V=520,H=1.5e5;
int n,q;
ci mod=998244353;
int pre[N][N],tmp[N],sum[N];
vector<int>vec[N];
inline void add(int &x,ci v){
x+=v,x-=x<mod?0:mod;
}
#define ull unsigned long long
ull calc(vector<int>&vec){
ull ret=0;
for(int x:vec)ret=ret*10007+x;
return ret;
}
struct DAT{
// map<vector<int>,int>mp;
unordered_map<ull,int>mp;
int n;
int a[N],tot,dp[H][N];
int cnt;
int pre[N][N];
int z[1<<8][N];
int con[1<<8][1024];
void check(ci k){
int sum=0;
for(int i=1;i<=n;++i)sum+=a[i];
if(sum>V)return;
vector<int>vec(n);
for(int i=1;i<=n;++i)vec[i-1]=a[i];
const ull v1=calc(vec);
if(mp.find(v1)!=mp.end())return;
mp[v1]=++tot;
int pos=0;
for(int i=1;i<=n;++i)
if(a[i]==k){
pos=i;
break;
}
// assert(pos!=0);
vec[pos-1]=k+1;
const ull v2=calc(vec);
if(mp.find(v2)!=mp.end()){
ci cur=mp[v2];
for(int i=0;i<=V;++i)add(dp[tot][i],dp[cur][i]);
}
vec[pos-1]=0;
const ull v3=calc(vec);
if(mp.find(v3)!=mp.end()){
ci cur=mp[v3];
for(int i=k;i<=V;++i)add(dp[tot][i],(ll)dp[cur][i-k]*pre[pos][k]%mod);
}
// bool ok=1;for(int i=1;i<=n;++i)ok&=a[i]!=k+1;
// if(ok){
// mp2[vec]=++cnt;
// }
}
void dfs(ci d,ci k,ci tp){
if(d==n+1){
check(k);
return;
}
a[d]=0,dfs(d+1,k,tp);
if(k!=0){
if(tp==0){
a[d]=k+1,dfs(d+1,k,tp);
a[d]=k,dfs(d+1,k,1);
}
else{
a[d]=k,dfs(d+1,k,1);
}
}
else{
for(int k=V;k;--k)
a[d]=k+1,dfs(d+1,k,0),
a[d]=k,dfs(d+1,k,1);
}
}
void Solve(ci _n){
n=_n;
vector<int>vec;
for(int i=1;i<=n;++i)vec.push_back(0);
mp[calc(vec)]=++tot;
dp[1][0]=1;
dfs(1,0,0);
z[0][0]=1;
for(int s=1;s<1<<n;++s){
ci x=__builtin_ctz(s&-s)+1;
NTT::Mul(z[s^(s&-s)],pre[x],z[s]);
// for(int i=0;i<=V;++i)
// for(int j=0;i+j<=V;++j)
// add(z[s][i+j],(ll)z[s^(s&-s)][i]*pre[x][j]%mod);
}
for(int s=0;s<1<<n;++s){
for(int i=0;i<=511;++i)con[s][i]=z[s][i];
NTT::ntt(1024,con[s],1);
}
}
}L,R;
int f[N],trs[N],g[N];
int o[1<<15][521];bool vo[1<<15];
int main(){
NTT::Init();
n=read(),q=read();
read();
for(int t=1;t<=n;++t){
pre[t][0]=1;
for(int i=1;i<=V;++i)vec[i].clear();
for(int k=read();k;--k){
ci b=read(),c=read();
vec[b].push_back(c);
}
for(int i=1;i<=V;++i){
for(int j=0;j<=V;++j)tmp[j]=0;
tmp[0]=1;
for(int x:vec[i]){
for(int k=0;k<=V;++k)sum[k]=((k?sum[k-1]:0)+tmp[k])%mod;
for(int k=0;k<=V;++k)tmp[k]=(sum[k]-((k-x-1>=0)?sum[k-x-1]:0)+mod)%mod;
}
for(int j=V;~j;--j)
for(int k=1;j+k*i<=V;++k)
add(pre[t][j+k*i],(ll)pre[t][j]*tmp[k]%mod);
}
}
ci nl=n>>1,nr=n-nl;
for(int i=1;i<=nl;++i)for(int j=0;j<=V;++j)L.pre[i][j]=pre[i][j];
for(int i=1;i<=nr;++i)for(int j=0;j<=V;++j)R.pre[i][j]=pre[i+nl][j];
L.Solve(nl),R.Solve(nr);
while(q--){
int st=0,st0=0;
for(int i=1;i<=n;++i){
char c=getchar();
while(c!='0'&&c!='1')c=getchar();
if(c=='1')st|=1<<i-1;
else st0|=1<<i-1;
}
ci m=read(),k=read();
if(vo[st0]){
for(int i=0;i<=520;++i)f[i]=o[st0][i];
}
else{
NTT::Get(L.con[st0&((1<<nl)-1)],R.con[st0>>nl],L.z[st0&((1<<nl)-1)],R.z[st0>>nl],f,m);
vo[st0]=1;
for(int i=0;i<=520;++i)o[st0][i]=f[i];
}
// for(int i=0;i<=m;++i)f[i]=L.z[st0&((1<<nl)-1)][i];
// NTT::Mul(f,R.z[st0>>nl],f);
vector<int>vec;
for(int i=1;i<=nl;++i)
if(st>>(i-1)&1)vec.push_back(k);
else vec.push_back(0);
const ull vl=calc(vec);
if(L.mp.find(vl)==L.mp.end()){
puts("0");
continue;
}
else{
ci cur=L.mp[vl];
NTT::Mul(f,L.dp[cur],f,m);
// for(int i=0;i<=m;++i)g[i]=0;
// for(int i=0;i<=m;++i)trs[i]=L.dp[cur][i];
// for(int i=0;i<=m;++i)
// for(int j=0;i+j<=m;++j)
// add(g[i+j],(ll)f[i]*trs[j]%mod);
// for(int i=0;i<=m;++i)f[i]=g[i];
}
vec.clear();
for(int i=1;i<=nr;++i)
if(st>>(i+nl-1)&1)vec.push_back(k);
else vec.push_back(0);
const ull vr=calc(vec);
if(R.mp.find(vr)==R.mp.end()){
puts("0");
continue;
}
else{
ci cur=R.mp[vr];
// NTT::Mul(f,R.dp[cur],f,m);
for(int i=1;i<=m;++i)add(f[i],f[i-1]);
int ans=0;
for(int i=0;i<=m;++i)add(ans,(ll)R.dp[cur][i]*f[m-i]%mod);
out(ans),putchar(10);
// for(int i=0;i<=m;++i)g[i]=0;
// for(int i=0;i<=m;++i)trs[i]=R.dp[cur][i];
// for(int i=0;i<=m;++i)
// for(int j=0;i+j<=m;++j)
// add(g[i+j],(ll)f[i]*trs[j]%mod);
// for(int i=0;i<=m;++i)f[i]=g[i];
}
// int ans=0;
// for(int i=0;i<=m;++i)add(ans,f[i]);
// printf("%d\n",ans);
} return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Subtask #1:
score: 3
Accepted
Test #1:
score: 3
Accepted
time: 20ms
memory: 12612kb
input:
1 521 998244353 39 520 520 11 22 414 8 95 18 229 356 26 407 316 10 24 26 19 61 11 130 482 476 420 15 192 193 208 24 19 233 494 217 275 294 26 28 439 20 272 277 28 198 5 335 22 8 28 17 154 78 6 13 175 17 2 5 477 256 200 4 1 36 427 371 439 23 10 65 426 25 24 27 121 29 28 13 12 453 0 520 1 1 519 1 1 51...
output:
38813347 922143638 98254957 38813343 922143633 38813338 98254946 922143620 98254933 922143604 38813302 38813288 922143562 38813247 38813220 38813188 38813150 98254715 38813047 922143273 98254516 38812814 922142999 98254191 922142723 38812257 38812058 98253436 922141847 38811240 922141173 38810463 38...
result:
ok 521 lines
Test #2:
score: 3
Accepted
time: 67ms
memory: 16428kb
input:
2 1561 998244353 151 520 520 511 30 121 396 25 16 113 11 6 175 242 20 8 5 61 13 518 447 404 8 220 177 4 19 18 15 70 233 9 14 26 512 17 9 9 19 30 8 495 20 13 27 277 22 396 14 4 29 345 442 19 25 14 5 16 295 19 65 134 10 10 296 245 6 7 30 253 15 187 26 482 454 28 414 170 404 11 27 27 25 13 509 1 5 291 ...
output:
883965618 144348435 762074635 112296779 385763651 821718611 673974966 879750066 927942969 136450507 436584627 612945970 768262217 613885343 39304132 852224740 215596261 151746110 965953558 969833936 664053020 458247365 881060255 878484499 781573019 616944059 850325449 296113117 674829177 887392623 6...
result:
ok 1561 lines
Subtask #2:
score: 0
Wrong Answer
Dependency #1:
100%
Accepted
Test #3:
score: 0
Wrong Answer
time: 161ms
memory: 18076kb
input:
3 4160 998244353 444 520 520 26 332 29 183 25 479 175 14 13 16 1 447 2 293 4 20 64 472 491 11 21 259 75 22 390 401 8 508 405 3 137 4 15 154 164 1 484 13 257 14 44 20 7 13 26 15 26 432 14 9 478 24 18 10 22 28 8 21 260 25 431 22 7 6 20 26 8 27 239 19 1 134 2 322 16 225 6 42 517 6 197 407 268 500 433 5...
output:
516056999 990096150 497048298 345860798 899328070 577475723 191997503 533625761 516056999 863614705 652318084 514747110 811600228 92531482 136793394 218097588 516056999 821305819 739754364 569418540 402235631 844207347 78271439 896568337 516056999 243958673 201200148 634787992 552693501 893938722 98...
result:
wrong answer 17th lines differ - expected: '352553395', found: '516056999'
Subtask #3:
score: 0
Skipped
Dependency #2:
0%
Subtask #4:
score: 0
Wrong Answer
Test #9:
score: 10
Accepted
time: 2747ms
memory: 492268kb
input:
15 52099 998244353 1 9 3 1 9 4 1 9 2 1 8 10 1 4 4 1 3 1 1 2 5 1 4 9 1 1 4 1 9 4 1 7 6 1 1 6 1 2 5 1 5 2 1 3 5 101000000001010 516 1 010001001010101 520 2 000000101000001 519 2 101011111100011 518 1 010110001000111 520 2 000110111100111 516 1 000100101001011 519 3 000111001010011 518 1 00001110010111...
output:
993379058 496689529 866368587 797687294 481245176 481245176 39022588 269889529 552778235 769822588 331666941 99789529 903956470 112750588 756797435 519045176 870912000 361582588 594280447 494747647 597778941 178845176 435456000 493445999 461733882 308912117 271186941 496689529 919511294 85533882 894...
result:
ok 52099 lines
Test #10:
score: 0
Wrong Answer
time: 2771ms
memory: 491688kb
input:
15 52099 998244353 1 444 1 1 184 2 1 427 1 1 23 1 1 513 2 1 413 4 1 304 2 1 214 5 1 108 2 1 304 1 1 283 5 1 465 1 1 277 1 1 57 2 1 160 4 111010111100101 520 25 010010011100110 520 3 010000011011000 519 36 110011011100110 516 21 010110001101101 516 42 100001101000111 520 31 110001100010001 519 10 110...
output:
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ...
result:
wrong answer 39738th lines differ - expected: '204', found: '201'
Subtask #5:
score: 0
Skipped
Dependency #3:
0%
Subtask #6:
score: 0
Skipped
Dependency #4:
0%
Subtask #7:
score: 0
Skipped
Dependency #6:
0%