网易笔试 分饼干

[编程题] 分饼干

时间限制:1秒

空间限制:32768K

易老师购买了一盒饼干,盒子中一共有k块饼干,但是数字k有些数位变得模糊了,看不清楚数字具体是多少了。易老师需要你帮忙把这k块饼干平分给n个小朋友,易老师保证这盒饼干能平分给n个小朋友。现在你需要计算出k有多少种可能的数值
输入描述:
输入包括两行:

第一行为盒子上的数值k,模糊的数位用X表示,长度小于18(可能有多个模糊的数位)

第二行为小朋友的人数n


输出描述:
输出k可能的数值种数,保证至少为1

输入例子:
9999999999999X
3

输出例子:
4
思路:动态规划

设dp[i][j]表示前i个数字且余数为j的可能数。 则递推公式为

dp[i][newj] += dp[i-1][j]

其中newj表示由前i-1位组成的数对n的余数和第i位组成的新数 对n 的余数

最后的结果为dp[len][0],即所有位且余数为0的可能性

#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;

int main() {
   // freopen("input.txt","r",stdin);
    string str;
    int n;
    cin>>str>>n;
    long long dp[20][10004];
    memset(dp,0,sizeof(dp));
    dp[0][0] =1 ;
    int len = (int)str.length();
    for(int i = 1; i <= len; i++) {
        for(int j = 0; j < n; j++) {
            if(str[i-1] == 'X') {
                for(int k = 0; k < 10; k++) {
                    int newj = (j * 10 + k) % n;
                    dp[i][newj] += dp[i-1][j];
                }
            }
            else {
                int newj = (j * 10 + str[i-1] - '0') % n;
                dp[i][newj] += dp[i-1][j];
            }
        }
    }
    cout<



你可能感兴趣的:(★算法入门,------数论,------动态规划)