hdu3652（数位dp）

时间：2022-10-18 03:23:38

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=3652

题意：求1~n含有13且能被13整除的数的个数。

分析：数位dp,dp数组加一维来维护到pos位模13的余数，则dp[pos][mod][2]表示非限制条件下到pos位模13余mod且已含有13的总个数，dp[pos][mod][1]表示没含有13但前一位是1且模13余mod的总个数，dp[pos][mod][0]表示没含有13前一位不为1模13余mod的总个数。。。

#include <cstdio>

#include <cstring>

#include <string>

#include <cmath>

#include <iostream>

#include <algorithm>

#include <queue>

#include <cstdlib>

#include <stack>

#include <vector>

#include <set>

#include <map>

#define LL long long

#define mod 10007

#define inf 0x3f3f3f3f

#define N 100010

#define FILL(a,b) (memset(a,b,sizeof(a)))

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

using namespace std;

int dp[][][];

int dig[];

int dfs(int pos,int pre,int md,int flag,int limit)

{

    if(!pos)return flag&&md==;

    if(!limit&&flag&&~dp[pos][md][])return dp[pos][md][];

    if(!limit&&!flag&&pre==&&~dp[pos][md][])return dp[pos][md][];

    if(!limit&&!flag&&pre!=&&~dp[pos][md][])return dp[pos][md][];

    int len=limit?dig[pos]:;

    int ans=;

    for(int i=;i<=len;i++)

    {

        ans+=dfs(pos-,i,(md*+i)%,flag||(pre==&&i==),limit&&i==len);

    }

    if(!limit)

    {

        if(flag)dp[pos][md][]=ans;

        else if(pre==)dp[pos][md][]=ans;

        else dp[pos][md][]=ans;

    }

    return ans;

}

int solve(int x)

{

    int len=;

    while(x)

    {

        dig[++len]=x%;

        x/=;

    }

    return dfs(len,,,,);;

}

int main()

{

    int n;

    while(scanf("%d",&n)>)

    {

        memset(dp,-,sizeof(dp));

        printf("%d\n",solve(n));

    }

}

相关文章