2015 北京现场赛 A Math Problem

题意：
给你两个数$n,k$，定义$g(x)$为集合$\{ i|f(i) \% k =x,1 \leqslant i \leqslant n\}$的元素个数，求$g(0) \otimes g(1) \otimes \cdots \otimes g(k-1)$。
方法：
官方题解是数位dp，我用的是二分。

1. 想得到1~2n+1对数组g的贡献，只需先求出1~n对数组g的贡献，这样2~2n+1对数组g的贡献就可由递推式算出，1的贡献单独补上即可。
2. 想得到1~2n对数组g的贡献，重复步骤1后，2n+1的贡献单独删去即可。

$$f(2n)=3*f(n)$$
$$f(2n+1)=3*f(n)+1$$

#include <cstdio>
const int N = 7e4 + 5;

int mod;
long long g[N], ng[N];

long long f(long long n)
{
    if(n == 1) return 1;
    if(n % 2 == 0)
        return 3 * f(n/2) % mod;
    else
        return (3 * f(n/2) + 1) % mod;
}
void dfs(long long n)
{
    if(n == 1)
    {
        for(int i = 0; i < mod; i++)
            g[i] = 0;
        g[1] = 1;
        return ;
    }

    dfs(n / 2);

    for(int i = 0; i < mod; i++)
        ng[i] = 0;
    for(int i = 0; i < mod; i++)
    {
        ng[(3 * i) % mod] += g[i];
        ng[(3 * i + 1) % mod] += g[i];
    }
    ng[1]++;
    if(n % 2 == 0) ng[f(n+1)]--;

    for(int i = 0; i < mod; i++)
        g[i] = ng[i];
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        long long n;
        scanf("%lld%d", &n, &mod);

        dfs(n);

        long long res = 0;
        for(int i = 0; i < mod; i++)
            res ^= g[i];
        printf("%lld\n", res);
    }

    return 0;
}