nyoj1122-士兵杀敌(四)

来源:互联网 发布:矿山工程预算软件 编辑:程序博客网 时间:2024/06/05 15:21

题目来源:http://acm.nyist.melem.php?pid=1122

题意

中问题意。。。

思路

貌似是延迟标记,,,自己手打,,,并没有什么模板。。。
解释下延迟标记,如果单点更新会的话,就会知道一个节点更新需要O(logn)的时间,如果是一个区间更新,如果还是每次每个节点都更新,最坏的时间复杂度是O(n*nlogn),那么就会不那么理想。
这里,好比样例1,我要更新1-3,那么我开始遍历,找到能被他包围的区间,1-10变成1-5变成1-3,这个时候,刚好,然后我就把1-3所在的节点的sum域加上我要加的值。然后,停,。。。。(果断相当干脆)
接下来我要输出节点3的值,那么依旧是1-10变成1-5变成1-3,然后,但是这个时候会发现1-3的sum域有值,所以我就1-3的子区间进行加上sum,但是如果子区间是一个叶节点(别问我叶节点是啥,就是没有分支的点),那么sum域的值无关紧要,只要把他的val域加上sum就好,然后输出,,,当然,输出之前,别忘了把1-3的节点的sum域清为0,因为已经向下传递,。。。

代码

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;const int maxn=1000000;struct pp{    int sum;    int val;} tree[(maxn<<2)+10];int q,n;void build(int l,int r,int rt){    if(l==r)    {        tree[rt].sum=tree[rt].val=0;    }    else    {        tree[rt].sum=0;        int m=(l+r)/2;        build(l,m,rt<<1);        build(m+1,r,(rt<<1)+1);    }}void update(int l,int r,int L,int R,int tmp,int rt){    if(L<=l&&r<=R)    {        if(l!=r) tree[rt].sum+=tmp;        tree[rt].val+=tmp;        return ;    }    int m=(l+r)/2;    if(m>=L)    {        update(l,m,L,R,tmp,rt<<1);    }    if(m+1<=R)    {        update(m+1,r,L,R,tmp,(rt<<1)+1);    }}int query(int l,int r,int x,int rt){    if(l==r)    {//        printf("I will accept!!!\n");        return tree[rt].val;    }    int m=(l+r)/2,res=0;    if(tree[rt].sum!=0)    {//        printf("I will down down!!!\n");        if(l==m)            tree[rt<<1].val+=tree[rt].sum;        else            tree[rt<<1].sum+=tree[rt].sum;        if(m+1==r)            tree[(rt<<1)+1].val+=tree[rt].sum;        else            tree[(rt<<1)+1].sum+=tree[rt].sum;        tree[rt].sum=0;//        printf("I give node %d and node %d,ranges are %d %d and %d %d\n",rt<<1,(rt<<1)+1,l,(l+r)/2,(l+r)/2+1,r);    }    if(m>=x)    {//        printf("come in!!!\n");        res=query(l,m,x,rt<<1);//        printf("come out!!!\n");    }    else    {//        printf("come in again!!!\n");        res=query(m+1,r,x,(rt<<1)+1);//        printf("come out again!!!\n");    }    return res;}int main(){    int q,n;    scanf("%d%d",&q,&n);    build(1,n,1);    char s[10];    while(q--)    {        scanf("%s",s);        if(s[0]=='Q')        {            int x;            scanf("%d",&x);            printf("%d\n",query(1,n,x,1));        }        else if(s[0]=='A')        {            int l,r,tmp;            scanf("%d%d%d",&l,&r,&tmp);            update(1,n,l,r,tmp,1);        }    }}

附:别人的线段树代码我看不懂,,,,只好自己写了一个。。。

原创粉丝点击