UVA1400 Ray, Pass me the dishes!" 【线段树 区间合并】

"Ray, Pass me the dishes!" UVA - 1400 

https://vjudge.net/problem/UVA-1400

题意

给出一个长度为n的整数序列D，对m个询问做出回答，对询问(a,b)找到(x,y)使得a<=x<=y<=b且Dx+Dx+1+……+Dy最大。如有多组答案取字典序最小的一组。

题解

sum[i]记录结点i控制的区间[l,r]中区间和最大的区间 
lsum[i]记录结点i控制的区间[l,r]中,从l开始往后最大连续区间和的区间右端点 
rsum[i]记录结点i控制的区间[l,r]中，从r开始往前最大连续区间和的区间左端点 
pre[N]存放初始数组的前缀和

C++代码

#include<iostream>
#include<utility>
#include<algorithm>

using namespace std;

#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1

typedef long long ll;
typedef pair<int,int> interval;

const int N=500500;

interval sum[N<<2];//sum[i]记录结点i控制的区间[l,r]中区间和最大的区间 
int lsum[N<<2];//lsum[i]记录结点i控制的区间[l,r]中,从l开始往后最大连续区间和的区间右端点 
int rsum[N<<2];//rsum[i]记录结点i控制的区间[l,r]中，从r开始往前最大连续区间和的区间左端点 
ll pre[N];//存放初始数组的前缀和

//获取区间[l,r]的元素和 
ll get_sum(int l,int r)
{
	return pre[r]-pre[l-1];
}

ll get_sum(interval a)
{
	return get_sum(a.first,a.second);
}

//比较两个区间的区间和，返回区间和大的区间
//如果区间和一样，返回左端点小的，左端点一样则返回右端点小的 
interval better(interval a,interval b)
{
	ll v1=get_sum(a),v2=get_sum(b);
	if(v1==v2)
	  return a<b?a:b;
	else
	  return v1<v2?b:a;
}

//根据左右孩子结点的信息更新父亲结点的信息 
void pushup(int l,int r,int rt)
{
	//最大和的区间
	sum[rt]=better(sum[rt<<1],sum[rt<<1|1]); 
	sum[rt]=better(sum[rt],make_pair(rsum[rt<<1],lsum[rt<<1|1]));
	
	//前缀延深的最长距离（从l位置往后），记录右端点 
	ll v1=get_sum(l,lsum[rt<<1]);
	ll v2=get_sum(l,lsum[rt<<1|1]);
	lsum[rt]=(v2>v1?lsum[rt<<1|1]:lsum[rt<<1]);
	
	//后缀延深的最长距离（从r位置往前），记录左端点
	v1=get_sum(rsum[rt<<1|1],r);
	v2=get_sum(rsum[rt<<1],r);
	rsum[rt]=(v2>=v1?rsum[rt<<1]:rsum[rt<<1|1]); 
}

void build(int l,int r,int rt)
{
	if(l==r)
	{
		sum[rt]=make_pair(l,r);
		lsum[rt]=rsum[rt]=l;
		return;
	}
	int m=(l+r)>>1;
	build(ls);
	build(rs);
	pushup(l,r,rt);
}

//调用此函数的前提是l在区间[L,R]中
//求以l为起始元素，终止元素也在[L,R]中其值最大的连续字段和的区间 
interval lquery(int L,int R,int l,int r,int rt) 
{
	if(lsum[rt]<=R) return make_pair(l,lsum[rt]);
	int m=(l+r)>>1;
	if(R<=m) return lquery(L,R,ls);
	interval a=lquery(L,R,rs);
	a.first=l;
	return better(a,make_pair(l,lsum[rt<<1]));
}

//调用此函数的前提是r在区间[L,R]中
//求以r为终止元素，起始元素也在[L,R]中其值最大的连续字段和的区间 
interval rquery(int L,int R,int l,int r,int rt) 
{
	if(rsum[rt]>=L) return make_pair(rsum[rt],r);
	int m=(l+r)>>1;
	if(m<L) return rquery(L,R,rs);
	interval a=rquery(L,R,ls);
	a.second=r;
	return better(a,make_pair(rsum[rt<<1|1],r));
}

interval query(int L,int R,int l,int r,int rt)
{
	if(L<=l&&r<=R) return sum[rt];
	int m=(l+r)>>1;
	if(R<=m) return query(L,R,ls);//在左子树 
	if(m<L) return query(L,R,rs);//在右子树
	//在中间
	int a=rquery(L,R,ls).first;
	int b=lquery(L,R,rs).second;
	//printf("a=%d b=%d\n",a,b);
	interval ans=better(query(L,R,ls),query(L,R,rs));//最大子段和完全在左边和完全在右边
	//最大子段和在中间
	return better(ans,make_pair(a,b)); 
}

int main()
{
	int n,m,t=1,x,L,R;
	while(~scanf("%d%d",&n,&m)) 
	{
		pre[0]=0;//前缀和
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&x);
			pre[i]=pre[i-1]+x;
		}
		build(1,n,1);
		printf("Case %d:\n",t++);
		while(m--)
		{
			scanf("%d%d",&L,&R);
			interval ans=query(L,R,1,n,1);
			printf("%d %d\n",ans.first,ans.second);
		 } 
	}
	return 0;
}

标签：me,rt,dishes,rsum,int,UVA1400,区间,lsum,include
来源： https://blog.csdn.net/SongBai1997/article/details/89927033