1026: [SCOI2009]windy数
Time Limit: 1 Sec Memory Limit: 162 MB
Description
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,
在A和B之间,包括A和B,总共有多少个windy数?
Input
包含两个整数,A B。
Output
一个整数
Sample Input
1 10
25 50
Sample Output
9 20
HINT
100%的数据,满足 1 <= A <= B <= 2000000000 。
【题目链接】 link
【题意】
RT
【思路】
用dp[i][j]表示长度为i,最高位为j的windy数个数,可以用暴力求得。
然后就是不断地分解输入的这个数,累加答案即可。
唯一值得注意的是,solve函数求出的是[1,x-1]内windy数的个数,因为每次都是从比当前位小的数去统计,没有考虑x是否为windy数。
具体细节见代码。
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
#define mst(a,b) memset((a),(b),sizeof(a))
#define rush() int T;scanf("%d",&T);while(T--)
typedef long long ll;
const int maxn = 15;
const ll mod = 1e9+7;
const ll INF = 1e18;
const double eps = 1e-6;
int digit[maxn];
ll dp[maxn][maxn];
void init()
{
mst(dp,0);
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];
}
}
ll solve(ll x)
{
ll ans=0;
if(x==0) return 0;
mst(digit,0);
int len=0;
while(x)
{
digit[++len]=x%10;
x/=10;
}
for(int i=1;i<=digit[len]-1;i++) ans+=dp[len][i]; //长度相等,最高位小于
for(int i=1;i<=len-1;i++) //长度小于
for(int j=1;j<=9;j++)
{
ans+=dp[i][j];
}
for(int i=len-1;i>0;i--) //前面固定后,枚举每一位
{
for(int j=0;j<=digit[i]-1;j++)
{
if(abs(digit[i+1]-j)>=2) ans+=dp[i][j];
}
if(abs(digit[i+1]-digit[i])<2) break;
}
return ans;
}
int main()
{
ll a,b;
scanf("%lld%lld",&a,&b);
init();
printf("%lld\n",solve(b+1)-solve(a));
}