数位DP习题

1.度的数量

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
const int N=35;
int f[N][N];
using namespace std;
int l,r,k,b;
int dp(int n){
    vector<int>v;
    while(n){
        v.push_back(n%b);
        n=n/b;
    }
    int last=0,res=0;
    for(int i=v.size()-1;i>=0;i--){
        int num=v[i];
        if(num>0){  
            res+=f[i][k-last];  // 当前位置取0,且限制大于0 剩下的数字可以随便选
            //下面是当前位置选1
            if(num>1){   //如果当前位置限制大于1,则当前位置选1后,后面的数选择没有限制
            //直接从剩下的位置中选出来剩下来的1即可
            //注意这里last不能直接++,下面还会用到last如果直接++的话,会影响下面的判断
                if(k-last-1>=0) res+=f[i][k-last-1];
                break;
            }
            else {   // 当前位置的限制正好等于1,则当前位置选1后,后面的数有限制(如果随便选的话
            //有可能大于n) ,所以要继续枚举
                last++;//当前位置选1
                if(last>k) break;// 如果发现当前位置选1后超出限定个1,则不合法,直接break
            }
        }
        if(!i&&k==last) res++;  // 这种情况只会发生在取到最后一位数字,最后一位数字恰好为0
        //且前面刚好取了k个1,如果最后一位数字大于0且前面已经取完了k个1,那么这种方案会在第19行加上
    }
    return res;
}
void init(){
    for(int i=0;i<N;i++){
        for(int j=0;j<=i;j++){
            if(!j) f[i][j]=1;
            else f[i][j]=f[i-1][j-1]+f[i-1][j];
        }
    }
}
int main(){
    cin >>  l >> r >> k >> b;
    init();
    cout << dp(r)-dp(l-1);
    return 0;
}

参考题解

//记忆化搜索做法
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=35;
int f[N][N];//f[i][j]=k 从前i位中选 且已选1的个数为j的方案数为k(不贴合状态下)
int l,r,k,b;
int a[N];
int dfs(int i,int j,bool limit){ // 当前枚举到第i个数 且 已经选了j个1,当前贴合状态为limit
    //if(j>k) return 0;
    if(i==0){ // 数枚举完了
        if(j==k) return 1;  // 如果符合题意,说明当前方案数有效,返回1
        else return 0;
    }
    if(!limit&&f[i][j]!=-1){  //如果当前不贴合 而且f[i][j] 已经计算过了 则直接记忆化操作
        return f[i][j];
    }
    int res=0,m;
    if(limit){ //如果贴合则当前位数只能取到当前位的最大值 比如a[i]=6 第i位只能取到6,如果不贴合可以取到9
        m=min(1,a[i]);
    }
    else {
        m=1;  // 不贴合 m本可以取到9 但是本题取大于1的数字无意义 所以取到1
    }
    for(int op=0;op<=m;op++){
      //  if ((op == 1 && j == k) || op > 1) continue;
        res+=dfs(i-1,j+(op==1),limit&&(op==a[i]));
    }
    if(!limit){
        return f[i][j]=res;
    }
    else {
        return res;
    }
}
int dp(int n){
    memset(f,-1,sizeof f);
    int cnt=0;
    while(n) {
        a[++cnt]=n%b;
        n=n/b;
    }
    return dfs(cnt,0,1);
}
int main(){
    cin >> l >> r >> k >> b;
    cout << dp(r)-dp(l-1);
    return 0;
}

参考题解
参考题解
参考题解

2.数字游戏

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=50;
int l,r;
int a[N],f[N][N];//表示前i个数且最后一位是j的符合题意的个数比如f[2][3]=5 表示 x3xxxxx 符合题意的有5种
int dfs(int i,int j,bool limit){
    if(i==0){
        return 1;
    }
    if(!limit&&f[i][j]!=-1) return f[i][j];
    int res=0,m;
    if(limit){
        m=a[i];
    }
    else {
        m=9;
    }
    for(int op=0;op<=m;op++){
        if(op<j) continue;
        res+=dfs(i-1,op,limit&&(op==a[i]));
    }
    if(limit){
        return res;
    }
    else {
        return f[i][j]=res;
    }
}
int dp(int n){
    int cnt=0;
    memset(f,-1,sizeof f);
    while(n){
        a[++cnt]=n%10;
        n=n/10;
    }
    return dfs(cnt,0,1);
}
int main(){
    while(cin >> l >> r){
       // cout << dp(r) << "here";
        cout << dp(r)-dp(l-1) << endl;
    }
    return 0;
}

3.Windy数

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=35;
int f[N][N][2]; // f[i][j][1]=k 表示前i位最后一位是j且含有前导0(即前面一位是0)的方案
int l,r;
int a[N];
int dfs(int i,int j,bool limit,int lead){
    if(i==0) return 1;
    if(!limit&&f[i][j][lead]!=-1) return f[i][j][lead];
    int res=0,m=(limit)?a[i]:9;
    for(int op=0;op<=m;op++){
        if(lead||abs(op-j)>1) res+=dfs(i-1,op,limit&&op==a[i],lead&&op==0);
    }
    return limit?res:f[i][j][lead]=res;
}
int dp(int n){
    int cnt=0;
    memset(f,-1,sizeof f);
    while(n){
        a[++cnt]=n%10;
        n=n/10;
    }
    return dfs(cnt,0,1,1);
}
int main(){
    cin >> l >> r;
    cout << dp(r)-dp(l-1);
    return 0;
}

