InTheBloodHorse

每日一题(13) BZOJ1026

字数统计: 353阅读时长: 1 min
2018/09/24 Share

题目地址
题意:
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,
在A和B之间,包括A和B,总共有多少个windy数?
思路:
数位dp,但是前导0让人很不舒服
dp[pos][num]表示位数为pos,上一位为num,满足条件的个数。
前导0的时候,要注意,这里传入-2,因为1,2,3,等一位数是满足条件的,所以他们与-2的差值都满足条件,所以可以dfs下去,不然就直接GG了。原本用sta来记录是否全为0,但是结果不理想,所以用-2来代表全是前导0。
AC代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#include<bits/stdc++.h>
using namespace std;
int a[20];
int dp[20][20];
long long dfs(int pos,int num,bool lim){
if(pos<0) return 1;
if(num >= 0 && !lim && dp[pos][num]!=-1) return dp[pos][num];
int up = lim ? a[pos]:9;
int ans=0;
for(int i=0;i<=up;i++){
if(abs(num-i)<=1) continue;
int pre = i;
if(i==0 && num==-2) pre=-2;
ans += dfs(pos-1,pre,lim && i==a[pos]);
}
if(num>=0 && !lim) dp[pos][num]=ans;
return ans;
}
long long solve(long long x){
int pos=0;
while(x){
a[pos++]=x%10;
x/=10;
}
return dfs(pos-1,-2,true);
}
int main()
{
long long x,y;
scanf("%lld %lld",&x,&y);
memset(dp,-1,sizeof(dp));
printf("%lld\n",solve(y)-solve(x-1));
}

原文作者:InTheBloodHorse

原文链接:http://pyking.cn/2018/09/24/每日一题-13-BZOJ1026/

发表日期:September 24th 2018, 7:27:22 pm

更新日期:September 24th 2018, 7:32:52 pm

版权声明:Have a fun

CATALOG