HDU 3652 B-number 数位DP

题目大意：给出一个数n，求1~n中满足：即含有13（连续），有能被13整除的数的个数.

题目思路：数位DP，一开始读错题意了，以为13可以不连续……，比较水的数位DP。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<stdlib.h>
#include<queue>
#include<math.h>
#include<map>
#define INF 0x3f3f3f3f
#define MAX 1000005
#define Temp 1000000000

using namespace std;

int num[20],dp[25][25][25];

//st==0 前一位不为1
//st==1 前一位为1
//st==2 已含13

long long dfs(int pos,int st,int limit,long long k)
{
    if(pos<0)
        return st==2&&k==0;

    if(!limit && dp[pos][st][k]!=-1)
        return dp[pos][st][k];
    int len=limit?num[pos]:9;
    long long ans=0;
    for(int i=0;i<=len;i++)
    {
        long long temp=(k*10+i)%13;
        if(st==0 && i==1)
            ans+=dfs(pos-1,1,limit&&i==len,temp);
        else if(st==0 && i!=1)
            ans+=dfs(pos-1,0,limit&&i==len,temp);
        else if(st==1 && i==3)
            ans+=dfs(pos-1,2,limit&&i==len,temp);
        else if(st==1 && i==1)
            ans+=dfs(pos-1,1,limit&&i==len,temp);
        else if(st==1)
            ans+=dfs(pos-1,0,limit&&i==len,temp);
        else
            ans+=dfs(pos-1,2,limit&&i==len,temp);
    }
    if(!limit)
        dp[pos][st][k]=ans;
    return ans;
}

long long solve(long long a)
{
    memset(dp,-1,sizeof(dp));
    int len=0;
    while(a)
    {
        num[len++]=a%10;
        a/=10;
    }
    long long ans=dfs(len-1,0,1,0);
    return ans;
}

int main()
{
    long long n;
    while(scanf("%lld",&n)!=EOF)
    {
        long long ans=solve(n);
        printf("%lld
",ans);
    }
    return 0;
}