FZU2171 - 防守阵地 II 线段树的区间更新

最新推荐文章于 2019-09-27 19:58:10 发布

Merc_A

最新推荐文章于 2019-09-27 19:58:10 发布

阅读量506

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/Merc_A/article/details/38660061

版权

数据结构专栏收录该内容

3 篇文章 0 订阅

订阅专栏

很明显的线段树区间更新问题，第一次写，就tle了，开始翻以前发的资料，说是更新到最后一个节点肯定会超，要在结构体里多设一个变量来记录该节点往下的节点是否需要修改，这样就不会超了，更新就是要加上一个pushdown，研究了好久，最后发现int就过了，__int64 就wa了，郁闷。

#include<stdio.h>
#include<string.h>

struct s
{
    int l,r;
    int ans,mark;
};
s a[500000];
int b[500000];
void pushdown(int rt)
{
     if(a[rt].mark!=0)
        {
            a[2*rt].mark+=a[rt].mark;
            a[2*rt].ans+=a[rt].mark*(a[2*rt].r-a[2*rt].l+1);
            a[2*rt+1].mark+=a[rt].mark;
            a[2*rt+1].ans+=a[rt].mark*(a[2*rt+1].r-a[2*rt+1].l+1);
            a[rt].mark=0;
        }

}
void build (int left,int right,int key)
{
    int mid=(left+right)/2;
    a[key].mark=0;
    a[key].l=left;
    a[key].r=right;
    if(left==right)
    {
        a[key].ans=b[right];
        return ;
    }
    //  {
    build (left,mid,2*key);
    build (mid+1,right,2*key+1);
    //  }
    a[key].ans=a[2*key].ans+a[2*key+1].ans;
    return ;
}
int ques(int left,int right,int rt)
{
    int temp1,temp2;
    int mid=(a[rt].l+a[rt].r)/2;
    if(a[rt].l==left&&a[rt].r==right) return a[rt].ans;
  //  printf("heh=%I64d?\n",a[rt].mark);
       pushdown(rt);

        if(right<=mid) return ques(left,right,2*rt);
        else if(left>mid) return ques(left,right,2*rt+1);
        else
        {
            temp1=ques(left,mid,2*rt);
            temp2=ques(mid+1,right,2*rt+1);
            return temp1+temp2;
        }
}

void edit(int xx,int yy,int rt,int hehe)
{
    if(a[rt].l>=xx&&a[rt].r<=yy)
    {
        a[rt].mark+=hehe;
        a[rt].ans+=hehe*(a[rt].r-a[rt].l+1);
        return ;
    }
    pushdown(rt);
        int mid=(a[rt].l+a[rt].r)/2;
        if(yy<=mid) edit(xx,yy,rt*2,hehe);
        else if(xx>mid) edit(xx,yy,rt*2+1,hehe);
        else
        {
            edit(xx,mid,rt*2,hehe);
            edit(mid+1,yy,rt*2+1,hehe);
        }
    a[rt].ans=a[rt*2].ans+a[rt*2+1].ans;
}
int main()
{
    int x,n,m,q,i;
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        for(i=1; i<=n; i++)
        {
            scanf("%d",&b[i]);
        }
        build (1,n,1);
       // printf("mm%I64d\n",m);
        while(q--)
        {
            scanf("%d",&x);
            printf("%d\n",ques(x,x+m-1,1));
            edit(x,x+m-1,1,-1);
            // printf(" xinde =? %I64d\n",ques(x,x+m-1,1));
        }
    }
    return 0;
}