码迷,mamicode.com
首页 > 其他好文 > 详细

COJ 1010 WZJ的数据结构(十) 线段树的地狱

时间:2015-06-12 20:54:37      阅读:155      评论:0      收藏:0      [点我收藏+]

标签:

传送门:http://oj.cnuschool.org.cn/oj/home/problem.htm?problemID=1001

WZJ的数据结构(十)
难度级别:D; 运行时间限制:3000ms; 运行空间限制:51200KB; 代码长度限制:2000000B
试题描述

请你设计一个数据结构,高效执行以下过程:

#include<iostream>
using namespace std;
const int maxn=100010;
int A[maxn];
int tp,ql,qr,v;
int _max,_min,_sum;
int main()
{
 int n,Q,tp;
 cin>>n>>Q;
 for(int i=1;i<=n;i++) cin>>A[i];
 while(Q--)
 {
  cin>>tp>>ql>>qr;
  if(tp==2)
  {
   _sum=0; _max=-1000000000; _min=1000000000;
   for(int i=ql;i<=qr;i++)
   {
    _max=max(_max,A[i]);
    _min=min(_min,A[i]);
    _sum+=A[i];
   }
   cout<<_max<<" "<<_min<<" "<<_sum<<endl;
  }
  else
  {
   cin>>v;
   for(int i=ql;i<=qr;i++)
   {
    if(!tp) A[i]=0;
    A[i]+=v;
   }
  }
 }
 return 0;
}

输入
见代码
输出
见代码
输入示例
5 5
1 2 3 4 5
2 1 3
0 1 5 2
1 2 3 3
2 1 4
2 1 5
输出示例
3 1 6
5 2 14
5 2 16
其他说明
1<=n,Q<=100000
0<=tp<=2
1<=A[i],v<=1000
1<=ql<=qr<=n

更新:指针版:

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cmath>
  4 #include<algorithm>
  5 #include<queue>
  6 #include<cstring>
  7 #define PAU putchar(‘ ‘)
  8 #define ENT putchar(‘\n‘)
  9 #define CH for(int d=0;d<=1;d++) if(ch[d])
 10 using namespace std;
 11 const int maxn=100000+10,maxnode=200000+10,inf=-1u>>1;
 12 struct node{
 13     node*ch[2];int mi,mx,sm,add,set,siz;node(){mi=inf;mx=-inf;sm=add=0;set=inf;}
 14     void sett(int tag){mi=mx=set=tag;sm=tag*siz;add=0;return;}
 15     void addt(int tag){add+=tag;mi+=tag;mx+=tag;sm+=tag*siz;return;}
 16     void down(){
 17         if(set!=inf){CH{ch[d]->sett(set);}set=inf;}
 18         if(add){CH{ch[d]->addt(add);}add=0;}
 19         return;
 20     }
 21     void update(){
 22         mi=inf;mx=-inf;sm=0;
 23         CH{mi=min(mi,ch[d]->mi);mx=max(mx,ch[d]->mx);sm+=ch[d]->sm;}
 24         return;
 25     }
 26 }seg[maxnode],*nodecnt=seg,*root;
 27 int A[maxn],ql,qr,cv,tp;
 28 void build(node*&x,int L,int R){
 29     x=nodecnt++;
 30     if(L==R) x->mi=x->mx=x->sm=A[L];
 31     else{
 32         int M=L+R>>1;
 33         build(x->ch[0],L,M);
 34         build(x->ch[1],M+1,R);
 35         x->update();
 36     } x->siz=R-L+1;return;
 37 }
 38 void update(node*&x,int L,int R){
 39     if(ql<=L&&R<=qr){
 40         if(tp) x->addt(cv);
 41         else x->sett(cv);
 42     }
 43     else{
 44         int M=L+R>>1;
 45         x->down();
 46         if(ql<=M) update(x->ch[0],L,M);
 47         if(qr>M) update(x->ch[1],M+1,R);
 48         x->update();
 49     }
 50     return;
 51 }
 52 int _mi,_mx,_sm;
 53 void query(node*x,int L,int R){
 54     if(ql<=L&&R<=qr){
 55         _mi=min(_mi,x->mi);
 56         _mx=max(_mx,x->mx);
 57         _sm+=x->sm;
 58     }
 59     else{
 60         int M=L+R>>1;
 61         x->down();
 62         if(ql<=M) query(x->ch[0],L,M);
 63         if(qr>M) query(x->ch[1],M+1,R);
 64     } return;
 65 }
 66 inline int read(){
 67     int x=0,sig=1;char ch=getchar();
 68     while(!isdigit(ch)){if(ch==-)sig=-1;ch=getchar();}
 69     while(isdigit(ch))x=10*x+ch-0,ch=getchar();
 70     return x*sig;
 71 }
 72 inline void write(int x){
 73     if(x==0){putchar(0);return;}if(x<0)putchar(-),x=-x;
 74     int len=0,buf[15];while(x)buf[len++]=x%10,x/=10;
 75     for(int i=len-1;i>=0;i--)putchar(buf[i]+0);return;
 76 }
 77 int n,Q;
 78 void init(){
 79     n=read();Q=read();
 80     for(int i=1;i<=n;i++) A[i]=read();
 81     build(root,1,n);
 82     return;
 83 }
 84 void work(){
 85     char ch;
 86     while(Q--){
 87         tp=read();ql=read();qr=read();
 88         if(tp!=2){
 89             cv=read();
 90             update(root,1,n);
 91         }
 92         else{
 93             _mi=inf;_mx=-inf;_sm=0;
 94             query(root,1,n);
 95             write(_mx);PAU;write(_mi);PAU;write(_sm);ENT;
 96         }
 97     }
 98     return;
 99 }
