[总结]数位统计模板

偷看了下7k+大牛的数位统计dp写法,通常的数位dp可以写成如下形式:

int dfs(int i, int s, bool e) {

    if (i==-1) return s==target_s;

    if (!e && ~f[i][s]) return f[i][s];

    int res = 0;

    int u = e?num[i]:9;

    for (int d = first?1:0; d <= u; ++d)

        res += dfs(i-1, new_s(s, d), e&&d==u);

    return e?res:f[i][s]=res;

}

其中:

f为记忆化数组;

i为当前处理串的第i位(权重表示法,也即后面剩下i+1位待填数);

s为之前数字的状态(如果要求后面的数满足什么状态,也可以再记一个目标状态t之类,for的时候枚举下t);

e表示之前的数是否是上界的前缀(即后面的数能否任意填)。

for循环枚举数字时,要注意是否能枚举0,以及0对于状态的影响,有的题目前导0和中间的0是等价的,但有的不是,对于后者可以在dfs时再加一个状态变量z,表示前面是否全部是前导0,也可以看是否是首位,然后外面统计时候枚举一下位数。It depends.

于是关键就在怎么设计状态。当然做多了之后状态一眼就可以瞄出来。

注意:

不满足区间减法性质的话(如hdu 4376),不能用solve(r)-solve(l-1),状态设计会更加诡异。

下面给几个例子:

codeforces 55d / spoj JZPEXT(7k+的鬼畜题,卡时限以及代码长度,所以状态设计要注意)

//JZPEXT 

 #include<cstdio> 

 #include<cstring> 

 typedef long long ll; 

 int c[2555],n[20],g[2555][10],h[255][10]; 

 ll f[20][50][255],l,r; 

   

 int gcd(int x,int y){ 

     return y?gcd(y,x%y):x; 

 } 

   

 ll dfs(int l,int m,int r,bool z){ 

     if(l==-1)return !(r%m); 

     if(!z&&f[l][c[m]][r]!=-1) return f[l][c[m]][r]; 

     ll res=0; 

     int u=z?n[l]:9; 

     for(int d=0;d<=u;++d) 

         res+=dfs(l-1,g[m][d],l?h[r][d]:r*10+d,z&&d==u); 

     return z?res:f[l][c[m]][r]=res; 

 } 

   

 ll s(ll x){ 

     int l=0; 

     for (;x;x/=10)n[l++]=x%10; 

     return dfs(l-1,1,0,1); 

 } 

   

 int main(){ 

     memset(f,-1,sizeof f); 

     int i,j,t; 

     for(i=1,r=-1;i<=2520;++i) 

         c[i]=r+=!(2520%i); 

     for(j=0;j<10;++j){ 

         for(i=1;i<=2520;++i) 

             g[i][j]=j?i*j/gcd(i,j):i; 

         for(i=0;i<252;++i) 

             h[i][j]=(i*10+j)%252; 

     } 

     scanf("%d",&t); 

     while(t--){ 

         scanf("%lld%lld",&l,&r); 

         printf("%lld\n",s(r)-s(l-1)); 

     } 

     return 0; 

 }

hdu 4352

#include <cstdio> 

 #include <cstring> 

 #include <algorithm> 

 using namespace std; 

   

 typedef long long ll; 

   

 ll f[20][1<<10][11]; 

 int num[20]; 

 int news[10][1<<10]; 

 ll l, r; 

 int k; 

   

 int getnews(int x, int s) { 

     for (int i = x; i < 10; ++i) 

         if (s&(1<<i)) return s^(1<<i)|(1<<x); 

     return s|(1<<x); 

 } 

   

 ll dfs(int i, int s, bool e, bool z) { 

     if (i==-1) return __builtin_popcount(s)==k; 

     if (!e && f[i][s][k]!=-1) return f[i][s][k]; 

     ll res = 0; 

     int u = e?num[i]:9; 

     for (int d = 0; d <= u; ++d) 

         res += dfs(i-1, z&&!d?0:news[d][s], e&&d==u, z&&!d); 

     return e?res:f[i][s][k]=res; 

 } 

   

 ll solve(ll x) { 

     int len =0; 

     for (; x; x/=10) 

         num[len++] = x%10; 

     return dfs(len-1, 0, 1, 1); 

 } 

   

 int T; 

   

 int main() { 

     int i, j; 

     for (i = 0; i < 10; ++i) 

         for (j = 0; j < 1<<10; ++j) 

             news[i][j] = getnews(i, j); 

     memset(f, -1, sizeof(f)); 

     scanf("%d", &T); 

     for (int t = 1; t <= T; ++t) { 

         scanf("%I64d%I64d%d", &l, &r, &k); 

         printf("Case #%d: ", t); 

         printf("%I64d\n", solve(r)-solve(l-1)); 

     } 

     return 0; 

 }

spoj BALNUM

//BALNUM

#include <cstdio>

#include <cstring>

#include <algorithm>

using namespace std;



typedef long long ll;



int num[22];

ll f[22][60000];

int T;

int news[60000][10];

int exp[10];

bool ok[60000];

ll l, r;



ll dfs(int i, int s, bool e) {

    if (i==-1) return ok[s];

    if (!e && ~f[i][s]) return f[i][s];

    ll res = 0;

    int u = e?num[i]:9;

    for (int d = s?0:1; d <= u; ++d)

        res += dfs(i-1, news[s][d], e&&d==u);

    return e?res:f[i][s]=res;

}



int getnum(ll x, int num[]) {

    int len = 0;

    for (; x; x/=10)

        num[len++] = x%10;

    return len;

}



ll solve(ll x) {

    int len = getnum(x, num);

    ll res = 0;

    for (int i = 1; i < len; ++i)

        res += dfs(i-1, 0, 0);

    return res+=dfs(len-1, 0, 1);

}



bool judge(int exp[]) {

    for (int i = 0; i < 10; ++i) {

        if (!exp[i]) continue;

        if (i&1) {

            if (exp[i]&1) return 0;

        }

        else {

            if (~exp[i]&1) return 0;

        }

    }

    return 1;

}



int main() {

    int i, j, k;

    memset(f, -1, sizeof f);

    for (int s = 0; s < 59049; ++s) {

        j = s;

        for (i = 0; i < 10; ++i, j/=3)

            exp[i] = j%3;

        if (s) ok[s] = judge(exp);

        else ok[s] = 0;

        for (i = 0; i < 10; ++i) {

            int t = exp[i];

            if (exp[i]&1) exp[i] = 2;

            else exp[i] = 1;

            for (j = 9; j >= 0; --j, news[s][i]*=3)

                news[s][i] += exp[j];

            news[s][i] /= 3;

            exp[i] = t;

        }

    }

    //printf("#%d\n", news[0][3]);

    scanf("%d", &T);

    while (T--) {

        scanf("%lld%lld", &l, &r);

        printf("%lld\n", solve(r)-solve(l-1));

    }

    return 0;

}

其实和翁教主,肖神讨论后,想到了更多的鬼畜题,以后再补充吧。

你可能感兴趣的:(总结)