然后我们考虑不停分解 x 出来,因为上的数量肯定比下面的多,那我们就把这个情况看到每个 2i 中,这都应该是要满足的,所以它们每一位应该都是一样的: ∀i∈N,⌊n2i⌋=∑i⌊ax2i⌋
那你就考虑这个是什么鬼,那我们会发现这个其实跟二进制有关,就是所有 ax 加在一起不能有进位。
那这个其实我们不难想到一个东西就是把 n 的二进制拆成若干份,每个 x 贡献是 1x!,然后每个的贡献乘起来,最后另外乘上 n!。
然后你这个乘你可以用循环卷积来做,就 k 个字符依次加入然后 FWT 一下。
然后这样还是太慢所以可以 k 这个地方快速幂一下。
代码
#include<cstdio>#define ll long longusingnamespace std;
constint N = 1 << 19;
int n, k, mo, jc[N], inv[N], num1[N];
int f[20][N], g[20][N];
intjia(int x, int y){return x + y >= mo ? x + y - mo : x + y;}
intjian(int x, int y){return x < y ? x - y + mo : x - y;}
intcheng(int x, int y){return1ll * x * y % mo;}
intksm(int x, int y){
int re = 1;
while (y) {
if (y & 1) re = cheng(re, x);
x = cheng(x, x); y >>= 1;
}
return re;
}
voidInit(){
jc[0] = 1; for (int i = 1; i < N; i++) jc[i] = cheng(jc[i - 1], i);
inv[0] = inv[1] = 1; for (int i = 2; i < N; i++) inv[i] = cheng(inv[mo % i], (mo - mo / i));
for (int i = 1; i < N; i++) inv[i] = cheng(inv[i - 1], inv[i]);
for (int i = 1; i < N; i++) num1[i] = num1[i ^ (i & (-i))] + 1;
}
voidFWT(int *f, int limit, int op){
for (int mid = 1; mid < limit; mid <<= 1) {
for (int R = mid << 1, j = 0; j < limit; j += R) {
for (int k = 0; k < mid; k++) {
ll x = f[j | k], y = f[j | mid | k];
f[j | k] = x; f[j | mid | k] = jia(cheng(x, op), y);
}
}
}
}
intwork(int n, int k){
int m = 1; while (m <= n) m <<= 1;
f[0][0] = 1; for (int i = 0; i <= n; i++) g[num1[i]][i] = inv[i];
for (int i = 0; i < num1[n]; i++) FWT(g[i], m, 1);
FWT(f[0], m, 1);
for (int kk = k; kk; kk >>= 1) {
if (kk & 1) {
for (int i = num1[n]; i >= 0; i--) {
for (int j = 0; j < m; j++)
f[i][j] = cheng(f[i][j], g[0][j]);
for (int j = 0; j < i; j++)
for (int k = 0; k < m; k++) {
f[i][k] = jia(f[i][k], cheng(f[j][k], g[i - j][k]));
}
}
}
for (int i = num1[n]; i >= 0; i--) {
for (int j = 0; j < m; j++)
g[i][j] = cheng((i ? 2 : 1), cheng(g[i][j], g[0][j]));
for (int j = 1; j < i; j++)
for (int k = 0; k < m; k++) {
g[i][k] = jia(g[i][k], cheng(g[j][k], g[i - j][k]));
}
}
}
FWT(f[num1[n]], m, mo - 1);
returncheng(f[num1[n]][n], jc[n]);
}
intmain(){
scanf("%d %d %d", &n, &k, &mo);
if (n & 1) {
printf("%d", ksm(k, n)); return0;
}
Init();
printf("%d", jian(ksm(k, n), work(n, k)));
return0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现