2.3 01序列

算法设计与分析 2.3 01序列

题目描述

给定一个序列，刚开始只有一个数。

序列里大于的数会进行分裂，即在序列中删除，并在原来的位置插入 X/2，X%2 ，X/2 三个数，直到序列中只有0和1。

问最后序列中第L个数到第R个数之间有多少个1。

输入格式

一行三个空格隔开的正整数。

对于50%的数据，0 <= N <= 100。

对于100%的数据，0 <=N <= 10e9，1 <= L <= R，0 <= R-L <= 10e5。

数据保证R不大于最后序列的长度。

输出格式

一个整数，最后序列中第个数到第个数之间有多少个1。

样例输入

4 3 6

样例输出

2

提示

样例中，序列变化如下：4 ➡ 2 0 2 ➡ 1 0 1 0 1 0 1

第3个数到第6个数之间有2个1。

参考代码

#include <iostream>
#include <cmath>
/*
递归
中序遍历三叉树
定位L的位置——每进入一个子树，判断L是在左子树，中子树还是右子树；
从L开始中序遍历，同时记录1个个数
不遍历L的左边序列，也不遍历R的右边序列

序列个数 —— pow(2,int((log(N)/log(2)))+1)-1

T:O(R-L)
S:O(log N)?

*/
long L,R;

long long int midOrder(long n, long long int len, long long int mid) {
	if(mid+len/2<L||mid-len/2>R) return 0;
	if(len==1) return n%2;

	if(L<=mid&&mid<=R)
		return midOrder(n/2,len/2,mid-(len+1)/4)+n%2+midOrder(n/2,len/2,mid+(len+1)/4);
	else if(R<mid)
		return midOrder(n/2,len/2,mid-(len+1)/4);
	else
		return midOrder(n/2,len/2,mid+(len+1)/4);
}

int main(void) {
	long N;

	std::cin>>N>>L>>R;

	long long int len=pow(2,int((log(N)/log(2)))+1);

//	std::cout<<int((log(N)/log(2)))+1<<std::endl;
//	std::cout<<len<<std::endl;

	std::cout<<midOrder(N,len,len/2);
}