题意:
给定一个正整数 n,找出小于或等于 n 的非负整数中,其二进制表示不包含 连续的1 的个数。
思路:
一开始先写了一下dfs+剪枝,勉强可以通过,但是耗时上只打败了7%
所以下面只写dp思路
先把官方的图贴过来
我们以高度为2的一层中的两个0来举例,即它们的高度相同,节点的值也相同,均为1,并且以它们为根结点的两棵子树都是满二叉树,那么它们包含的无连续 1 的从根结点到叶结点的路径个数是相同的。
此外,在高度为 t、根结点为 0 的满二叉树中:其左孩子是高度为 t−1、根结点为 0 的满二叉树。其右孩子是高度为 t−1、根结点为 1 的满二叉树;但是因为路径中不能有连续 1,所以右子结点下只有其左子结点包含的从根结点到叶结点的路径才符合要求,而其左子结点是高度为 t−2、根结点为 0 的满二叉树。
dp[i]表示高度为i 根节点值为0的满二叉树的符合题意路径的个数
因此有状态转移方程:
dp[0]=dp[1]=1
dp[i]=dp[i-1]+dp[i-2] i>=2
但是题目给出的n并不可能总是满二叉树,当给出的不是满二叉树时,我们就要递归的处理
对于某个子树,如果该子树有右子树,说明其左子树是满二叉树。而我们已经求出了满二叉树中的路径数量,直接考虑其右子树即可。同理,若该子树没有右子树,我们递归考虑其左子树即可。
public int findIntegers(int n) {
int dp[]=new int[35];
dp[0]=dp[1]=1;
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){//判断 当前子树是否有右子树,即第i层是否有1
ans+=dp[i+1];
if(pre==1){
break;
// 上一层为 1,之后要处理的右子树根节点肯定也为1,此时连续两个 1,不满足题意,直接退出
}
pre=1;//本层为1 加标记
}else{
pre=0;//本层不为1 清楚标记
}
if(i==0){
ans++;
}
}
return ans;
}
时间复杂度O(log(n)) 其中n为题目数据最大值 1e9
空间复杂度O(log(n)) 其中n为题目数据最大值 1e9