100 void print(){
101     return;
102 }
103 int main(){init();work();print();return 0;}

题解:传两个lazy标:setv,addv,addv是建立在setv上的增加量,这样来一个setv就清掉addv,来一个addv就加上去,维护懒标的时候先维护setv再维护addv就好了。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cmath>
  4 #include<algorithm>
  5 #include<queue>
  6 #include<cstring>
  7 #define PAU putchar(‘ ‘)
  8 #define ENT putchar(‘\n‘)
  9 using namespace std;
 10 const int maxn=100000+10,maxn3=300000+10,inf=-1u>>1;
 11 int maxv[maxn3],minv[maxn3],sumv[maxn3],setv[maxn3],addv[maxn3],A[maxn];
 12 void pushup(int o,int lc,int rc){
 13     minv[o]=min(minv[lc],minv[rc]);
 14     maxv[o]=max(maxv[lc],maxv[rc]);
 15     sumv[o]=sumv[lc]+sumv[rc];
 16     return;
 17 }
 18 void pushdown(int o,int lc,int rc){
 19     if(setv[o]>=0){
 20         setv[lc]=setv[rc]=setv[o];
 21         addv[lc]=addv[rc]=0;
 22         setv[o]=-1;
 23     } if(addv[o]){
 24         addv[lc]+=addv[o];
 25         addv[rc]+=addv[o];
 26         addv[o]=0;
 27     } return;
 28 }
 29 void maintain(int o,int L,int R){
 30     int lc=o<<1,rc=lc|1;
 31     if(L<R&&setv[o]<0) pushup(o,lc,rc);
 32     if(setv[o]>=0){
 33         minv[o]=maxv[o]=setv[o];
 34         sumv[o]=(R-L+1)*setv[o];
 35     } if(addv[o]){
 36         minv[o]+=addv[o];
 37         maxv[o]+=addv[o];
 38         sumv[o]+=(R-L+1)*addv[o];
 39     } return;
 40 }
 41 int _min,_max,_sum,ql,qr,tp,cv;
 42 void update(int o,int L,int R){
 43     if(ql<=L&&R<=qr){
 44         if(tp) addv[o]+=cv;
 45         else setv[o]=cv,addv[o]=0;
 46     } else{
 47         int M=L+R>>1,lc=o<<1,rc=lc|1;
 48         pushdown(o,lc,rc);
 49         if(ql<=M) update(lc,L,M); else maintain(lc,L,M);
 50         if(qr>M) update(rc,M+1,R); else maintain(rc,M+1,R);
 51     } maintain(o,L,R);return;
 52 }
 53 void build(int o,int L,int R){
 54     if(L==R) setv[o]=A[L];
 55     else{
 56         int M=L+R>>1,lc=o<<1,rc=lc|1;
 57         build(lc,L,M);build(rc,M+1,R);
 58     } maintain(o,L,R);return;
 59 }
 60 void query(int o,int L,int R,int add){
 61     if(setv[o]>=0){
 62         int change=setv[o]+addv[o]+add;
 63         _sum+=(min(R,qr)-max(L,ql)+1)*change;
 64         _min=min(_min,change);
 65         _max=max(_max,change);
 66     } else if(ql<=L&&R<=qr){
 67         _sum+=sumv[o]+(R-L+1)*add;
 68         _min=min(_min,minv[o]+add);
 69         _max=max(_max,maxv[o]+add);
 70     } else{
 71         int M=L+R>>1,lc=o<<1,rc=lc|1;
 72         if(ql<=M) query(lc,L,M,add+addv[o]);
 73         if(M<qr) query(rc,M+1,R,add+addv[o]);
 74     } return;
 75 }
 76 inline int read(){
 77     int x=0,sig=1;char ch=getchar();
 78     while(!isdigit(ch)){if(ch==-)sig=-1;ch=getchar();}
 79     while(isdigit(ch))x=10*x+ch-0,ch=getchar();
 80     return x*=sig;
 81 }
 82 inline void write(int x){
 83     if(x==0){putchar(0);return;}if(x<0)putchar(-),x=-x;
 84     int len=0,buf[15];while(x)buf[len++]=x%10,x/=10;
 85     for(int i=len-1;i>=0;i--)putchar(buf[i]+0);return;
 86 }
 87 int n,Q;
 88 void init(){
 89     memset(setv,-1,sizeof(setv));
 90     n=read();Q=read();
 91     for(int i=1;i<=n;i++) A[i]=read();
 92     build(1,1,n);
 93     return;
 94 }
 95 void work(){
 96     while(Q--){
 97         tp=read();ql=read();qr=read();
 98         if(tp==2){//query
 99             _sum=0;_min=inf;_max=-inf;
100             query(1,1,n,0);
101             write(_max);PAU;write(_min);PAU;write(_sum);ENT;
102         } else cv=read(),update(1,1,n);
103     }
104     return;
105 }
106 void print(){
107     return;
108 }
109 int main(){init();work();print();return 0;}

 

COJ 1010 WZJ的数据结构(十) 线段树的地狱

标签:

原文地址:http://www.cnblogs.com/chxer/p/4474006.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!