hihocoder 1033 交错和 （数位dp）

描述

给定一个数 x，设它十进制展从高位到低位上的数位依次是 a0, a1, ..., an - 1，定义交错和函数：

f(x) = a0 - a1 + a2 - ... + ( - 1)n - 1an - 1

例如：

f(3214567) = 3 - 2 + 1 - 4 + 5 - 6 + 7 = 4

给定 

输入

输入数据仅一行包含三个整数，l, r, k(0 ≤ l ≤ r ≤ 1018, |k| ≤ 100)。

输出

输出一行一个整数表示结果，考虑到答案可能很大，输出结果模 109 + 7。

提示

对于样例 ，满足条件的数有 110 和 121，所以结果是 231 = 110 + 121。

解题思路：

看题面可以明显的看到是一道数位dp问题。与以往的数位dp问题不同之处在于，它不仅仅需要知道答案的个数还需要知道这些数字的和。

因为数位dp算法的核心，就是把问题按照每一位分解下去，所以我们通过正常数位dp方法计算答案的个数可以知道每一位数字为 i 的个数，把和累加起来就是最后的答案。

细节请参考代码。

AC代码：

/*
* @Author: wchhlbt
* @Last Modified time: 2017-09-13
*/

#include <bits/stdc++.h>

#define Fori(x) for(int i=0;i<x;i++)
#define Forj(x) for(int j=0;j<x;j++)
#define maxn 1000007
#define inf 0x3f3f3f3f
#define ONES(x) __builtin_popcount(x)
#define _  << "  " <<
using namespace std;

typedef long long ll ;
const double eps =1e-8;
const int mod = 1000000007;
const double PI = acos(-1.0);
int dx[5] = {0,0,1,-1,0};
int dy[5] = {1,-1,0,0,0};
inline int read(){ int cnt;    scanf("%d",&cnt);   return cnt;}

ll l,r,k;
ll f[20];
struct node{
	ll n;
	ll sum;
	node(ll n = 0, ll sum = 0):n(n),sum(sum){}
};
node dp[20][400][3];//dp[i][j][k] 第i位 到当前位的交错和为j 下一位的符号k 0表示前导零 1表示+ -1表示- 
int bit[20];
node zero(0,0);
node one(1,0);
node dfs(int pos,int sum,int limit,int flag)//flag标记当前位数字前的符号应该是+ - 
{
	node res(0,0);
    if(pos < 1){
    	if(sum==k)	return one;
    	else	return zero;
    }
    int x = sum + 200;
    int o = flag + 1;
    if(!limit && dp[pos][x][o].n != -1) return dp[pos][x][o];
    int len = limit?bit[pos]:9;    
    for(int i = 0; i <= len; i++)
    {
    	node temp;
    	if(flag==0 && i!=0)
    		temp = dfs(pos-1,  i, limit&&i==len,-1);
    	else
    		temp = dfs(pos-1, sum+flag*i, limit&&i==len, -flag);
        if(temp.n>0){
        	res.n += temp.n;
        	res.sum += (temp.n % mod * i * f[pos-1]%mod);//加上这一位的值*个数
        	res.sum += temp.sum;//加上后面若干位的值
        	res.sum %= mod;        
        }      
    }
    if(!limit) dp[pos][x][o] = res;
    return res;
}

ll solve(ll n)//预处理函数，将所给定的数字，处理为每个位置上的数字（上界）
{
    int len = 0;
    while(n)
    {
        bit[++len] = n%10;
        n /= 10;
    }
    node ans = dfs(len, 0 , 1, 0);
    return ans.sum;
}

int main()
{
	f[0] = 1;
	for(int i = 1; i<=18; i++){
		f[i] = f[i-1]*10%mod;
	}
	cin>>l>>r>>k;
	memset(dp,-1,sizeof dp);//初始化为0 有可能导致大量重复计算
	cout << (solve(r)-solve(l-1)+mod)%mod << endl;
    return 0;
}