UESTC 250 windy数(数位DP)

时间:2021-07-28 09:12:22

题意:题意:求区间[A,B]之间的,不含前导0,且相邻两数位之间相差至少为2的正整数有多少个.

分析:dp[i][j]表示,长度为i、以j为结尾的表示的个数,再加一个前导0判断即可

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
const ll INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod = ;
int dp[][],a,b;
int bit[],len;
//f表示前导0是否有效
int dfs(int i,int j,int f,int e){
if(i==)
{
if(f)return ;
else return ;
}
if(!f&&!e&&dp[i][j]!=-)return dp[i][j];
int l=e?bit[i]:;
int num=;
for(int v=;v<=l;++v){
if(f){
if(v==)
num+=dfs(i-,v,,e&&(v==l));
else
num+=dfs(i-,v,,e&&(v==l));
}
else if(abs(j-v)>=)
num+=dfs(i-,v,,e&&(v==l));
}
if(!f&&!e)dp[i][j]=num;
return num;
}
int solve(ll x){
len=;
while(x){
bit[++len]=x%;
x/=;
}
return dfs(len,,,);
}
int main()
{
memset(dp,-,sizeof(dp));
scanf("%d%d",&a,&b);
printf("%d\n",solve(b)-solve(a-));
return ;
}