1026-windy数+数位DP+记忆化搜索

1026: [SCOI2009]windy数

题意：数位DP模板题；

目前只理解了记忆化搜索，就想练练手，

------给递推写法留一个位子

------

　　注意这道题要判断前导0的情况，1 ）可以加一个bool lead，或者在（i==0&&pre==-10)特判

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdio>
using namespace std;
typedef long long ll;
ll dp[][],digit[];

ll dfs(int pos,int pre,bool limit,bool lead)//lead是判断前导0的，如果（i==0&&pre==-10）需要在dfs中继续pre=-10，避免满足记忆化的条件
{
    if(pos==)return ;
    if(!limit&&dp[pos][pre]>=&&!lead)return dp[pos][pre];
    int num = limit?digit[pos]:;
    ll ans = ;
    for(int i=;i<=num;i++)
    {
        if(abs(i-pre)<)continue;
        if(lead&&i==)
            ans += dfs(pos-,pre,limit&&i==digit[pos],true);
        else ans+= dfs(pos-,i,limit&&i==digit[pos],false);
    }
    if(!lead && !limit)return dp[pos][pre] = ans;
    else return ans;
}

ll solve(ll x)
{
    int len = ;
    while(x>)
    {
        digit[++len] = x%;
        x/=;
    }
    return dfs(len,-,true,true);
}
int main(){
    ll n,m;
    memset(dp,-,sizeof(dp));
    while(~scanf("%lld%lld",&n,&m))
    {
        printf("%lld\n",solve(m)-solve(n-));
    }
}

贴一张特判的，感觉这个目的性和思路比较清晰；

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

int a, b,num[],dp[][];

int dfs(int len, int last, bool shangxian)
{
    int p;
    if (len <= )
        return ;
    if (!shangxian && dp[len][last] != -&& last >= )
        return dp[len][last];
    int cnt = , maxx = (shangxian ? num[len] : );
    for (int i = ; i <= maxx; i++)
    {
        if (abs(i - last) < )
            continue;
        p = i;
        if (i ==  && last == -)
            p = last;
        cnt += dfs(len - , p, shangxian && (i == maxx));
    }
    //return cnt;
    if (last >=  && !shangxian)
        dp[len][last] = cnt;
    return cnt;
}

int solve(int x)
{
    int k = ;
    while (x)
    {
        num[++k] = x % ;
        x /= ;
    }
    memset(dp, , sizeof(dp));
    return dfs(k, -, true);
}

int main()
{
    scanf("%d%d", &a, &b);
    printf("%d\n", solve(b) - solve(a - ));

    return ;
}