数位DP --Windy数

满足一下条件的数字称为windy数

不考虑前导0,所有相邻的两个数字的差至少为2。

求出任意区间内的所有windy数

思路:dp[i][j]表示所有长度为 i 的数字以 j 开头的windy数的个数。0<=j<=9

仔细考虑前导0 的影响,已注释在代码中

#include<iostream>
#include<cstring>
#include<cstdio>
#define bug
using namespace std;
const int MAXN = 20;
template<typename T>
T oabs(T a){return a<0?-a:a;}
int dp[MAXN][10];
//dp[i][j] 表示长度为i的所有数字以j开头的windy数字的个数
int a,b;
bool init()
{
	if(scanf("%d%d",&a,&b)==EOF)
		return false;
	return true;
}
void DP_init()
{
	memset(dp,0,sizeof(dp));
	int tar = 0;
	for(int i=0;i<10;i++)
		dp[1][i]=1;
	for(int i=2;i<MAXN;i++)
	{
		//如果前导为0,可以看做这一位不存在,加上后一位所有的数字开头的windy数个数
		for(int j=0;j<10;j++)dp[i][0]+=dp[i-1][j];
		for(int j=1;j<10;j++)
		{
			tar = j+2;
			if(tar<10)
			{
				for(int k=tar;k<10;k++)
					dp[i][j]+=dp[i-1][k];
			}
			tar=j-2;
			if(tar>=0)
			{
				for(int k=tar;k>=0;k--)
					{
						dp[i][j]+=dp[i-1][k];
						if(k==0)
						{
							//此时由于前一位不为0,所以此时的前导0不可以当做0这一位不存在,所以减去多加上两个数
							//表示这个trick找了好久,T_T
							dp[i][j]-=dp[i-2][0]+dp[i-2][1];
						}
					}
			}
		}
	}
}
int num[MAXN],len;
int solve(int n)
{
	len = 1;
	while(n)
	{
		num[len++]=n%10;
		n/=10;
	}
	num[len]=20;
	len--;
	int res =0;
	for(int i=len;i;i--)
	{
		for(int j =0;j<num[i];j++)
		{
			if(oabs(num[i+1]-j)<2)continue;
			res+=dp[i][j];
			if(j==0&&i!=len)
			{
				res-=dp[i-1][0]+dp[i-1][1];
			}
		}	

		if(oabs(num[i]-num[i+1])<2)break;
	}
	return res;
}
int main()
{
	DP_init();
	while(init())
	{
		printf("%d\n",solve(b+1)-solve(a));
	}
	return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值