Harmony Pairs

原题请看这里

题目描述：

设S(x)S(x)S(x)表示十进制表示下xxx的每位数字之和，当S(A)S(A)S(A) >>> S(B)S(B)S(B)时(A,B)(A,B)(A,B)表示一个和谐对。
给定一个数NNN，求满足0≤A≤B≤N0≤A≤B≤N0≤A≤B≤N的和谐对(A,B)(A,B)(A,B)的数量，答案对109+710^9+7109+7取模。

输入描述：

输入只有一行，表示一个整数NNN((( 111 ≤≤≤ NNN ≤≤≤ 1010010^{100}10100 )))

输出描述：

输出一个整数，表示答案。

样例输入：

100

样例输出：

967

思路：

一看到这个NNN的取值范围，又看到和数字有关，那不就是数位dpdpdp嘛…
首先，我们回想数位dpdpdp的做法：：：暴力枚举，使枚举方式满足dpdpdp数组各维度的性质，最后再记忆化。
数位DP详解
接下来我们就可以看这道题目了，我们确定dpdpdp数组的第一个维度为pospospos，表示当前所在的位置。
然后我们很容易想到第二个和第三个维度分别表示AAA前面的位数和和BBB前面的位数和，第四个和第五个维度分别表示AAA的高位是否与BBB的相同，BBB的高位是否与NNN相同，那么问题来了，如果这样开数组，那么dpdpdp数组就是这个样子的：
dp[100][900][900][2][2]dp[100][900][900][2][2]dp[100][900][900][2][2]
可这明显开不下呀！
这时我们就想第二第三个维度是否可以简化？答案是肯定的！我们将二三两个维度压缩成一个维度，表示AAA前面的位数和−-−BBB前面的位数和，那么dpdpdp数组就可以变为：
dp[100][2000][2][2]dp[100][2000][2][2]dp[100][2000][2][2]
解决了dpdpdp数组的问题，接下来就好办了，具体做法详见代码。

AC Code:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MAXN=105;
const int mod=1e9+7;
int dp[MAXN][MAXN*20][2][2],dig[MAXN],len;
char s[MAXN];
ll dfs(int pos,int sum,int lit1,int lit2)
{
if(!pos) return sum>1000;//1000的偏移量可以保证差值非负
if(~dp[pos][sum][lit1][lit2]) return dp[pos][sum][lit1][lit2];//如果已经搜过了就直接返回值
ll ret=0;
for(int i=0;i<=(lit1?dig[pos]:9);i++)//限制条件B<=N
for(int j=0;j<=(lit2?i:9);j++)//限制条件A<=B
ret=(ret+dfs(pos-1,sum+j-i,lit1&i==dig[pos],lit2&i==j))%mod;
return dp[pos][sum][lit1][lit2]=ret;//记忆化
}
int main()
{
memset(dp,-1,sizeof(dp));
scanf("%s",s);
len=strlen(s);
for(int i=0;i<len;i++)
dig[len-i]=s[i]-'0';
printf("%lld\n",dfs(len,1000,1,1)%mod);
}