【HDU5306】【DTOJ2481】Gorgeous Sequence【线段树】
题目大意:给你一个序列a,你有三个操作,0: x y t将a[x,y]和t取min;1:x y求a[x,y]的最大值;2:x y求a[x,y]的sum
题解:首先很明显就是线段树裸题,那么考虑如何维护
区间最大值和区间sum很好维护,0操作不好做,那么考虑怎么快速解决0操作
很容易想到维护区间最大值和区间是否全部相同,这是一个做法,但是时间复杂度上却不正确,但也给了我们一个思路,可以通过维护最大值之类的数来加快操作
一个不行就两个,于是考虑维护次大值,那么每次0操作就有三种情况,一:比区间最大值还大,那么直接返回;二:大于区间次大值小于区间最大值,那么可以直接修改区间;三:小于区间次大值,那么继续递归下去;
这么做是可行的,在更新区间时因为要维护sum值,于是多维护一个区间max有多少个即可
证明略,时间复杂度o(nlog^2n)——来源:吉老师
代码:
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<string> #define ll long long using namespace std; int T,n,m; ll a[1000001]; class Segtree { public: ll sum[1000001*5],mx[1000001*5],smx[1000001*5],cnt[1000001*5],fl[1000001*5]; void pushup(int pos) { sum[pos]=sum[pos<<1]+sum[pos<<1|1]; if(mx[pos<<1]>mx[pos<<1|1]) { mx[pos]=mx[pos<<1]; cnt[pos]=cnt[pos<<1]; if(mx[pos<<1|1]>smx[pos<<1])smx[pos]=mx[pos<<1|1]; else smx[pos]=smx[pos<<1]; } if(mx[pos<<1]<mx[pos<<1|1]) { mx[pos]=mx[pos<<1|1]; cnt[pos]=cnt[pos<<1|1]; if(mx[pos<<1]>smx[pos<<1|1])smx[pos]=mx[pos<<1]; else smx[pos]=smx[pos<<1|1]; } if(mx[pos<<1]==mx[pos<<1|1]) { mx[pos]=mx[pos<<1]; cnt[pos]=cnt[pos<<1]+cnt[pos<<1|1]; if(smx[pos<<1]>smx[pos<<1|1])smx[pos]=smx[pos<<1]; else smx[pos]=smx[pos<<1|1]; } } void pushdown(int pos) { if(fl[pos]>=0) { if(mx[pos<<1]>fl[pos]) { sum[pos<<1]-=(mx[pos<<1]-fl[pos])*cnt[pos<<1]; fl[pos<<1]=mx[pos<<1]=fl[pos]; } if(mx[pos<<1|1]>fl[pos]) { sum[pos<<1|1]-=(mx[pos<<1|1]-fl[pos])*cnt[pos<<1|1]; fl[pos<<1|1]=mx[pos<<1|1]=fl[pos]; } fl[pos]=-1; } } void build(int l,int r,int pos) { if(l==r) { sum[pos]=mx[pos]=a[l]; cnt[pos]=1; fl[pos]=smx[pos]=-1; return; } int mid=l+r>>1; build(l,mid,pos<<1); build(mid+1,r,pos<<1|1); pushup(pos); fl[pos]=-1; } void change(int l,int r,int al,int ar,ll v,int pos) { int mid=l+r>>1; if(l==al && r==ar) { if(v>=mx[pos])return; if(smx[pos]<v && v<mx[pos]) { sum[pos]-=(mx[pos]-v)*cnt[pos]; fl[pos]=mx[pos]=v; return; } change(l,mid,al,mid,v,pos<<1); change(mid+1,r,mid+1,ar,v,pos<<1|1); pushup(pos); return; } pushdown(pos); if(ar<=mid)change(l,mid,al,ar,v,pos<<1); if(al>mid)change(mid+1,r,al,ar,v,pos<<1|1); if(al<=mid && ar>mid){change(l,mid,al,mid,v,pos<<1);change(mid+1,r,mid+1,ar,v,pos<<1|1);} pushup(pos); } ll askmax(int l,int r,int al,int ar,int pos) { if(l==al && r==ar)return mx[pos]; int mid=l+r>>1; pushdown(pos); if(ar<=mid)return askmax(l,mid,al,ar,pos<<1); if(al>mid)return askmax(mid+1,r,al,ar,pos<<1|1); if(al<=mid && ar>mid) { ll t1=askmax(l,mid,al,mid,pos<<1),t2=askmax(mid+1,r,mid+1,ar,pos<<1|1); return t1>t2?t1:t2; } } ll asksum(int l,int r,int al,int ar,int pos) { if(l==al && r==ar)return sum[pos]; int mid=l+r>>1; pushdown(pos); if(ar<=mid)return asksum(l,mid,al,ar,pos<<1); if(al>mid)return asksum(mid+1,r,al,ar,pos<<1|1); if(al<=mid && ar>mid)return asksum(l,mid,al,mid,pos<<1)+asksum(mid+1,r,mid+1,ar,pos<<1|1); } }segtree; int main() { scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++)scanf("%lld",&a[i]); segtree.build(1,n,1); int mod,x,y;ll t; while(m--) { scanf("%d",&mod); if(mod==0) { scanf("%d%d%lld",&x,&y,&t); segtree.change(1,n,x,y,t,1); } if(mod==1) { scanf("%d%d",&x,&y); printf("%lld\n",segtree.askmax(1,n,x,y,1)); } if(mod==2) { scanf("%d%d",&x,&y); printf("%lld\n",segtree.asksum(1,n,x,y,1)); } } } return 0; }
心得:有时候多维护一个次大值能解决很多问题,更多时候在不知道如何证明时间复杂度的情况下往往维护次大值是正确的