hihoCoder 1259 A Math Problem 数位dp

 真的是区域赛没弱题这题貌似是区分北京站银牌与铜牌的一道题。

代码就先看一下别人的吧。

思路:我一直以为从低位向高位进行计算,然而发现很麻烦,例如11001100010000

你需要先算10000000000000,再算1000000000000,再算1000000000,会超时。

因此我们需要另一种方法。

例如110101

先算1,读取数据的时候从最高为开始读取,1,1,0,1,0,1这才是读取顺序,注意一下,转移一次之后,末尾要看他是多少。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>

using namespace std;
typedef long long ll;
const int maxn = 65540;
const int maxm = 65;

int K;
ll N, dp[maxm][maxn];

ll solve () {
    int s = 0, bit[maxm];
    for (int i = 60; i >= 0; i--) {
        bit[i] = N % 2;
        N /= 2;
    }

    memset(dp, 0, sizeof(dp));
    for (int i = 0; i <= 60; i++) {
        for (int j = 0; j < bit[i]; j++)
            dp[i][(s*3+j)%K]++;
        s = (s * 3 + bit[i]) % K;

        for (int j = 0; j < K; j++) {
            dp[i+1][j*3 % K] += dp[i][j];
            dp[i+1][(j*3+1)%K] += dp[i][j];
        }
    }
    dp[60][s]++;
    dp[60][0]--;

    ll ans = 0;
    for (int i = 0; i < K; i++)
        ans ^= dp[60][i];
    return ans;
}

int main () {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        scanf("%lld%d", &N, &K);
        printf("%lld\n", solve());
    }
    return 0;
}




你可能感兴趣的:(hihoCoder 1259 A Math Problem 数位dp)