[CodeForces - 768B] Code For 1

4536 단어 데이터 구조
링크: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 
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 
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;
}

좋은 웹페이지 즐겨찾기