BZOJ1026: [SCOI2009]windy数

Portal

f[i][j]表示i位，最高位为j的windy数个数
统计答案用前缀和的思想，sum[B]-sum[A-1]
然后就是暴力了

【代码】

#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 100005
using namespace std;
typedef long long ll;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m;    
int f[11][10],p[11];

void Input_Init()
{
    n=read(),m=read();p[1]=1;
    for(register int i=2;i<=10;i++) p[i]=p[i-1]*10;
    for(register int i=0;i<=9;i++) f[1][i]=1;
    for(register int i=2;i<=10;i++)
    for(register int j=0;j<10;j++)
    for(register int k=0;k<10;k++)
        if(abs(j-k)>=2) f[i][j]+=f[i-1][k];
}

int Query(int x)
{
    if(!x)return 0;
    int rtn=0,w=10;
    while(p[w]>x)w--;
    for(int i=1;i<w;i++)
        for(int j=1;j<=9;j++)
            rtn+=f[i][j];
    int cur=x/p[w];
    for(int i=1;i<cur;i++) rtn+=f[w][i];
    x%=p[w];
    int pre=cur;
    for(int i=w-1;i;i--)
    {
        cur=x/p[i];
        for(int j=0;j<cur;j++)
            if(abs(pre-j)>=2) rtn+=f[i][j];
        if(i==1&&abs(cur-pre)>=2) rtn+=f[1][cur];
        if(abs(cur-pre)<2)break;
        pre=cur;
        x%=p[i];
    }
    return rtn;
}

int main()
{
    Input_Init();
    printf("%d\n",Query(m)-Query(n-1));
    return 0;
}