2020牛客多校第六场H-Harmony Pairs（数位DP）

最新推荐文章于 2020-11-12 16:22:46 发布

ding_ning123

最新推荐文章于 2020-11-12 16:22:46 发布

阅读量1.1k

点赞数 2

分类专栏： 2020牛客暑期多校训练营

本文链接：https://blog.csdn.net/ding_ning123/article/details/107633794

版权

2020牛客暑期多校训练营专栏收录该内容

25 篇文章 0 订阅

订阅专栏

Description

在这里插入图片描述

题解

$10^{100}$ 的数据明示算法：数位DP
令状态 $d p [x] [d] [l] [s]$
$x$ :搜到位置
$d$ : $A$ 前面位数和 $- B$ 前面位数和（防负数偏移 $1000$ ）
$l$ :limit,当前位之前 $B$ 是否等于 $N$ ,反映限制 $B\leq N$
$s$ :当前位之前 $A$ 是否等于 $B$ ,反映限制 $A\leq B$

#include <cstdio>
#include <cstring>
const int N=110,mod=1e9+7;
using namespace std;
int n,digit[N],dp[N][N*20][2][2];
char s[N];
int dfs(int x,int d,bool l,bool s){
	if(!x) return d>1000;//S(A)>S(B)
	if(~dp[x][d][l][s]) return dp[x][d][l][s];
	int ret=0,lim=l?digit[x]:9;
	for(int i=0;i<=lim;i++)
		for(int j=0;j<=(s?i:9);j++)
			ret=(ret+dfs(x-1,d+j-i,l&(digit[x]==i),s&(i==j)))%mod;
	return dp[x][d][l][s]=ret;		
}int main(){
	memset(dp,-1,sizeof(dp));
	scanf("%s",s),n=strlen(s);
	for(int i=0;i<n;i++) digit[n-i]=s[i]-'0';
	printf("%d\n",dfs(n,1000,1,1));
}