题意: 给出一个长度为n的整数序列D,你的任务是对m个询问作出回答。对于询问(a,b),需要找到两个下标x和y,使得a≤x≤y≤b,并且Dx+Dx+1+...+Dy尽量大。如果有多组满足条件的x和y,x应该尽量小。如果还有多解,y应该尽量小。
tags: 分治思想,线段树 大白书201
区别于静态最大连续和,只能用分治算法: 最优解要么完全在左半序列,要么完全在右半序列,要么跨越中点。
我们构造一棵线段树,维护 3 个值:最大连续和 max_sub ,最大前缀和 max_prefix , 最大后缀和 max_suffix 。按照上述分治算法维护即可。
因为这个题要求出标号,所以我们还要记录标号。 标号要求尽可能小,我们只要在维护的时候注意一下顺序就好了。
#include<bits/stdc++.h> using namespace std; #pragma comment(linker, "/STACK:102400000,102400000") #define rep(i,a,b) for (int i=a; i<=b; ++i) #define per(i,b,a) for (int i=b; i>=a; --i) #define mes(a,b) memset(a,b,sizeof(a)) #define INF 0x3f3f3f3f #define MP make_pair #define PB push_back #define fi first #define se second #define mid (l+(r-l)/2) typedef long long ll; const int N = 500005; struct Item { ll sum, sub, prefix, suffix; int id1, id2, id3, id4; } tr[N<<2]; ll ai; int n, m; Item pushup(Item ro1, Item ro2) { Item ret; ret.sum = ro1.sum+ro2.sum; // sum ret.sub = ro1.sub; // sub ret.id1=ro1.id1, ret.id2=ro1.id2; if(ret.sub < ro1.suffix+ro2.prefix) { ret.sub = ro1.suffix+ro2.prefix; ret.id1=ro1.id4, ret.id2=ro2.id3; } if(ret.sub < ro2.sub) { ret.sub = ro2.sub; ret.id1=ro2.id1, ret.id2=ro2.id2; } ret.prefix = ro1.prefix; //prefix ret.id3 = ro1.id3; if(ret.prefix < ro1.sum+ro2.prefix) { ret.prefix = ro1.sum+ro2.prefix; ret.id3 = ro2.id3; } ret.suffix = ro1.suffix+ro2.sum; //suffix ret.id4 = ro1.id4; if(ret.suffix < ro2.suffix) { ret.suffix = ro2.suffix; ret.id4 = ro2.id4; } return ret; } void build(int ro, int l, int r) { if(l==r) { scanf("%lld", &ai); tr[ro].sum=tr[ro].sub=tr[ro].prefix=tr[ro].suffix=ai; tr[ro].id1=tr[ro].id2=tr[ro].id3=tr[ro].id4=l; return ; } build(ro<<1, l, mid); build(ro<<1|1, mid+1, r); tr[ro] = pushup(tr[ro<<1], tr[ro<<1|1]); } Item query(int ro, int l, int r, int ql, int qr) { if(ql<=l && r<=qr) { return tr[ro]; } if(qr<=mid) return query(ro<<1, l, mid, ql, qr); else if(mid<ql) return query(ro<<1|1, mid+1, r, ql, qr); else return pushup(query(ro<<1,l,mid,ql,qr), query(ro<<1|1,mid+1,r,ql,qr)); } int main() { int cas = 0; while(~scanf("%d%d", &n, &m)) { printf("Case %d:\n", ++cas); build(1, 1, n); int a, b; Item ans; while(m--) { scanf("%d %d", &a, &b); ans = query(1, 1, n, a, b); printf("%d %d\n", ans.id1, ans.id2); } } return 0; }
看了别人的代码修改了一下:
#include<bits/stdc++.h> using namespace std; #pragma comment(linker, "/STACK:102400000,102400000") #define rep(i,a,b) for (int i=a; i<=b; ++i) #define per(i,b,a) for (int i=b; i>=a; --i) #define mes(a,b) memset(a,b,sizeof(a)) #define INF 0x3f3f3f3f #define MP make_pair #define PB push_back #define fi first #define se second #define mid (l+(r-l)/2) #define PII pair<int , int > typedef long long ll; const int N = 500005; struct Item { PII sub, pre, suf; int l, r; } tr[N<<2]; ll ai, sum[N]; int n, m; PII getMax(PII x, PII y) { ll sumx=sum[x.se]-sum[x.fi-1], sumy=sum[y.se]-sum[y.fi-1]; if(sumx!=sumy) return sumx>sumy ? x : y; return x<y ? x : y; } Item pushup(Item ro1, Item ro2) { Item ret; ret.l=ro1.l, ret.r=ro2.r; ret.sub = getMax(ro1.sub, getMax(ro2.sub, MP(ro1.suf.fi, ro2.pre.se))); ret.pre = getMax(ro1.pre, MP(ro1.l, ro2.pre.se)); ret.suf = getMax(ro2.suf, MP(ro1.suf.fi, ro2.r)); return ret; } void build(int ro, int l, int r) { if(l==r) { scanf("%lld", &ai); tr[ro].sub=tr[ro].pre=tr[ro].suf=MP(l,l); tr[ro].l=tr[ro].r=l; sum[l] = sum[l-1]+ai; return ; } build(ro<<1, l, mid); build(ro<<1|1, mid+1, r); tr[ro] = pushup(tr[ro<<1], tr[ro<<1|1]); } Item query(int ro, int l, int r, int ql, int qr) { if(ql<=l && r<=qr) return tr[ro]; if(qr<=mid) return query(ro<<1, l, mid, ql, qr); else if(mid<ql) return query(ro<<1|1, mid+1, r, ql, qr); else return pushup(query(ro<<1,l,mid,ql,qr), query(ro<<1|1,mid+1,r,ql,qr)); } void Init() { mes(sum, 0); } int main() { int cas = 0; while(~scanf("%d%d", &n, &m)) { Init(); build(1, 1, n); printf("Case %d:\n", ++cas); int a, b; Item ans; while(m--) { scanf("%d %d", &a, &b); ans = query(1, 1, n, a, b); printf("%d %d\n", ans.sub.fi, ans.sub.se); } } return 0; }