【bzoj1799】 AHOI2009 同类分布数位dp

最新推荐文章于 2019-07-04 22:58:56 发布

qingdaobaibai

最新推荐文章于 2019-07-04 22:58:56 发布

阅读量1.3k

点赞数 1

分类专栏：动态规划

本文链接：https://blog.csdn.net/u012288458/article/details/47057091

版权

动态规划专栏收录该内容

63 篇文章 0 订阅

订阅专栏

好久没写数位dp了，学习了一种新的数位dp的写法，没有以前的复杂，不需要分dp和计算两步了，而且少了一位状态，快了很多。

题解可以参照：http://timeplayer.blog.163.com/blog/static/203718254201521485450999/

因为考虑到直接dp不可行，我们先枚举数字之和，共有9*18种，f[i][j][k][2]表示长度为i数字之和为j，模当前枚举的数字之和为k的是否严格小于该数的种类数。

那么f[i][j][k]-->f[i+1][j+p][(k*10+p)%mod]大概就是这样，因为取模的时候模0了，所以WA了一发。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>

using namespace std;

long long f[21][200][200][2];
int n;
int sum=9*18+1;
long long a,b;
int dight[20];

long long calc(long long x,int mod)
{
	if (!x) return 0;
	memset(f,0,sizeof(f));
	int num=0;
	while (x)
	{
		dight[++num]=x%10;
		x/=10;
	}
	f[num+1][0][0][0]=1;
	for (int i=num+1;i>1;i--)
	  for (int j=0;j<=mod;j++)
	    for (int k=0;k<mod;k++)
	    {
	    	if (!f[i][j][k][0] && !f[i][j][k][1]) continue;
	    	for (int p=0;p<=9;p++)
	    	{
	    		if (p<dight[i-1]&&(j+p<=mod)) f[i-1][j+p][(10*k+p)%mod][1]+=f[i][j][k][0];
	    		else if (p==dight[i-1]&&(j+p<=mod)) f[i-1][j+p][(10*k+p)%mod][0]+=f[i][j][k][0];
	    		if (f[i][j][k][1]&&(j+p<=mod)) f[i-1][j+p][(10*k+p)%mod][1]+=f[i][j][k][1];
	    	}
	    }
	return f[1][mod][0][0]+f[1][mod][0][1];
}

int main()
{
	scanf("%lld%lld",&a,&b);
	long long ans=0;
	for (int i=1;i<sum;i++)
	  ans+=calc(b,i)-calc(a-1,i);
	printf("%lld\n",ans);
	return 0;
}