第一次写博客= = 题目= =HDU1166

最新推荐文章于 2021-02-25 04:32:17 发布

路-过

最新推荐文章于 2021-02-25 04:32:17 发布

阅读量681

点赞数

本文链接：https://blog.csdn.net/u012896700/article/details/16943321

版权

刚开始学线段树。。。

第一次做的线段树题目。。。

题目意思大概就是

有N个营地。每个营地初始有多少人。

然后他会给你操作。在某个营地上加或减多少人。。。。

然后询问你I到J和营地共有多少人。。。。。

我的思路：

一开始建立线段树。。。

然后在输入的时候把包含该营地的所有区间都加上该营地的值。。。

最后就是让所有的区间就是该区间内所有营地内人数的总和。。。。

他给你操作的时候就和上面输入的时候一样操作= =

询问的时候。。

就是搜索。。。。一直搜索到满足的区间。然后都加起来= =

贴上代码。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
int m;
int a[50000];
struct tree{
  int left;
  int right;
  int renshu;

}tree[2000000];
void inset(int inst,int le,int ri)
{   tree[inst].left=le;
    tree[inst].right=ri;
    tree[inst].renshu=0;
   // cout<<le<<" "<<ri<<endl;
    if(le==ri)
    {
        return ;
    }
    int mid=(le+ri)>>1;
    inset(2*inst,le,mid);
    inset(2*inst+1,mid+1,ri);

}
void add(int inst,int k,int p)
{
    if(k>=tree[inst].left&&k<=tree[inst].right)
    {
        tree[inst].renshu+=p;
       // cout<<tree[inst].left<<" "<<tree[inst].right<<endl;
        if(tree[inst].left==tree[inst].right)
            return ;
        int mid=(tree[inst].left+tree[inst].right)>>1;
        if(mid>=k)
        add(2*inst,k,p);
        else
            add(2*inst+1,k,p);
    }

}
void Query(int inst,int zuo,int you)
{
    if(tree[inst].left==zuo&&tree[inst].right==you)
    {
             m+=tree[inst].renshu;
        return ;
    }
    int mid=(tree[inst].left+tree[inst].right)>>1;
    if(mid>=you)
       Query(2*inst,zuo,you);
    else if(mid<zuo)
        Query(2*inst+1,zuo,you);
    else
    {
      Query(2*inst,zuo,mid);
      Query(2*inst+1,mid+1,you);
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int i=1;i<=t;i++)
    {
        int n;
        scanf("%d",&n);
        inset(1,1,n);

        for(int j=0;j<n;j++)
        {
            scanf("%d",&a[j]);
            add(1,j+1,a[j]);
        }
        char z[10];
        printf("Case %d:\n",i);
        while(1)
        { scanf("%s",z);
            int t,p;
            if(strcmp(z,"End")==0)
                break;
            scanf("%d%d",&t,&p);
            if(strcmp(z,"Add")==0)
            {
                add(1,t,p);
                //cout<<1<<endl;
            }
            else if(strcmp(z,"Sub")==0)
            {   p=-p;
                add(1,t,p);
            }
            else if(strcmp(z,"Query")==0)
            {
                m=0;
                Query(1,t,p);
                printf("%d\n",m);
            }
        }
    }
}