【Codeforces 768 B Code For 1】+ 简单线段树

B. Code For 1
time limit per test2 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
Jon fought bravely to rescue the wildlings who were attacked by the white-walkers at Hardhome. On his arrival, Sam tells him that he wants to go to Oldtown to train at the Citadel to become a maester, so he can return and take the deceased Aemon’s place as maester of Castle Black. Jon agrees to Sam’s proposal and Sam sets off his journey to the Citadel. However becoming a trainee at the Citadel is not a cakewalk and hence the maesters at the Citadel gave Sam a problem to test his eligibility.

Initially Sam has a list with a single element n. Then he has to perform certain operations on this list. In each operation Sam must remove any element x, such that x > 1, from the list and insert at the same position , , sequentially. He must continue with these operations until all the elements in the list are either 0 or 1.

Now the masters want the total number of 1s in the range l to r (1-indexed). Sam wants to become a maester but unfortunately he cannot solve this problem. Can you help Sam to pass the eligibility test?

Input
The first line contains three integers n, l, r (0 ≤ n < 250, 0 ≤ r - l ≤ 105, r ≥ 1, l ≥ 1) – initial element and the range l to r.

It is guaranteed that r is not greater than the length of the final list.

Output
Output the total number of 1s in the range l to r in the final sequence.

Examples
input
7 2 5
output
4
input
10 3 10
output
5
Note
Consider first example:

Elements on positions from 2-nd to 5-th in list is [1, 1, 1, 1]. The number of ones is 4.

For the second example:

Elements on positions from 3-rd to 10-th in list is [1, 1, 1, 0, 1, 0, 1, 0]. The number of ones is 5.

N 每次可分为 N / 2 , N % 2, N / 2 ? 线段树递归?

AC代码:

#include<cstdio>
typedef long long LL;
int q(LL N,LL L,LL R,LL l,LL r){
    if(N == 0 || L > r || R < l) return 0;
    if(N == 1) return 1;
    LL m = (l + r) / 2;
    return q(N / 2,L,R,l,m - 1) + q(N % 2,L,R,m,m) + q(N / 2,L,R,m + 1,r);
}
int main()
{
    LL N,L,R;
    scanf("%lld %lld %lld",&N,&L,&R);
    LL nl = 1,k = N;
    while(k > 1) nl = nl * 2 + 1,k /= 2;
    printf("%d\n",q(N,L,R,1,nl));
    return 0;
}
    原文作者:B树
    原文地址: https://blog.csdn.net/wyk1823376647/article/details/56853401
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