【数位DP】【P4317】花神的数论题

【数位DP】【P4317】花神的数论题

Description

给定 $n$，求 $n$ 以内所有正整数二进制下 $1$ 的个数的乘积，答案对 $10^7 + 7$ 取模

Limitations

$1 \leq n \leq 10^{15}$

Solution

这是我见过最简单的数位 DP，然而我还是翻车了……

考虑 $1$ 最多有 $O(\log n)$ 个，因此 $1$ 的个数是可以枚举的。

枚举求有 $k$ 个 $1$ 的数的个数，设 $f_{i, j}$ 是考虑前 $i$ 位，剩下的位置还需要填 $j$ 个 $1$ 且不顶上界的方案数。转移显然：

$$f_{i,j} = f_{i-1,j+1} f_{i-1,j}$$

分别对应第 $i$ 位填 $0$ 和填 $1$。

顶上界的情况记一下已经有几个 $1$ 就可以更新了。

因为只考虑 $1$ 的个数，所以前导 $0$ 对答案没有贡献，不需要做特殊处理。

需要注意的是，$10^7 + 7$ 并不是一个质数，$10^7 + 7 ~=~941 \times 10627$，因此在 DP 的时候并不能模 $p - 1$，因为他的欧拉函数并不是减一简单。事实上并不需要取模，因为方案数不会超过 $n$，因此直接计算即可。

Code

#include <cstdio>
#include <cstring>

const int maxn = 64;
const int MOD = 10000007;

ll n;
int cnt, upc = -1;
ll ans = 1;
ll frog[maxn][maxn];
int MU[maxn];

ll calc(const int x);
ll mpow(const int x, ll y);

int main() {
  freopen("1.in", "r", stdin);
  qr(n);
  do cnt += (MU[++upc] = n & 1); while (n >>= 1);
  for (int i = 1; i < 60; ++i) {
    ans = ans * mpow(i, calc(i)) % MOD;
  }
  ans = ans * cnt % MOD;
  qw(ans, '\n', true);
  return 0;
}

ll calc(const int x) {
  int cnt = x - 1;
  memset(frog, 0, sizeof frog);
  frog[upc][x] = 1;
  for (int i = upc - 1; ~i; --i) {
    int di = i + 1;
    for (int j = 0; j <= x; ++j) {
      frog[i][j] = frog[di][j + 1] + frog[di][j];
    }
    if (MU[i] && ~cnt) {
      ++frog[i][cnt--];
    }
  }
  return frog[0][0];
}

ll mpow(const int x, ll y) {
  ll _ret = 1, _tmp = x;
  while (y) {
    if (y & 1) {
      _ret = _ret * _tmp % MOD;
    }
    _tmp = _tmp * _tmp % MOD;
    y >>= 1;
  }
  return _ret;
}

Summary

神tm $10^7+7$ 竟然不是个质数