2019年icpc上海网络赛 L Digit sum（数位DP）

题目链接：点击这里

这道题对友用的暴力超时，他们打表过了，不过应该是数据水，这道题的正解应该是数位DP。

就是找1-n所有数，b进制下各个位的数的和，这不就是数位DP吗，虽然我DP很差，但是我套了个板子就过了。

AC代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<stdlib.h>
#include<queue>
#include<set>
#include<map>
#include<iomanip>
#include<math.h>
using namespace std;
typedef long long ll;
typedef double ld;
const ll INF=1e14;
int i,j,k;
int n,t,m,b;
int x,y;
int res,cnt,temp,sum;
int cas=0;

int slove1(int temp, int n, int k)
{
    int res=0,cnt=1;
    for (int i=0; i<n; i++)
        cnt*=k;
    res=temp*cnt+cnt*n*(k-1)/2;
    return res;
}

int slove2(int temp, int n, int k)
{
    if(n<k)
    {
        int res=0;
        for(int i=0; i<=n; i++)
            res+=temp+i;
        return res;
    }
    int nn=n,cnt=1,res=0;
    int mm=0;
    while(nn>=k)
    {
        cnt*=k;
        mm++;
        nn/=k;
    }
    for(int i=0; i<nn; i++)
        res+=slove1(temp+i,mm,k);
    res+=slove2(temp+nn,n-nn*cnt,k);
    return res;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&n,&b);
        printf("Case #%d: %d\n",++cas,slove2(0,n,b));
    }
    return 0;
}