A - 敌兵布阵 (HDU - 1166)

Posted 2020-10-26 Alpacaddhh

- 题目大意

    中文题，一共有4种命令，然后来根据命令来进行操作。

- 解题思路

    典型的线段树的问题，进行点的更新，然后进行区间的查询（查询的时候查的是这个区间上的点的和）。

- 代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=50001;
int ans;
struct node
{
    int l,r,sum;
}t[4*maxn];

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

void add(int n,int d,int k)
{
    int mid;
    if(t[k].l==t[k].r&&t[k].l==d)
    {
        t[k].sum+=n;
        return;
    }
    mid=(t[k].r+t[k].l)/2;
    if(d<=mid)
        add(n,d,2*k);
    else
        add(n,d,2*k+1);
    t[k].sum=t[2*k].sum+t[2*k+1].sum;
}

void findd(int l,int r,int k)
{
    int mid;
    if(t[k].l==l&&t[k].r==r)
    {
        ans+=t[k].sum;
        return;
    }
      mid=(t[k].r+t[k].l)/2;
    if(r<=mid)
        findd(l,r,2*k);
    else if(l>mid)
        findd(l,r,2*k+1);
    else
        {
            findd(l,mid,2*k);
            findd(mid+1,r,2*k+1);
        }

}

char str[20];
int main()
{
    int n,x,y;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        build(1,x,1);
        for(int j=1;j<=x;j++)
        {
            scanf("%d",&y);
            add(y,j,1);
        }
        printf("Case %d:\n",i);
        while(scanf("%s",str),strcmp(str,"End"))
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(strcmp(str,"Add")==0)
                add(b,a,1);
            else if(strcmp(str,"Sub")==0)
                add(-b,a,1);
            else
            {
                ans=0;
                findd(a,b,1);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}