CodeForces 215E Periodical Numbers 数位DP

这题做了有好几天了，终于过了= =

完全不懂网上题解的递推写法，只能自己用记忆化搜索瞎搞，总算是搞出来了。

具体策略就是记忆化搜索的时候用一个tmp数组记录最前面len个的值，然后后面的数字必须要和前len相同。

但是麻烦的是这样子会有重复，想了一个很挫的写法就是枚举算出恰好长度为ll并且循环节长度为l的数字的个数，然后全部记录下来，从小到大依次减去重复的。

真是挫。。。sad

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
#include <string>
#include <iostream>
#include <map>
#include <cstdlib>
#include <list>
#include <set>
#include <queue>
#include <stack>

using namespace std;

typedef long long LL;
const int maxn = 70;
int lim[maxn],len;
int tmp[maxn];
LL f[maxn][maxn];

void getlim(LL num) {
    len = 0;
    memset(lim,0,sizeof(lim));
    while(num) {
        lim[len++] = num % 2;
        num /= 2;
    }
}

LL dfs(int now,int l,int ll,int first,int bound) {
    if(now == 0) {
        return first % l == 0 && first / l >= 2 && first == ll;
    }
    int m = bound ? lim[now - 1] : 1;
    if(first != -1 && !bound && f[now][first] != -1) return f[now][first] ;
    LL ret = 0;
    for(int i = 0;i <= m;i++) {
        if(i && first == -1) first = now;
        if(first == -1) ret += dfs(now - 1,l,ll,-1,bound && i == m);
        else {
            int pos = first - now;
            tmp[pos] = i;
            if(pos < l) ret += dfs(now - 1,l,ll,first,bound && i == m);
            else if(tmp[pos % l] == i) {
                ret += dfs(now - 1,l,ll,first,bound && i == m);
            }
        }
    }
    if(first != -1 && !bound) f[now][first] = ret;
    return ret;
}

LL solve(LL num) {
    getlim(num);
    LL ret = 0,cnt[maxn][maxn];
    memset(cnt,0,sizeof(cnt));
    for(int i = 1;i <= len / 2 + 1;i++) {
        for(int j = len;j >= 1;j--) {
            memset(f,-1,sizeof(f));
            cnt[i][j] = dfs(len,i,j,-1,1);
        }
    }
    for(int i = 2;i <= len / 2;i++) {
        for(int j = 1;j < i;j++) if(i % j == 0) {
            for(int k = i * 2;k <= len;k++) if(k % i == 0) {
                cnt[i][k] -= cnt[j][k];
            }
        }
    }
    for(int i = 1;i <= len / 2;i++) {
        for(int j = 1;j <= len;j++) {
            ret += cnt[i][j];
        }
    }
    return ret;
}

int main() {
    LL a,b; cin >> a >> b;
    cout << solve(b) - solve(a - 1) << endl;
    return 0;
}