hdu4217 Data Structure? 线段树

最新推荐文章于 2018-07-29 23:59:58 发布

凉拌灰土

最新推荐文章于 2018-07-29 23:59:58 发布

阅读量771

点赞数

分类专栏：线段树文章标签： hdu4217 Data Structure 线段树 hdu

本文链接：https://blog.csdn.net/cnh294141800/article/details/20214961

版权

线段树专栏收录该内容

4 篇文章 0 订阅

订阅专栏

题意：不解释了。

看了别人的博客才知道要这么用线段树，用的灵活。

线段树的节点s[k].n记录剩下节点的数目。当查找到s[k].l=s[k].r即叶子节点的时候，就取这个值，然后将s[k].n=0说明这个点已经被拿了。

应该说来是线段树 + 二分查找。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int n,k;
const int N=262144;
struct node
{
    int l, r;
    int n;
}s[N*4+1];

void bulid(int l,int r,int k)
{
    if(l==r){
        s[k].l=l;
        s[k].r=r;
        s[k].n=1;
        return ;
    }
    s[k].l=l;
    s[k].r=r;
    s[k].n=0;
    int mid=(l+r)>>1;
    bulid(l,mid,k<<1);
    bulid(mid+1,r,k<<1|1);
    s[k].n=s[2*k].n+s[2*k+1].n;
}
int query(int k,int a)
{
    if(s[k].l==s[k].r){
        s[k].n=0;
        return s[k].l;
    }
    int ans;
    if(a<=s[k<<1].n) ans=query(k<<1,a);
    else if(a>s[k<<1].n) ans=query(k<<1|1,a-s[k<<1].n);
    s[k].n=s[2*k].n+s[2*k+1].n;
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int w=1;w<=t;w++){
        scanf("%d%d",&n,&k);
        bulid(1,n,1);
        __int64  ans=0;
        for(int i=1;i<=k;i++){
            int a;
            scanf("%d",&a);
            ans+=query(1,a);
        }
        printf("Case %d: ",w);
        printf("%I64d\n",ans);
    }
}