UVA-1400 Ray,Pass me the Dishes,LA 3938,セグメントツリー,区間クエリー
3623 ワード
題意:一列数(n個)、m次クエリ区間[l,r]の最大連続区間[x,y](l<=x<=y<=r)を与える.(n,m<=500 000)
構想:ダイナミッククエリー区間最大連続区間;
最大連続区間とを求める場合:
セグメントツリーで最大連続とsum_を維持するsub、最大接頭辞、sum_prefix、最大接尾辞、sum_suffix.
root.sum_sub = max{l.sum_sub, r.sum_sub, (l.sum_suffix + r.sum_prefix) };
タイトル要求区間、類似:
最大連続区間max_を線分ツリーで維持するsub、最大接頭辞右端max_prefix、最大接尾辞左端max_suffix.
詳細は、コードを参照してください.
構想:ダイナミッククエリー区間最大連続区間;
最大連続区間とを求める場合:
セグメントツリーで最大連続とsum_を維持するsub、最大接頭辞、sum_prefix、最大接尾辞、sum_suffix.
root.sum_sub = max{l.sum_sub, r.sum_sub, (l.sum_suffix + r.sum_prefix) };
タイトル要求区間、類似:
最大連続区間max_を線分ツリーで維持するsub、最大接頭辞右端max_prefix、最大接尾辞左端max_suffix.
詳細は、コードを参照してください.
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define lc rt<<1
#define rc rt<<1|1
const int maxn = 500000 + 5;
typedef long long LL;
LL num[maxn], max_prefix[maxn<<2], max_suffix[maxn<<2];
struct node{
int l, r;
node(int ll=0, int rr=0):l(ll),r(rr){}
} max_sub[maxn<<2];
LL prefix_sum[maxn];
int ql, qr;
LL sum(int l, int r){
return prefix_sum[r] - prefix_sum[l-1];
}
LL sum(node a){
return sum(a.l, a.r);
}
node better(node a, node b){
if(sum(a) != sum(b)) return sum(a) > sum(b) ? a:b;
return (a.l<b.l||(a.l==b.l&&a.r<b.r))? a:b;
}
void build(int rt, int l, int r){
if(l==r){
max_prefix[rt] = max_suffix[rt] = l;
max_sub[rt] = node(l,l);
return ;
}
int m = (l+r)>>1;
build(lc, l, m);
build(rc, m+1, r);
LL v1 = sum(l, max_prefix[lc]);
LL v2 = sum(l, max_prefix[rc]);
if(v1 == v2) max_prefix[rt] = min(max_prefix[lc], max_prefix[rc]);
else max_prefix[rt] = v1 > v2 ? max_prefix[lc] : max_prefix[rc];
v1 = sum(max_suffix[lc], r);
v2 = sum(max_suffix[rc], r);
if(v1 == v2) max_suffix[rt] = min(max_suffix[lc], max_suffix[rc]);
else max_suffix[rt] = v1 > v2 ? max_suffix[lc] : max_suffix[rc];
max_sub[rt] = better(max_sub[lc], max_sub[rc]);
max_sub[rt] = better(max_sub[rt], node(max_suffix[lc], max_prefix[rc]));
}
int query_prefix(int rt, int l, int r){
if(qr >= max_prefix[rt]) return max_prefix[rt];
int m = (l+r)>>1;
//l<=qr<=m
if(qr <= m) return query_prefix(lc, l, m);
//m+1<=qr<=r
int rr = query_prefix(rc, m+1, r);
node ret = better(node(l,rr), node(l, max_prefix[lc]));
return ret.r;
}
int query_suffix(int rt, int l, int r){
if(ql <= max_suffix[rt]) return max_suffix[rt];
int m = (l+r)>>1;
//m+1<=ql<=r
if(ql > m) return query_suffix(rc, m+1, r);
//l<=ql<=m
int ll = query_suffix(lc, l, m);
node ret = better(node(ll, r), node(max_suffix[rc], r));
return ret.l;
}
node query(int rt, int l, int r){
if(ql <= l && r <= qr) return max_sub[rt];
int m = (l+r)>>1;
if(qr <= m) return query(lc, l, m);
if(ql > m) return query(rc, m+1, r);
//ql <= m <= qr
int ll = query_suffix(lc, l, m); //l_max_suffix
int rr = query_prefix(rc, m+1, r); //r_max_prefix
node mid = node(ll, rr);
node sub = better( query(lc, l, m), query(rc, m+1, r));
return better( mid, sub);
}
int main()
{
int n, m, i, cas = 1, l, r;
while(~scanf("%d%d", &n, &m)){
printf("Case %d:
", cas++);
prefix_sum[0] = 0;
for(i=1; i<=n; ++i) {
scanf("%lld", &num[i]);
prefix_sum[i] = prefix_sum[i-1] + num[i];
}
node v = node(1,3);
build(1, 1, n);
while(m--){
scanf("%d%d", &l, &r);
ql = l; qr = r;
node ans = query(1, 1, n);
printf("%d %d
", ans.l, ans.r);
}
}
return 0;
}