链接:http://codeforces.com/problemset/problem/768/B
Problem description
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.
Input
The first line contains three integers n, l, r (0 ≤ n < 2^50, 0 ≤ r - l ≤ 10^5, 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.
分析:第一眼看上去是一个递归,想直接暴力出答案,但是2^50的n写个递归暴力做一定会T了,因此想到线段树的查询操作,可以在logn的时间内查询区间,但是要使用查询操作就必须知道当前“节点”所代表的区间,因此需要写一个函数得到当前节点处理后得到01序列的长度来得到这个区间。好在可以在logn的时间得到一个数n的序列长度。因此再使用类似线段树的查询操作就可以在log^2(n)的时间复杂度下求得结果了。
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n, le, ri;
ll getlen(ll x)
{
return x > 1 ? 1 + 2 * getlen(x >> 1) : 1;
}
ll query(ll l, ll r, ll rt, ll L, ll R)
{
if (L <= l && r <= R)
return rt;
ll m = l + getlen(rt >> 1);
ll ans = 0;
if (m - 1 >= L)
ans += query(l, m - 1, rt >> 1, L, R);
if (L <= m && m <= R)
ans += rt & 1;
if (m + 1 <= R)
ans += query(m + 1, r, rt >> 1, L, R);
return ans;
}
int main()
{
cin >> n >> le >> ri;
ll len = getlen(n);
cout << query(1, len, n, le, ri) << endl;
return 0;
}