【BNUOJ】 信息战(四)——战场演练(树状数组,查询第k大值)

本文介绍了一种利用树状数组和二分查找算法解决生命值问题的方法,通过代码实现和详细解释,展示了如何高效地查询和修改生命值,并找到特定生命值的人。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >




用一个普通数组p[i]保存第i个人的生命值,一个树状数组c[h]表示生命值为h的人有多少个,h的范围为1--3*65535;

此时,query(x)表示生命值不多于x的人的个数,用二分来找生命值第k的的人,用一个数 tol 保存当前存活的人数。



代码:

#include <cstdio>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <cctype>//tolower toupper isdigit isalpha islower isupper isspace
#include <algorithm>
#include <iostream>
#include <map>
#define LL long long
#define db double
#define pi acos(-1.0)
#define pr printf
#define sc scanf
#define mod
#define N 3*65535+5
using namespace std;

int c[N];
int p[30011];
int low(int i)
{
    return i&(-i);
}
void add(int i,int x)
{
    while(i<=N)
    {
        c[i]+=x;
        i+=low(i);
    }
}
int query(int i)
{
    int r=0;
    while(i)
    {
        r+=c[i];
        i-=low(i);
    }
    return r;
}

int main()
{
    int n,m,i,j,x,y,tol;
    char s[10];
    memset(c,0,sizeof(c));
    while(sc("%d",&n)+1&&n)
    {
        tol=n;
        for(i=0; i<n; ++i)
        {
            sc("%d",&x);
            p[i+1] = x ;
            add(x,1);
        }
        sc("%d",&m);
        for(i=0; i<m; ++i)
        {
            sc("%s",s);
            if(s[0]=='A')
            {

                sc("%d%d",&x,&y);
                add(p[x],-1);

                p[x]-=y;
                if(p[x]>0)
                    add(p[x],1);
                else
                    tol--;
            }
            else if(s[0]=='C')
            {
                sc("%d%d",&x,&y);
                add(p[x],-1);
                p[x]+=y;
                add(p[x],1);
            }
            else
            {

                sc("%d",&x);
                if(tol<x)
                {
                    puts("-1");
                    continue;
                }

                int l=0,r=N-5,m;
                while(l<=r)
                {
                    m=(l+r)>>1;
                    if(x<=tol-query(m))
                        l=m+1;
                    else
                        r=m-1;

                }
                pr("%d\n",l);

            }
        }
        pr("%d\n",tol);
    }

    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值