BZOJ3267/3272 KC采花/Zgg吃東西(線段樹)
阿新 • • 發佈:2018-09-23
end ini clas tdi 操作 else void ++ 最優
直接維護選k個子段時的最優解似乎也可以做,然而復雜度是O(nk2logn),顯然跑不過。
考慮一種費用流做法。序列裏每個點拆成入點和出點,源連入匯連出,入點和出點間連流量1費用ai的邊,相鄰點出點向入點連流量1費用0的邊,整體限流k。
直接跑當然還不如暴力。觀察一下這個做法是在幹啥:每次選擇費用最大的一段,然後利用反向邊將這一段的費用取反。
這個做法看起來非常貪心(不過費用流本質上也挺貪心的),不過看起來確實是對的。當然從貪心角度就完全不會證了。
於是考慮利用這種做法維護。那麽線段樹維護區間最大子段和和最小子段和,取反時交換,剩下的是基本操作了。
#include<iostream> #include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<algorithm> using namespace std; int read() { int x=0,f=1;char c=getchar(); while (c<‘0‘||c>‘9‘) {if (c==‘-‘) f=-1;c=getchar();} while (c>=‘0‘&&c<=‘9‘) x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } #define N 100010 int n,m,a[N]; struct data{int L,R,max,maxl,maxr,min,minl,minr,sum,rmax,rmaxid,rmin,rminid,lmax,lmaxid,lmin,lminid,rev; }tree[N<<2],q[22]; data merge(data a,data b) { data u; u.rev=0;u.L=a.L,u.R=b.R; u.sum=a.sum+b.sum; if (a.lmax>a.sum+b.lmax) u.lmax=a.lmax,u.lmaxid=a.lmaxid;else u.lmax=a.sum+b.lmax,u.lmaxid=b.lmaxid; if (a.lmin<a.sum+b.lmin) u.lmin=a.lmin,u.lminid=a.lminid; else u.lmin=a.sum+b.lmin,u.lminid=b.lminid; if (b.rmax>b.sum+a.rmax) u.rmax=b.rmax,u.rmaxid=b.rmaxid; else u.rmax=b.sum+a.rmax,u.rmaxid=a.rmaxid; if (b.rmin<b.sum+a.rmin) u.rmin=b.rmin,u.rminid=b.rminid; else u.rmin=b.sum+a.rmin,u.rminid=a.rminid; u.max=a.rmax+b.lmax;u.maxl=a.rmaxid;u.maxr=b.lmaxid; u.min=a.rmin+b.lmin;u.minl=a.rminid;u.minr=b.lminid; if (a.max>u.max) u.max=a.max,u.maxl=a.maxl,u.maxr=a.maxr; if (a.min<u.min) u.min=a.min,u.minl=a.minl,u.minr=a.minr; if (b.max>u.max) u.max=b.max,u.maxl=b.maxl,u.maxr=b.maxr; if (b.min<u.min) u.min=b.min,u.minl=b.minl,u.minr=b.minr; return u; } void work(int k) { tree[k].rev^=1; tree[k].sum=-tree[k].sum; swap(tree[k].max,tree[k].min);tree[k].max=-tree[k].max,tree[k].min=-tree[k].min; swap(tree[k].maxl,tree[k].minl);swap(tree[k].maxr,tree[k].minr); swap(tree[k].lmax,tree[k].lmin);tree[k].lmax=-tree[k].lmax,tree[k].lmin=-tree[k].lmin; swap(tree[k].lmaxid,tree[k].lminid); swap(tree[k].rmax,tree[k].rmin);tree[k].rmax=-tree[k].rmax,tree[k].rmin=-tree[k].rmin; swap(tree[k].rmaxid,tree[k].rminid); } void down(int k){work(k<<1),work(k<<1|1),tree[k].rev=0;} void build(int k,int l,int r) { tree[k].L=l,tree[k].R=r; if (l==r) { tree[k].sum=tree[k].max=tree[k].min=tree[k].rmax=tree[k].rmin=tree[k].lmax=tree[k].lmin=a[l]; tree[k].maxl=tree[k].maxr=tree[k].minl=tree[k].minr=tree[k].rmaxid=tree[k].rminid=tree[k].lmaxid=tree[k].lminid=l; return; } int mid=l+r>>1; build(k<<1,l,mid); build(k<<1|1,mid+1,r); tree[k]=merge(tree[k<<1],tree[k<<1|1]); } void rev(int k,int l,int r) { if (tree[k].L==l&&tree[k].R==r) {work(k);return;} if (tree[k].rev) down(k); int mid=tree[k].L+tree[k].R>>1; if (r<=mid) rev(k<<1,l,r); else if (l>mid) rev(k<<1|1,l,r); else rev(k<<1,l,mid),rev(k<<1|1,mid+1,r); tree[k]=merge(tree[k<<1],tree[k<<1|1]); } void modify(int k,int p,int x) { if (tree[k].L==tree[k].R) { tree[k].sum=tree[k].max=tree[k].min=tree[k].rmax=tree[k].rmin=tree[k].lmax=tree[k].lmin=x; tree[k].rev=0; return; } if (tree[k].rev) down(k); int mid=tree[k].L+tree[k].R>>1; if (p<=mid) modify(k<<1,p,x); else modify(k<<1|1,p,x); tree[k]=merge(tree[k<<1],tree[k<<1|1]); } data query(int k,int l,int r) { if (tree[k].L==l&&tree[k].R==r) return tree[k]; if (tree[k].rev) down(k); int mid=tree[k].L+tree[k].R>>1; if (r<=mid) return query(k<<1,l,r); else if (l>mid) return query(k<<1|1,l,r); else return merge(query(k<<1,l,mid),query(k<<1|1,mid+1,r)); } int main() { #ifndef ONLINE_JUDGE freopen("bzoj3272.in","r",stdin); freopen("bzoj3272.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(); for (int i=1;i<=n;i++) a[i]=read(); m=read(); build(1,1,n); while (m--) { int op=read(); if (op) { int l=read(),r=read(),k=read(),ans=0; for (int i=1;i<=k;i++) { q[i]=query(1,l,r); if (q[i].max>0) ans+=q[i].max,rev(1,q[i].maxl,q[i].maxr); else {k=i-1;break;} } for (int i=1;i<=k;i++) rev(1,q[i].maxl,q[i].maxr); printf("%d\n",ans); } else { int x=read(),y=read(); modify(1,x,y); } } return 0; }
BZOJ3267/3272 KC采花/Zgg吃東西(線段樹)