魔力数

题目描述

如果一个数字模m等于k（即该数字除以m余数为k），则称这个数字为魔力数 现在魔法师jrq有一个十进制下的n位数，初始每个数位都是0，jrq有d点魔力，他每花费1点魔力可以使这个n位数的某一位加1，但不能超过9（即最大只能加到9），jrq可以不花光他的所有魔力，问jrq可以得到多少种不同的魔力数 魔力数可以有前导零

输入数据

四个正整数，分别代表n,d,m,k 1<=n<=10 1<=d<=15 1<=m<=1000000007 0<=k<m

输出数据

10 15 1000000007 1000000007

样例输入

2 1 10 0

样例输出

2

样例说明

满足样例条件的魔力数有00和10

解析：采用动态规划+枚举，用 dp[i][j][k] 表示 i 位数，使用 j 点魔力，余数为 k 时的，魔力数个数。

 1、动态方程：dp[i+1][j+z][(k+(POW[i+1]*z)%m)%m]+=dp[i][j][k];

当前 i 位数，消耗 j 点魔力，余数为 k 时，魔力个数已知，对 i + 1位数进行操作， 添加 z 点魔力，余数变为 (k+(POW[i+1]*z)%m)%m

dp[i+1][p][q] = dp[i][j][k]， p=j+z (0<=z<=d-j)，q= (k+(POW[i+1]*z)%m)%m

long long ORZ(int n,int d,int m,int mod){
	long long dp[11][16][1005];
	dp[0][0][0]=1;
	for(int i=0;i<n;i++){
	    for(int j=0;j<=d;j++){
	        for(int k=0;k<m;k++){
	            for(int z=0;(z<=d-j)&&(z<=9);z++){
	                dp[i+1][j+z][(k+(POW[i+1]*z)%m)%m]+=dp[i][j][k];
	            }
	            
	        }
	    }
	}
	long long ans=0;
	for(int i=0;i<=d;i++)
		ans+=dp[n][i][mod];
	return ans;
}

2、当 余数 过大时， 数组容量过大，此时采用枚举的方法。

bool check(long long s,int d){
    while(s){
        d-=s%10;
        s/=10;
        if(d<0) return 0;
    }
    return 1;
}

long long value=k,ans=0;
while(value<POW[n+1]){
    ans+=check(value,d);
    value+=m;
}

 

 完整代码：

#include<iostream>
using namespace std;
typedef long long ll;
long long POW[12]; //储存 10 的 n 次方

long long ORZ(int n,int d,int m,int mod){
	long long dp[11][16][1005];
	dp[0][0][0]=1;
	for(int i=0;i<n;i++){
	    for(int j=0;j<=d;j++){
	        for(int k=0;k<m;k++){
	            for(int z=0;(z<=d-j)&&(z<=9);z++){
	                dp[i+1][j+z][(k+(POW[i+1]*z)%m)%m]+=dp[i][j][k];
	            }
	            
	        }
	    }
	}
	long long ans=0;
	for(int i=0;i<=d;i++)
		ans+=dp[n][i][mod];
	return ans;
}

bool check(long long s,int d){
    while(s){
        d-=s%10;
        s/=10;
        if(d<0) return 0; //当操作次数大于魔力值
    }
    return 1;
}

int main()
{
    POW[1]=1;
    for(int i=2;i<=11;i++){
    	POW[i]=POW[i-1]*10;
    }

    int n,d,m,k;
    cin>>n>>d>>m>>k;

    if(m<=1000) //当余数小于1000
    	cout<<ORZ(n,d,m,k)<<endl;
    else //枚举 value = num * m + k, 枚举次数为 num + 1
    {
        long long value=k,ans=0;
        while(value<POW[n+1]){ //当前值小于上界
            ans+=check(value,d);
            value+=m; 
        }
        cout<<ans<<endl;
    }
    return 0;
}

 

参考资料： https://blog.csdn.net/qq_37868325/article/details/109140834