QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#67054 | #5169. 夹娃娃 | Dual | 0 | 84ms | 18504kb | C++ | 5.8kb | 2022-12-09 21:17:00 | 2022-12-09 21:17:03 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
struct fastmod {
typedef unsigned long long u64;
typedef __uint128_t u128;
int m;
u64 b;
fastmod(int m) : m(m), b(((u128)1 << 64) / m) {}
int operator ()(u64 a) {
u64 q = ((u128)a * b) >> 64;
int r = a - q * m;
return r < m ? r : r - m;
}
} Red(2);
const int N = 15,M = 520,Lim = 5000;
const int B = 5,C = 3,S = 1 << B; //L : 块数,B : 块长
int n,Q,P;
inline int Add(const int &a,const int &b) { return (a + b >= P) ? (a + b - P) : (a + b);}
inline int Sub(const int &a,const int &b) { return (a < b) ? (a - b + P) : (a - b);}
inline void Plus(int &a,const int &b) { a += b;if(a >= P) a -= P;}
inline void Minus(int &a,const int &b) { a -= b;if(a < 0) a += P;}
inline int qpow(int a,int b) { int res = 1;while(b) {if(b&1) res = 1ll * res * a % P;a = 1ll * a * a % P;b >>= 1;} return res;}
typedef std::vector<int> vint;
vint f[C][S][M + 5]; // 块数,集合,k
vint g[C][S]; // 无限制
int fac[Lim + 5],ifac[Lim + 5],inv[Lim + 5];
int Poly[Lim + 5][M + 5];
void initfac(int n)
{
fac[0] = 1;
for(int i = 1;i <= n;i++) fac[i] = Red(1ll * fac[i - 1] * i);
inv[1] = 1;
for(int i = 2;i <= n;i++) inv[i] = Red(1ll * inv[P % i] * (P - P / i));
ifac[0] = 1;
for(int i = 1;i <= n;i++) ifac[i] = Red(1ll * ifac[i - 1] * inv[i]);
}
// 点值范围:0-Lim
void init_Lagrange(int n) // 系数表示法只需要保留 520 项,但是点值需要保留 2600/1560 项
{
for(int i = 0;i <= n;i++)
for(int j = 0;j <= min(n,M);j++)
Poly[i][j] = 0;
static int f[Lim + 5],tmp[Lim + 5];
initfac(n + 1);
memset(f,0,sizeof f);
f[0] = 1;
for(int i = 1;i <= n;i++)
{
for(int j = min(n,M) + 1;j >= 1;j--)
f[j] = Red(1ll * f[j] * (P - i)),Plus(f[j],f[j - 1]);
f[0] = Red(1ll * f[0] * (P - i));
}
for(int j = 0;j <= min(n,M);j++)
{
Poly[0][j] = Red(1ll * f[j] * ifac[n]);
if(n & 1) Poly[0][j] = P - Poly[0][j];
}
for(int j = min(n,M);j;j--) f[j] = f[j - 1];
f[0] = 0;
for(int i = 1;i <= n;i++)
{
memcpy(tmp,f,sizeof f);
for(int j = 0;j <= min(n,M);j++)
tmp[j] = P - Red(1ll * tmp[j] * inv[i]),Minus(tmp[j + 1],tmp[j]);
int res = Red(1ll * ifac[i] * ifac[n - i]);
if((n - i) & 1) res = P - res;
for(int j = 0;j <= min(n,M);j++)
Poly[i][j] = Red(1ll * res * tmp[j]);
}
}
inline int Solve(int S,int k,int m)
{
vector<int> tmp(M * (C + 1) + 1,1);
for(int p = 0;p <= (n - 1) / B;S >>= B,++p)
{
int now = S & ((1 << B) - 1);
for(int x = 0;x <= M * (C + 1);++x)
tmp[x] = Red(1ll * tmp[x] * f[p][now][k][x]);
}
long long ans = 0;
for(int x = 0;x <= M * (C + 1);++x)
ans += Red(1ll * tmp[x] * Poly[x][m]);
return ans % P;
}
int Cnt[S];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
cin >> n >> Q >> P;
Red = fastmod(P);
for(int i = 0;i < n;i++)
{
int a;
cin >> a;
static int t[M + 5];
memset(t,0,sizeof t);
t[0] = 1;
for(int _ = 1;_ <= a;_++)
{
int b,c;
cin >> b >> c;
for(int j = b;j <= M;j++) Plus(t[j],t[j - b]);
int tmp = b * (c + 1);
if(tmp > M) continue;
for(int j = M;j >= tmp;j--) Minus(t[j],t[j - tmp]);
}
// 没有限制的那个 g
g[i / B][1 << (i % B)].resize(M * B + 1);
for(int x = 0;x <= M * B;x++)
{
int res = 0;
for(int j = M;j >= 0;j--)
res = Red(1ll * res * x + t[j]);
// printf("gres:%d\n",res)
g[i / B][1 << (i % B)][x] = res;
}
for(int k = 0;k <= M;k++) f[i / B][1 << (i % B)][k].resize(M * (C + 1) + 1);
for(int x = 0;x <= M * (C + 1);x++)
{
int powx = 1,res = 0;
for(int k = 0;k <= M;k++)
{
res = Red(res + 1ll * t[k] * powx);
f[i / B][1 << (i % B)][k][x] = P - res; // 容斥系数 -1
powx = Red(1ll * powx * x);
}
}
}
for(int i = 1;i < S;i++) Cnt[i] = Cnt[i >> 1] + (i & 1);
init_Lagrange(M * B); // 到时候要把点值个数由 M * B 变为 M * (C + 1) (由块内变为块间)
for(int p = 0;p <= (n - 1) / B;++p)
{
int siz = min(B,n - p * B);
g[p][0].assign(M * B + 1,1); // 空集的权都赋为 1
for(int k = 0;k <= M;k++) f[p][0][k].assign(M * (C + 1) + 1,1);
for(int S = 1;S < (1 << siz);++S)
if(Cnt[S] > 1)
{
g[p][S].resize(M * B + 1);
for(int x = 0;x <= M * B;++x)
g[p][S][x] = Red(1ll * g[p][S & (S - 1)][x] * g[p][S & (-S)][x]);
for(int k = 0;k <= M && (k + 1) * Cnt[S] <= M;k++)
{
f[p][S][k].resize(M * (C + 1) + 1);
for(int x = 0;x <= M * (C + 1);++x)
f[p][S][k][x] = Red(1ll * f[p][S & (S - 1)][k][x] * f[p][S & (-S)][k][x]);
}
}
// return 0;
// 只保留 M * (C + 1) + 1 个点值
for(int S = 0;S < (1 << siz);++S)
{
vector<int> tmp(M + 1,0);
for(int i = 0;i <= M;i++)
for(int j = 0;j <= M * B;j++)
Plus(tmp[i],Red(1ll * g[p][S][j] * Poly[j][i]));
g[p][S].resize(M * (C + 1) + 1);
for(int x = 0;x <= M * (C + 1);++x)
{
int res = 0;
for(int j = M;j >= 0;j--)
res = Red(1ll * res * x + tmp[j]);
g[p][S][x] = res;
}
}
// return 0;
for(int S = 0;S < (1 << siz);++S)
for(int k = 0;k <= M && (k + 1) * Cnt[S] <= M;k++)
for(int x = 0;x <= M * (C + 1);++x)
f[p][S][k][x] = Red(1ll * f[p][S][k][x] * g[p][((1 << siz) - 1) ^ S][x]);
// return 0;
for(int u = 0;u < siz;++u)
for(int S = 0;S < (1 << siz);++S)
if((S >> u) & 1)
for(int k = 0;k <= M;k++)
for(int x = 0;x <= M * (C + 1);++x)
Plus(f[p][S][k][x],f[p][S ^ (1 << u)][k][x]);
}
init_Lagrange(M * (C + 1));
for(int i = 0;i <= M * (C + 1);i++)
for(int j = 1;j <= M;j++)
Plus(Poly[i][j],Poly[i][j - 1]);
while(Q--)
{
string s;
int k,m;
cin >> s >> m >> k;
--k;
int S = 0;
for(int i = n - 1;i >= 0;i--)
S = S << 1 | (s[i] - '0');
cout << Solve(S,k,m) << endl;
}
return 0;
}
详细
Subtask #1:
score: 0
Runtime Error
Test #1:
score: 3
Accepted
time: 84ms
memory: 18504kb
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
Runtime Error
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:
result:
Subtask #2:
score: 0
Skipped
Dependency #1:
0%
Subtask #3:
score: 0
Skipped
Dependency #2:
0%
Subtask #4:
score: 0
Runtime Error
Test #9:
score: 0
Runtime Error
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%