C - Data Structure? HDU - 4217 线段树

很奇妙的线段树

#include<bits/stdc++.h>
using namespace std;
const int N = 262200;
struct node
{
    int l,r,s;
}t[3 * N];

int T,x,n,k;
long long ans;

void build(int x,int l,int r)
{
    t[x].l = l;
    t[x].r = r;
    t[x].s = r - l + 1;
    if(l == r) return;
    int mid = (l + r) / 2;
    build(2 * x,l,mid);
    build(2 * x + 1,mid + 1,r);
}

long long finds(int x,int num)
{
    t[x].s--;
    if(t[x].l == t[x].r) return t[x].l;
    if(t[2 * x].s >= num) return finds(2 * x,num);
    else return finds(2 * x + 1,num - t[2 * x].s);
}

int main()
{
    int tot = 0;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        build(1,1,n);
        ans = 0;
        for(int i = 0;i < k; ++i)
        {
            scanf("%d",&x);
            ans += finds(1,x);
        }
        printf("Case %d: %lld\n",++tot,ans);
    }
    return 0;
}