LeetCode 600 不含连续1的非负整数[字典数 动态规划] HERODING的LeetCode之路

Posted 2021-09-27 HERODING23

解题思路：
解题关键分为两步，第一步利用字典树的方法统计高度为 t + 1、根结点为 0 的满二叉树中，不包含连续 1 的从根结点到叶结点的路径数量。第二步回归n本身，寻找其在字典树中的位置，然后不断记录其不包含连续 1 的从根结点到叶结点的路径数量，代码如下：

class Solution {
public:
    int findIntegers(int n) {
        vector<int> dp(31);
        dp[0] = dp[1] = 1;
        for (int i = 2; i < 31; ++i) {
            dp[i] = dp[i - 1] + dp[i - 2];
        }

        int pre = 0, res = 0;
        for (int i = 29; i >= 0; --i) {
            int val = 1 << i;
            if ((n & val) != 0) {
                n -= val;
                res += dp[i + 1];
                if (pre == 1) {
                    break;
                }
                pre = 1;
            } else {
                pre = 0;
            }

            if (i == 0) {
                ++res;
            }
        }

        return res;
    }
};