leetcode 600. 不含连续1的非负整数

非常明显的数位dp。

有两种状态表示的方法。

1。dp[pos][pre[sta] 表示到第pos位，前一位为pre，当前状态为sta的个数。

class Solution {
public:
    int dp[40][2][2];
    int bit[40];
    int dfs(int pos,int pre,int sta,bool limit)
    {
        if(pos==-1) return sta;
        if(limit==0&&dp[pos][pre][sta]!=-1) return dp[pos][pre][sta];
        int up=limit?bit[pos]:1;
        int ans=0;
        for(int i=0;i<=up;i++)
        {
            if(pre==1&&i==1)
            {
                ans+=dfs(pos-1,i,0,limit&&i==up);
            }
            else
            {
                ans+=dfs(pos-1,i,sta&&1,limit&&i==up);
            }
        }
        if(limit==0) dp[pos][pre][sta]=ans;
        return ans;
    }
    int solve(int x)
    {
        int pos=0;
        while(x)
        {
            bit[pos++]=x%2;
            x/=2;
        }
        return dfs(pos-1,0,1,1);
    }
    int findIntegers(int num) {
        memset(dp,-1,sizeof(dp));
        return solve(num);
    }
};

2. 去掉sta。直接dp[pos][pre]也行。不过写法有点变化。

class Solution {
public:
    int dp[40][2];
    int bit[40];
    int dfs(int pos,int pre,bool limit)
    {
        if(pos==-1) return 1;
        if(limit==0&&dp[pos][pre]!=-1) return dp[pos][pre];
        int up=limit?bit[pos]:1;
        int ans=0;
        for(int i=0;i<=up;i++)
        {
            if(pre==1&&i==1)
            {
                //ans+=dfs(pos-1,i,limit&&i==up);
            }
            else
            {
                ans+=dfs(pos-1,i,limit&&i==up);
            }
        }
        if(limit==0) dp[pos][pre]=ans;
        return ans;
    }
    int solve(int x)
    {
        int pos=0;
        while(x)
        {
            bit[pos++]=x%2;
            x/=2;
        }
        return dfs(pos-1,0,1);
    }
    int findIntegers(int num) {
        memset(dp,-1,sizeof(dp));
        return solve(num);
    }
};