hihocoder 1259 推公式+数位dp（非dfs写法）

题意：

题目链接：http://hihocoder.com/problemset/problem/1259?sid=1199888
给出f(x)的定义，给出n和k，求f(1)~f(n)中模k==（0~k-1)的个数的异或和。

---

思路：

首先推公式，可以得到f(2n)=f(n)*3,f(2n+1)=f(2n)+1；
根据公式，写几个例子可以发现，对于一个数，比如5，二进制形式(101)，可以发现f(5)=3^2+3^0=10，也就是f(n)就是n二进制形式计算三进制的结果。
那么剩下就是数位dp，本来应该是很简单的统计模数的dp，但是这题要求算出mod k = (0~k-1)的所有结果，dfs对于每个k正着推那种显然会T，（dfs其实也能写，只是我不会），这题直接for循环反着推会比较好写。dp理出无限制的情况，再对于上界进行dp。

---

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

LL n;
int m;
int a[70];
LL dp[70][65555], three[100], cnt[65555];
LL res;

void solve() {
    three[0] = 1;
    for (int i = 1; i <= 70; i++) {
        three[i] = three[i - 1] * 3 % m;
    }
    int pos = 0;
    while (n) {
        a[pos++] = n % 2;
        n /= 2;
    }
    memset(dp, 0, sizeof(dp));
    dp[0][0] = 1;
    for (int i = 0; i < pos - 1; i++) { // 计算出无限制条件下数位长度为i且mod=j的数字个数
        for (int k = 0; k < m; k++) {
            if (dp[i][k] <= 0) continue;
            for (int t = 0; t < 2; t++) {
                int tmp = (three[i] * t % m + k) % m;
                dp[i + 1][tmp] += dp[i][k];
            }
        }
    }
    int cur = 0;
    memset(cnt, 0, sizeof(cnt));
    for (int i = pos; i > 0; i--) {     // 遍历上界每一位数字
        if (a[i - 1]) {                 // 当上界该位为1时，说明取0之后的数位就无限制
            for (int j = 0; j < m; j++) {
                int t = (cur + j) % m;
                cnt[t] += dp[i - 1][j];
            }
            cur += three[i - 1];        // cur保存到当前数位每一位都取上界之后得到的值
            cur %= m;
        }
    }
    cnt[cur]++;     // 最后要加上上界这个数字
    cnt[0]--;       // 提议要求0不可取，去掉0
    LL ans = 0;
    for (int i = 0; i < m; i++)
        ans ^= cnt[i];
    printf("%lld\n", ans);
}

int main() {
    //freopen("in.txt", "r", stdin);
    int T;
    scanf("%d", &T);
    while (T--) {
        scanf("%lld%d", &n, &m);
        solve();
    }
    return 0;
}