4.数字游戏2

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=35;
int f[N][110];
int l,r,mod;
int a[N];
int dfs(int i,int j,int limit){
    if(i==0) {
        if(j==0) return 1;
        else return 0;
    }
    if(!limit&&f[i][j]!=-1) return f[i][j];
    int res=0,m=limit?a[i]:9;
    for(int op=0;op<=m;op++){
        res+=dfs(i-1,(j+op)%mod,limit&&(op==a[i]));
        //cout <<i<<" "<<op<<" "<< res <<" "<<m<< endl;
    }
    return limit?res:f[i][j]=res;
}
int dp(int n){
    memset(f,-1,sizeof f);
    int cnt=0;
    while(n){
        a[++cnt]=n%10;
        n=n/10;
    }
    return dfs(cnt,0,1);
}
int main(){
    while(cin>>l>>r>>mod){
    cout << dp(r)-dp(l-1) << endl;}
    return 0;
}

5.不要62

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=35;
int l,r;
int a[N];
int f[N][N];
int dfs(int i,int j,int last,int limit){
    if(i==0) return 1;
    if(!limit&&f[i][j]!=-1) return f[i][j];
    int res=0,m=limit?a[i]:9;
    for(int op=0;op<=m;op++){
        if(op==4||(op==2&&last==6)) continue;
        res+=dfs(i-1,op,op,limit&&a[i]==op);
    }
    return limit?res:f[i][j]=res;
}
int dp(int n){
    int cnt=0;
    memset(f,-1,sizeof f);
    while(n){
        a[++cnt]=n%10;
        n=n/10;
    }
    return dfs(cnt,0,0,1);
}
int main(){
    while(cin >>l >> r){
    if(l!=0&&r!=0) cout << dp(r)-dp(l-1)<<endl;}
    return 0;
}

6.恨7不成妻

// f[i][j][k]={cnt,sum,sum2} 表示前i位每位数字之和%7为j,前i位所组成的数字%7为k,在满足
// i,j,k的条件下 第i-1位至第最后一位所能组成的无关7的个数为cnt,第i-1位至第最后一位所能组成的无关7的数的
//和为sum,第i-1位至第最后一位所能组成的无关7的数的平方和为sum2
//Q1: 怎么由后i-1位的无关数的和推出后i位无关数的和?
// res.sum+=tmp.cnt*B+tmp.sum;  (B为10的i次幂)
//Q2: 怎么由后i-1位的无关数的平方和推出后i位无关数的平方和?
// res.sum2+=tmp.cnt*B^2+tmp.sum2+2*B*tmp.sum
#include<iostream>
#include<cstring>
#include<algorithm>
#define int long long
using namespace std;
const int N=22,mod=1e9+7;
struct node{
    int cnt=0,sum,sum2;
}f[N][10][10];
int t,l,r;
int p[N];
int a[N];
node dfs(int i,int j,int k,int limit){
    if(i==0){
        if(j!=0&&k!=0) return {1,0,0};
        else return {0,0,0};
    }
    if(!limit&&f[i][j][k].cnt!=-1) return f[i][j][k];
    node res={0,0,0};   // 记得初始化
    int m=limit?a[i]:9;
    for(int op=0;op<=m;op++){
        if(op==7) continue;                                    //发现了一件事情 关于取余
        node tmp=dfs(i-1,(j+op)%7,(k*10+op)%7,limit&&op==a[i]);    // (k%7*10+op)%7==(k*10+op)%7 (k<=6)
        (res.cnt+=tmp.cnt)%=mod;
        int B=op*p[i-1]%mod;
        (res.sum+=tmp.cnt*B%mod+tmp.sum)%=mod;
        (res.sum2+=tmp.cnt*B%mod*B%mod+tmp.sum2%mod+2*B%mod*tmp.sum%mod)%=mod;
    }                                                      
    return limit?res:f[i][j][k]=res;
}
int dp(int n){
    int cnt=0;
    memset(f,-1,sizeof f);
    while(n){
        a[++cnt]=n%10;
        n=n/10;
    }
    return dfs(cnt,0,0,1).sum2;
}
signed main(){
    cin >> t;
    p[0]=1;
    for(int i=1;i<N;i++) p[i]=(10*p[i-1])%mod; // p[i]=10^i;
    while(t--){
        int l,r;
        cin >> l >> r;
        cout << (dp(r)-dp(l-1)+mod)%mod << endl;
    }
    return 0;
}

参考题解

