codves 4927 线段树练习5
有n个数和5种操作
add a b c:把区间[a,b]内的所有数都增加c
set a b c:把区间[a,b]内的所有数都设为c
sum a b:查询区间[a,b]的区间和
max a b:查询区间[a,b]的最大值
min a b:查询区间[a,b]的最小值
输入描述 Input Description
第一行两个整数n,m,第二行n个整数表示这n个数的初始值
接下来m行操作,同题目描述
输出描述 Output Description
对于所有的sum、max、min询问,一行输出一个答案
样例输入 Sample Input
10 6
3 9 2 8 1 7 5 0 4 6
add 4 9 4
set 2 6 2
add 3 8 2
sum 2 10
max 1 7
min 3 6
样例输出 Sample Output
49
11
4
#include<cstring> #include<cstdio> #include<algorithm> #define ll long long using namespace std; const int N=100010; const int inf=0x3f3f3f3f; struct node { ll a_la,s_la,sum,minx,maxx; int l,r,flag; }e[N*8]; int read() { int ans=0,f=1;char c=getchar(); while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();} while(c>='0'&&c<='9'){ans=ans*10+c-48;c=getchar();} return ans*f; } inline void pushup(int ro) { e[ro].sum=e[ro<<1].sum+e[ro<<1|1].sum; e[ro].minx=min(e[ro<<1].minx,e[ro<<1|1].minx); e[ro].maxx=max(e[ro<<1].maxx,e[ro<<1|1].maxx); } void build(int ro,int l,int r) { e[ro].l=l,e[ro].r=r; if(l==r) {int t=read();e[ro].minx=t,e[ro].maxx=t,e[ro].sum=t;} else { int mid=(l+r)>>1; build(ro<<1,l,mid);build(ro<<1|1,mid+1,r);pushup(ro); } return; } inline void down(int ro) { if(e[ro].flag) { e[ro<<1].flag=e[ro<<1|1].flag=1; e[ro].flag=0; int mid=(e[ro].l+e[ro].r)>>1; e[ro<<1].sum=e[ro].s_la*(mid-e[ro].l+1); e[ro<<1|1].sum=e[ro].s_la*(e[ro].r-mid); e[ro<<1].maxx=e[ro<<1].minx=e[ro<<1|1].maxx=e[ro<<1|1].minx=e[ro].s_la; e[ro<<1].s_la=e[ro<<1|1].s_la=e[ro].s_la; e[ro<<1].a_la=e[ro<<1|1].a_la=e[ro].s_la=0; } if(e[ro].a_la) { e[ro<<1].a_la+=e[ro].a_la; e[ro<<1|1].a_la+=e[ro].a_la; e[ro<<1].sum+=e[ro].a_la*(e[ro<<1].r-e[ro<<1].l+1); e[ro<<1].minx+=e[ro].a_la; e[ro<<1].maxx+=e[ro].a_la; e[ro<<1|1].sum+=e[ro].a_la*(e[ro<<1|1].r-e[ro<<1|1].l+1); e[ro<<1|1].minx+=e[ro].a_la; e[ro<<1|1].maxx+=e[ro].a_la; e[ro].a_la=0; } } void add(int ro,int l,int r,int x) { down(ro); if(l<=e[ro].l&&e[ro].r<=r) { if(e[ro].s_la) e[ro].s_la+=x; else e[ro].a_la+=x; e[ro].sum+=x*(e[ro].r-e[ro].l+1); e[ro].maxx+=x; e[ro].minx+=x; return; } int mid=(e[ro].l+e[ro].r)>>1; if(l<=mid) add(ro<<1,l,r,x); if(r>mid) add(ro<<1|1,l,r,x); pushup(ro); return; } void set(int ro,int l,int r,int x) { down(ro); if(l<=e[ro].l&&e[ro].r<=r) { e[ro].sum=x*(e[ro].r-e[ro].l+1); e[ro].minx=x; e[ro].maxx=x; e[ro].a_la=0; e[ro].s_la=x; e[ro].flag=1; return; } int mid=(e[ro].l+e[ro].r)>>1; if(l<=mid) set(ro<<1,l,r,x); if(r>mid) set(ro<<1|1,l,r,x); pushup(ro); return; } ll asum(int ro,int l,int r) { down(ro); if(l<=e[ro].l&&e[ro].r<=r) return e[ro].sum; int mid=(e[ro].l+e[ro].r)>>1; ll ans=0; if(l<=mid) ans+=asum(ro<<1,l,r); if(mid<r) ans+=asum(ro<<1|1,l,r); return ans; } ll amin(int ro,int l,int r) { down(ro); if(l<=e[ro].l&&e[ro].r<=r) return e[ro].minx; int mid=(e[ro].l+e[ro].r)>>1; ll answer=inf; if(l<=mid) answer=min(answer,amin(ro<<1,l,r)); if(mid<r) answer=min(answer,amin(ro<<1|1,l,r)); return answer; } ll amax(int ro,int l,int r) { down(ro); if(l<=e[ro].l&&e[ro].r<=r) return e[ro].maxx; int mid=(e[ro].l+e[ro].r)>>1; ll answer=-inf; if(l<=mid) answer=max(answer,amax(ro<<1,l,r)); if(mid<r) answer=max(answer,amax(ro<<1|1,l,r)); return answer; } int main() { int n,m; n=read(),m=read(); build(1,1,n); //for(int i=1;i<=20;i++) printf("%lld %lld %lld %d %d\n",e[i].sum,e[i].maxx,e[i].minx,e[i].l,e[i].r); for(int i=1;i<=m;i++) { char ch[10]; scanf("%s",ch); if(ch[1]=='d') {int l,r,x;scanf("%d %d %d",&l,&r,&x);add(1,l,r,x);} //for(int i=1;i<=20;i++) printf("%lld %lld %lld %d %d\n",e[i].sum,e[i].maxx,e[i].minx,e[i].l,e[i].r); if(ch[1]=='e') {int l,r,x;scanf("%d %d %d",&l,&r,&x);set(1,l,r,x);} if(ch[1]=='u') {int l,r;scanf("%d %d",&l,&r);printf("%lld\n",asum(1,l,r));} if(ch[1]=='a') {int l,r;scanf("%d %d",&l,&r);printf("%lld\n",amax(1,l,r));} if(ch[1]=='i') {int l,r;scanf("%d %d",&l,&r);printf("%lld\n",amin(1,l,r));} } return 0; }
模板!!!
调了好久。