hdu 5023 线段树

成端更新+统计区间内的值

挺模板的题。。。

一开始没想起来用set统计，傻傻地去排序了【大雾

 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<set>
using namespace std;
struct
{
    int l,r;
    int dat;
}t[3000010];
char ch;
int N,M,num,aa,bb,cc;
set<int> s;

//建树：
void build(int l,int r,int o)
{
    if (o>num) num=o;
    t[o].l=l;   t[o].r=r;   t[o].dat=2;
    if (l!=r)
    {
        int mid=(l+r)/2;
        build(l,mid,2*o);
        build(mid+1,r,2*o+1);
    }
}

//push_down：(向下更新一层)
void push_down(int o)
{
    if (t[o].dat!=0)
    {
        t[o<<1].dat=t[o].dat;
        t[o<<1|1].dat=t[o].dat;
        t[o].dat=0;
    }
}

//更新：cin>>ml>>mr>>md;    update(ml,mr,1,md);        ->令a[ml..mr]=md
void update(int l,int r,int o,int md)
{
    if (o>num) return;
    int tl=t[o].l,tr=t[o].r;
    if ((tl==l)&&(tr==r))
    {
        t[o].dat=md;
        return;
    }
    if(t[o].dat==md)    return;
    int mid=(tl+tr)>>1;
    //if (tl==tr) return;
    push_down(o);
    if (r<=mid)
        update(l,r,o<<1,md);
    else if (l>mid)
        update(l,r,o<<1|1,md);
    else
    {
        update(l,mid,o<<1,md);
        update(mid+1,r,o<<1|1,md);
    }
}

void sum(int l,int r,int o)
{
    if(o>num)   return;
    int tl=t[o].l,tr=t[o].r;
    //if((l==tl)&&(r==tr))
    if(t[o].dat)                //经过的地方若t[o].dat不为0就加入set，而不是对每个tl==tr的节点统计。lazy思想的精髓
    {
        s.insert(t[o].dat);
        return;
    }
    else
    {
        int mid=(tl+tr)/2;
        if(r<=mid)
            sum(l,r,2*o);
        else if (l>mid)
            sum(l,r,2*o+1);
        else
        {
            sum(l,mid,2*o);
            sum(mid+1,r,2*o+1);
        }
    }
}

int main()
{
    while(cin>>N>>M)
    {
        if((N==0)&&(M==0))
            break;
        build(1,N,1);
        for(int i=1;i<=M*2;i++)
        {
            scanf("%c ",&ch);
            //cout<<"iiiiiiii"<<i<<" "<<M<<endl;
            if(ch=='P')
            {
                scanf("%d%d%d",&aa,&bb,&cc);
                update(aa,bb,1,cc);
            }
            else if (ch=='Q')
            {
                s.clear();
                scanf("%d%d",&aa,&bb);
                sum(aa,bb,1);
                //printf("SS:  %lu\n", s.size());
                int tmp=s.size();
                for(set<int>::iterator it=s.begin();it!=s.end();it++)
                {
                    printf("%d",*it);
                    tmp--;
                    if(tmp!=0)  printf(" ");
                }
                printf("\n");
            }
        }
    }
    return 0;
}

 

 

 

 

Ref：http://blog.csdn.net/lyhvoyage/article/details/39518963