【洛谷P2657】Windy数【数位DP】

分析

显然数位DP，转化为只有上界的形式。

发现这道题限制还是比较少，只有差值。于是想怎么设置阶段和状态？如果第二维表示差值的话就不好枚举所有数，所以第二维设成最高位的数字，这样子试填的时候枚举数字，就可以顺手得到差值了。

设 $f[pos][j]$ 为填到第pos位，最高位数字为 $j$ 的数字总数。转移：$$dp[i][j]=\Sigma dp[i-1][k]，abs(j-k)>2$$

可以先把所有范围内的都处理出来先，然后每次统计答案就行。

但其实统计答案这里也是关键，需要保证这个数字不大于范围内，在最后一位的时候要枚举所有位，让所有位都不大于上界。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
//设dp[i][j]为长度为i中最高位是j的windy数的个数
//dp[i][j]=sum(dp[i-1][k]) 其中 abs(j-k)>=2 
int p,q,dp[15][15],a[15];
void init()
{
	for(int i=0;i<=9;i++) dp[1][i]=1;
	for(int i=2;i<=10;i++)
	{
		for(int j=0;j<=9;j++)
		{
			for(int k=0;k<=9;k++)
			{
				if(abs(j-k)>=2)	dp[i][j]+=dp[i-1][k]; 
			}
		}
	}
}
int work(int x)	
{
	memset(a,0,sizeof(a));
	int len=0,ans=0;
	while(x)
	{
		a[++len]=x%10;
		x/=10;
	}

	for(int i=1;i<=len-1;i++)
	{
		for(int j=1;j<=9;j++)
		{
			ans+=dp[i][j];
		} 
	}
	//len位
	for(int i=1;i<a[len];i++)
	{
		ans+=dp[len][i];
	} 
	//最高位与原数相同
	for(int i=len-1;i>=1;i--)
    {
        for(int j=0;j<=a[i]-1;j++)
		{
		   	if(abs(j-a[i+1])>=2) ans+=dp[i][j];	 
		} 
		if(abs(a[i+1]-a[i])<2) break;

    }
	return ans;
}
int main()
{
	init();
	cin>>p>>q;
	cout<<work(q+1)-work(p);
	return 0;
}