hdu4217

/*
分析:
    线段树。
    代码不长,思路都在那个find子函数里面,注意要用64位
整数,至于32位的会不会WA,我没试,也不准备试-、-I,其它
的就没啥用解释的了,看不明白了再留言吧。


                                                2012-10-20
*/












#include"stdio.h"
#include"string.h"
#include"stdlib.h"
struct Seg
{
    int l,r,mid;
    int sum;
}T[1111111];
void build(int l,int r,int k)
{
    T[k].l=l;
    T[k].r=r;
    T[k].mid=(l+r)>>1;
    if(l==r)    {T[k].sum=1;return ;}
    build(l,T[k].mid,2*k);
    build(T[k].mid+1,r,2*k+1);
    T[k].sum=T[2*k].sum+T[2*k+1].sum;
}
int find(int tot,int k)
{
    int ans;
    if(T[k].l==T[k].r)  {T[k].sum=0;return T[k].l;}
    if(tot<=T[2*k].sum) ans=find(tot,2*k);
    else                ans=find(tot-T[2*k].sum,2*k+1);
    T[k].sum=T[2*k].sum+T[2*k+1].sum;
    return ans;
}
int main()
{
    int T,Case;
    int n,m;
    int t,temp;
    __int64 ans;
    scanf("%d",&T);
    for(Case=1;Case<=T;Case++)
    {
        scanf("%d%d",&n,&m);
        build(1,n,1);
        ans=0;
        while(m--)
        {
            scanf("%d",&t);
            ans+=find(t,1);
        }
        printf("Case %d: %I64d\n",Case,ans);
    }
    return 0;
}


你可能感兴趣的:(hdu4217)