题意:
题目链接: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;
}