数位DP 求满足条件的和

这是一个板子：求数位DP中满足一些条件的数的和

 

方法：

如果我们求满足条件的数的个数只需要开一个数组就行。

如果我们求满足条件的数的和，则需要开一个数组结构体   node   里面有两个元素，一个是cnt，一个是sum。

cnt就是记录满足条件的数量，sum就是记录满足条件的数字和。

假设我们的答案 ans ,递归返回的 next  (ans,next 均为结构体)，

我们更新ans：

ans.cnt+=next.cnt (这个是基础数位DP的东西，不用多说)

ans.sum+=next.sum+i*10^(pos-1)*next.cnt;

为什么这么更新呢？ 我们假设我们求 [1,123]的和（用数位DP求）。

我们向下递归就是 求 0_ _和 1_ _ (_表示未知位置);

后面继续递归，递归完成后 1*100+（百位为1满足条件的和 个数（1+23）*23 / 2 +1，最后的1后面为0时的情况  ）

而1__,由10_,11_,12_递归回来的。后面可以自己模拟一下。

主要代码时：

ans.cnt+=next.cnt;
ans.sum+=next.sum+i*pre[pos]*next.cnt;

全部代码：

///#include<bits/stdc++.h>
///#include<unordered_map>
///#include<unordered_set>
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<bitset>
#include<set>
#include<stack>
#include<map>
#include<list>
#include<new>
#include<vector>
#define MT(a,b) memset(a,b,sizeof(a));
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const double E=2.718281828459;
const ll MOD=1e9+7;
const int INF=0x3f3f3f3f;

struct node
{
    ll cnt;
    ll sum;
}ans_sum,dp[20];

ll pre[20],num[22];

node dfs(ll pos,bool lead,bool limit)
{
    if(pos==0)
        return node{1,0};
    if(!lead&&!limit&&dp[pos].cnt!=-1)
        return dp[pos];
    ll up=limit?num[pos]:9;
    node ans={0,0};
    for(ll i=0;i<=up;i++)
    {
         node next=dfs(pos-1,limit&&i==num[pos],lead&&i==0);
         ans.cnt+=next.cnt;
         ans.sum+=next.sum+i*pre[pos]*next.cnt;
    }
    if(!lead&&!limit)
        dp[pos]=ans;
    return ans;
}

ll solve(ll x)
{
    ll sign=0;
    while(x)
    {
        num[++sign]=x%10;
        x=x/10;
    }
    memset(dp,-1,sizeof(dp));
    ans_sum=dfs(sign,1,1);
    return ans_sum.sum;
}

void init()
{
    pre[1]=1;
    for(int i=2;i<=10;i++)
        pre[i]=pre[i-1]*10;
    return ;
}

int main()
{
    ll x;
    init();
    while(~scanf("%lld",&x))
        printf("%lld\n",solve(x));
    return 0;
}