QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#656270 | #5169. 夹娃娃 | QBF | 0 | 0ms | 0kb | C++14 | 8.8kb | 2024-10-19 12:07:50 | 2024-10-19 12:07:52 |
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));
}
unsigned ll tmp[521];
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=0;i<=min(520,m);++i)tmp[i]=g[i];
for(int i=512;i<=min(520,m);++i)
for(int j=0;i+j<=min(520,m);++j)
tmp[i+j]+=(ll)A[i]*B[j];
for(int i=0;i<=min(520,m);++i)tmp[i]%=mo;
for(int j=512;j<=min(520,m);++j)
for(int i=0;i+j<=min(520,m);++i)
tmp[i+j]+=(ll)A[i]*B[j]%mo;
for(int i=0;i<=min(520,m);++i)C[i]=tmp[i]%mo;
}
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=0;i<=min(520,m);++i)tmp[i]=g[i];
for(int i=512;i<=min(520,m);++i)
for(int j=0;i+j<=min(520,m);++j)
tmp[i+j]+=(ll)A[i]*B[j];
for(int i=0;i<=min(520,m);++i)tmp[i]%=mo;
for(int j=512;j<=min(520,m);++j)
for(int i=0;i+j<=min(520,m);++i)
tmp[i+j]+=(ll)A[i]*B[j];
for(int i=0;i<=min(520,m);++i)C[i]=tmp[i]%mo;
}
}
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(){
freopen("doll.in","r",stdin);
freopen("doll.out","w",stdout);
// int st=clock();
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,520);
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);
}
// printf("%d ms.\n",clock()-st);
return 0;
}
详细
Subtask #1:
score: 0
Dangerous Syscalls
Test #1:
score: 0
Dangerous Syscalls
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:
result:
Subtask #2:
score: 0
Skipped
Dependency #1:
0%
Subtask #3:
score: 0
Skipped
Dependency #2:
0%
Subtask #4:
score: 0
Dangerous Syscalls
Test #9:
score: 0
Dangerous Syscalls
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:
result:
Subtask #5:
score: 0
Skipped
Dependency #3:
0%
Subtask #6:
score: 0
Skipped
Dependency #4:
0%
Subtask #7:
score: 0
Skipped
Dependency #6:
0%