LeetCode: 600. 不含连续1的非负整数（01字典树+dp）

给定一个正整数 n，找出小于或等于 n 的非负整数中，其二进制表示不包含 连续的1 的个数。

示例 1:

输入: 5
输出: 5
解释: 
下面是带有相应二进制表示的非负整数<= 5：
0 : 0
1 : 1
2 : 10
3 : 11
4 : 100
5 : 101
其中，只有整数3违反规则（有两个连续的1），其他5个满足规则。
说明: 1 <= n <= 109

分析：

       该题是一个01字典树+dp的问题，一开始除了暴力的方法想不到其他的，还是看了题解。不得不说，难度还是挺大的。由于小于等于n的非负整数的二进制表示存在许多的公共前缀，因此我们可以利用字典树来进行优化。可以发现组成的01字典树是一棵完全二叉树，如果某个节点有两个子节点，那么以左子节点为根的子树必是一棵满二叉树，以右子节点为根的子树必是一棵完全二叉树；如果某个节点只有一个子节点，那么以该子节点为根的子树必是一颗完全二叉树。

       那么该问题就转化为统计满二叉树和完全二叉树的不包含连续1的从根节点到叶节点的路径数量。对于满二叉树，我们可以如题解所说，采用dp的方法来计算；对于完全二叉树呢，我们只需要从根节点开始处理，如果当前节点包含两个节点，那我们把左子树（必是满二叉树）的路径数量计入答案，处理右子节点，如果只包含一个节点，那么继续处理该节点。

       通过该题学习到，如果遇到有公共前缀的问题，可以考虑通过字典树的方式来降低时间复杂度。同时，对于一棵完全二叉树，要注意上面提到的把完全二叉树进行拆分得到的性质。

class Solution {
public:
    int findIntegers(int n) {
        vector<int> dp(30);
        //dp[i]:表示高度为i，根节点为0的满二叉树中，不包含连续1的路径数量
        dp[0] = 1, dp[1] = 2;
        for(int i = 2; i < 30; i++)
            dp[i] = dp[i - 1] + dp[i - 2];
        int pre = 0, ans = 0;
        for(int i = 29; i >= 0; i--){
            int val = 1 << i;
            if((n & val) != 0){
                //n在该位置上为1，说明是两个子节点，其中左子节点必为满二叉树
                ans += dp[i];
                //如果父节点是1
                if(pre == 1)
                    break;
                //查询节点值为1的子节点
                pre = 1;
            }else{
                //查询节点值为0的子节点
                pre = 0;
            }
            if(i == 0){
                //加上该条路径
                ans++;
            }
        }
        return ans;
    }
};