（nice!!!）LeetCode 600. 不含连续1的非负整数（动态规划、数位dp）

题目：600. 不含连续1的非负整数

思路：这是典型的数位DP题，但本题每一位上的值只有0、1可选，要简单些。
细节看注释。

class Solution {
public:
	//状态记忆化，sta[u][pre]表示：在不受限制的情况下(符合要求的数不会超过n)，在第u位上选0或1，pre是第u+1位是选1还是0。该状态下符合要求的个数
    int sta[35][5];
    //u:二进制第u位上该选0还是1、pre:上一位选的是否是1、isLimit:当前u位上选的值是否受到不超过n的限制
    int dfs(int u,bool pre,bool isLimit,int n){
    	//当处理到的位置为-1时，说明前面的选取都没有连续的1，当前二进制的选择就是成功的，返回1即可
        if(u<0) return 1;
        //后续位选0、1操作都不会造成数值超过n，且当前的状态已经出现过
        if(isLimit==false&&sta[u][pre]>=0) return sta[u][pre];
        //判断当前u位上可选取的最大值
        //如果受到限制，那就看n的第u位上的取值情况，如果不搜限制，那就是0、1都可以选
        int up=isLimit? n>>u&1:1;
        //统计可以形成的符合要求的个数
        int res=0;
        //第u位选取0时
        res+=dfs(u-1,false,isLimit&&up==0,n);
        //如果u的上一位不是1，且第u位可选的值可以是1时
        if(pre==false&&up==1)
            res+=dfs(u-1,true,isLimit&&up==1,n);
        //如果当前状态为不受限制，那就可以保存当前的状态
        //因为如果当前状态受到限制，只会遍历到该位置一次，没有村存的必要
        if(isLimit==false) sta[u][pre]=res;
        return res;
    }
    int findIntegers(int n) {
        int u;
        //把记忆化状态初始化
        memset(sta,-1,sizeof sta);
        //找到n的二进制最高位
        for(int i=31;i>=0;i--){
            if(n>>i&1){
                u=i;
                break;
            }
        }
        //u:二进制第u位上该选0还是1、pre:上一位选的是否是1、isLimit:当前u位上选的值是否受到不超过n的限制
        return dfs(u,false,true,n);
    }
};