原题
题目描述
设
S
(
x
)
S(x)
S(x)表示十进制表示下
x
x
x的每位数字之和,当
S
(
A
)
S(A)
S(A)>
S
(
B
)
S(B)
S(B)时,(A,B)表示一个和谐对。(
1
≤
N
≤
10
1≤N≤10
1≤N≤10100)
给定
N
N
N,求满足
0
≤
A
≤
B
≤
N
0≤A≤B≤N
0≤A≤B≤N的和谐对的数量,答案对
10
10
109
+
7
+7
+7取模。
样例
输入
100
输出
967
思路
因为
1
≤
N
≤
10
1≤N≤10
1≤N≤10100,所以直接存储或者直接暴力都不行。
这时候我们会想到记忆化,而且输入的数字数位非常多,所以我们可以想到用树形DP做这道题。
首先我们会想到,
d
p
[
p
o
s
]
[
x
]
[
y
]
[
l
1
]
[
l
2
]
dp[pos][x][y][l1][l2]
dp[pos][x][y][l1][l2]表示考虑第
p
o
s
pos
pos个数字,
x
x
x表示
S
(
A
)
S(A)
S(A),
y
y
y表示
S
(
B
)
S(B)
S(B),
l
1
l1
l1表示前pos-1位数字
B
B
B与
N
N
N是否相等,
l
2
l2
l2表示前pos-1位数字
A
A
A与
B
B
B是否相等。
但是这样算出来的复杂度似乎有些问题。
因为我们并不要知道
A
A
A与
B
B
B具体各自的数字是多少,只需知道它们的差即可。
又因为它们的差值可能为负数,所以刚开始基准的数字至少要
9
×
100
=
900
9×100=900
9×100=900,所以数组大小至少要
2
×
900
=
1800
2×900=1800
2×900=1800。
因此,我们可以想到。
d
p
[
p
o
s
]
[
d
f
]
[
l
1
]
[
l
2
]
dp[pos][df][l1][l2]
dp[pos][df][l1][l2]表示考虑第
p
o
s
pos
pos个数字,
d
f
df
df表示
S
(
A
)
−
S
(
B
)
S(A)-S(B)
S(A)−S(B),
l
1
l1
l1表示前pos-1位数字
B
B
B与
N
N
N是否相等,
l
2
l2
l2表示前pos-1位数字
A
A
A与
B
B
B是否相等。
这样就可以解决复杂度的问题。若还不理解,可以看看代码。
代码
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7,maxn=105;
char a[maxn];
int n,dp[maxn][maxn*20][2][2];
int dfs(int pos,int df,bool limit1,bool limit2)
{
if(pos>=n)return df>1000;//大于基准表示S(A)>S(B),累加
if(dp[pos][df][limit1][limit2]!=-1)return dp[pos][df][limit1][limit2];
int x=limit1?(a[pos]-'0'):9,ans=0;
for(int i=0;i<=x;i++)
{
int y=limit2?i:9;
for(int j=0;j<=y;j++)
ans=(ans+dfs(pos+1,df+j-i,limit1&(i==x),limit2&(j==y)))%mod;
}
return dp[pos][df][limit1][limit2]=ans;//记忆化
}
int main()
{
scanf("%s",a);n=strlen(a);memset(dp,-1,sizeof(dp));
printf("%d",dfs(0,1000,1,1));//基准为1000
return 0;
}