7.计数统计

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=35;
int l,r;
struct node{
    int a=0,b=0,c=0,d=0,e=0,f=0,g=0,h=0,i=0,j=0,cnt=0;
}f[N][2];  //f[i].c 表示的是从i位到最后一位所含有的2的数量,f[i].cnt表示的是从i位到最后一位所能组成的所有数的个数
int a[N];  //f[i].c= 从第i-1位到最后一位含有2的数量+从第i-1位到最后一位所能组成数的个数
// 比如统计1XXX中含有1的个数  那么就是XXX中含有1个数 加上 i-1位到最后一位所能组成数的个数   
node dfs(int i,int limit,int lead){
    if(i==0) return {0,0,0,0,0,0,0,0,0,0,1};
    if(!limit&&f[i][lead].a!=-1) return f[i][lead];
    node res={0,0,0,0,0,0,0,0,0,0,0};
    int m=limit?a[i]:9;
    for(int op=0;op<=m;op++){
        node tmp=dfs(i-1,limit&&op==a[i],lead&&op==0);
        if(op==0&&lead!=1) res.a+=tmp.cnt;
        else if(op==1) res.b+=tmp.cnt;
        else if(op==2) res.c+=tmp.cnt;
        else if(op==3) res.d+=tmp.cnt;
        else if(op==4) res.e+=tmp.cnt;
        else if(op==5) res.f+=tmp.cnt;
        else if(op==6) res.g+=tmp.cnt;
        else if(op==7) res.h+=tmp.cnt;
        else if(op==8) res.i+=tmp.cnt;
        else if(op==9) res.j+=tmp.cnt;
        res.a+=tmp.a;
        res.b+=tmp.b;
        res.c+=tmp.c;
        res.d+=tmp.d;
        res.e+=tmp.e;
        res.f+=tmp.f;
        res.g+=tmp.g;
        res.h+=tmp.h;
        res.i+=tmp.i;
        res.j+=tmp.j;
        res.cnt+=tmp.cnt;
    }
    return limit?res:f[i][lead]=res;
}
node dp(int n){
    int cnt=0;
    memset(f,-1,sizeof f);
    memset(a,0,sizeof a);
    while(n){
        a[++cnt]=n%10;
        n=n/10;
    }
    return dfs(cnt,1,1);
}
int main(){
    while(cin>>l>>r){
        if(l==0&&r==0) break;
        if(l>r) swap(l,r);
        node tmpl=dp(l-1);
        node tmpr=dp(r);
        cout << tmpr.a-tmpl.a<<" "<< tmpr.b-tmpl.b<<" "<< tmpr.c-tmpl.c<<" "<< tmpr.d-tmpl.d<<" "<< tmpr.e-tmpl.e<<" "<< tmpr.f-tmpl.f<<" "<< tmpr.g-tmpl.g<<" "<< tmpr.h-tmpl.h<<" "<< tmpr.i-tmpl.i<<" "<< tmpr.j-tmpl.j<<" "<<endl;
    }
    return 0;
}




#include <iostream>
#include <cstring>
using namespace std;

const int N = 10;

int l, r;
int a[N], f[N][N];

//sum表示前面已经有了多少个x,前面的sum+后面的记忆化过的f[n][n]就是一个数中含x的数量
//这种思路其实就是穷举每一个数 看看每一个数里面有多少个1 或者其他的
// 比如12341(假设最后一位上限为1) 枚举完了 含有2个1
//     12340           枚举完了含有一个1
//那么 1234_这种一共含有3个1 即f[5][1]=3
// 如果我枚举到了1235_ 则可以直接返回3了 不必再枚举了
int dfs(int pos, int sum, int x, int lead, int limit)
{
    if (!pos)
        return sum;
    if (!limit && !lead && ~f[pos][sum])
        return f[pos][sum];

    int res = 0, up = limit ? a[pos] : 9;
    for (int i = 0; i <= up; ++i)
        if ((!lead || i) && (i == x)) 
            res += dfs(pos - 1, sum + 1, x, lead && !i, limit && i == up);
        else
            res += dfs(pos - 1, sum, x, lead && !i, limit && i == up);
    //什么时候能把当前填入的数计算进去呢?
    //1. 有前导0,但是当前数不为0,且计的数位和当前数相等,可以算进去,比如01____,1的个数+1
    //2. 有前导0,但是当前数为0,且计的数位和当前位相等,不能算进去,比如00_____,前面两个不算0的个数
    //3. 无前导0,当前数可以为0可以不为0,且计的数位和当前位相等,可以算进去,比如10___,可以算0的个数

    if (!limit && !lead)
        f[pos][sum] = res;
    return res;
}



int dp(int x, int i)
{
    memset(f, -1, sizeof(f));
    int len = 0;
    while (x)
        a[++len] = x % 10, x /= 10;
    return dfs(len, 0, i, 1, 1);
}

int main()
{
    while (cin >> l >> r, l || r)
    {        
        if (l > r)
            swap(l, r);
        for (int i = 0; i <= 9; ++i)
            cout << dp(r, i) - dp(l - 1, i) << ' ';
        puts("");
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值