UVA 1400 "Ray, Pass methe dishes!"(线段树,区间合并)
题意:
给出一个长度为n的整数序列D,你的任务是对m各询问做出回答。对于询问(a,b),需要找到两个下标x和y,使得a<=x<=y<=b,并且Dx+Dx+1+...+Dy尽量大。如果有多组满足条件的x和y,x应该尽量小。如果还有多解,y应该尽量小。
分析:
本题详解见:刘汝佳训练指南P201
其实如果对于线段树的本质有较深的理解本题其实不难。
建立一棵线段树,每个节点维护下面3种信息:
max_sub:本节点维护的最大连续和的子区间[L,R]
max_prefix:本节点的最大前缀和区间的右端坐标R
max_suffix:本节点的最大后缀和区间的左端坐标L
然后对于一个查询[20,50]区间的最大连续和,由于区间被分成了[20,32],[33,50]结果只有下面3中情况:
1. 最大连续和在[20,32]中,即max_sub[20,32]
2. 最大连续和在[33,50]中,即max_sub[33,50]
3. 最大连续和x在[20,32]中,y在[33,50]中即max_sum[20,50]=max_suffix[20,32]+max_prefix[33,50],并且更新x,y的坐标.
max_prefix和max_sufiix也可以用上面的分两段方式来求,不过还需要维护sum[]的值.
在build建树,和查询的时候都可以用上面的思想来求.
AC代码:179ms
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
const int MAXN=500000+1000;
typedef long long LL;
typedef pair<int ,int> interval;
interval max_sub[MAXN*4];
int max_prefix[MAXN*4], max_suffix[MAXN*4];
LL prefix_sum[MAXN];
LL get_sum(int L,int R)
{
return prefix_sum[R]-prefix_sum[L-1];
}
LL get_sum(interval p)
{
return get_sum(p.first, p.second);
}
interval better(interval a, interval b)
{
if(get_sum(a) != get_sum(b)) return get_sum(a) > get_sum(b) ? a : b ;
return a < b ? a : b;
}
void PushUp(int i,int l,int r)
{
//计算最大和
max_sub[i] = better(max_sub[i*2], max_sub[i*2+1]);
max_sub[i] = better(max_sub[i], make_pair(max_suffix[i*2], max_prefix[i*2+1]));
//计算前缀
LL v1 = get_sum(l, max_prefix[i*2]);
LL v2 = get_sum(l, max_prefix[i*2+1]);
if(v1 == v2) max_prefix[i] = min(max_prefix[i*2], max_prefix[i*2+1]);
else max_prefix[i] = v1 > v2 ? max_prefix[i*2] : max_prefix[i*2+1];
//计算后缀
v1 = get_sum(max_suffix[i*2], r);
v2 = get_sum(max_suffix[i*2+1], r);
if(v1 == v2) max_suffix[i] = min(max_suffix[i*2], max_suffix[i*2+1]);
else max_suffix[i] = v1 > v2 ? max_suffix[i*2] : max_suffix[i*2+1];
}
void build(int i,int l,int r)
{
if(l==r)
{
max_sub[i]=make_pair(l, r);
max_prefix[i]=max_suffix[i]=l;
//printf("l=%d,r=%d,interval is [%d,%d]\n",l,r,max_sub[i].first,max_sub[i].second);
return ;
}
int m=(l+r)/2;
build(lson);
build(rson);
if(l==1&&r==2)
{
int x=1;
int y=1;
}
PushUp(i,l,r);
//printf("l=%d,r=%d,interval is [%d,%d]\n",l,r,max_sub[i].first,max_sub[i].second);
}
interval query_prefix(int ql, int qr, int i, int l, int r)//调用该函数的前提是:[ql,qr]包含了左端点l
{
if(max_prefix[i] <= qr) return make_pair(l, max_prefix[i]);
else{
int m = (l+r)/2;
if(qr <= m) return query_prefix(ql,qr,lson);
interval a = query_prefix(ql,qr,rson);
a.first = l;
return better(a, make_pair(l, max_prefix[i*2]));
}
}
interval query_suffix(int ql, int qr, int i, int l, int r)//调用该函数的前提是:[ql,qr]包含了右端点r
{
if(max_suffix[i] >= ql) return make_pair(max_suffix[i], r);
else{
int m = (l+r)/2;
if(m < ql) return query_suffix(ql,qr,rson);
interval a = query_suffix(ql,qr,lson);
a.second = r;
return better(a, make_pair(max_suffix[i*2+1], r));
}
}
interval query(int ql,int qr,int i,int l,int r)//返回[ql,qr]与[l,r]相交的最大连续和区间
{
if(ql<=l && r<=qr) return max_sub[i];
else{
int m = (l+r)/2;
if(qr<=m) return query(ql,qr,lson);
if(ql>m) return query(ql,qr,rson);
int L = query_suffix(ql,qr,lson).first;
int R = query_prefix(ql,qr,rson).second;
interval a = better(query(ql,qr,lson), query(ql,qr,rson));
return better(make_pair(L, R), a);
}
}
int main()
{
int n,m;
int kase=1;
while(scanf("%d%d",&n,&m)==2&&n&&m)
{
prefix_sum[0]=0;
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
prefix_sum[i] = prefix_sum[i-1] + x;
}
build(1,1,n);
printf("Case %d:\n",kase++);
while(m--)
{
int x,y;
scanf("%d%d",&x,&y);
interval a = query(x,y,1,1,n);
printf("%d %d\n",a.first,a.second);
}
}
return 0;
}