POJ2750 Potted Flower 线段树+DP

这题是求一个圈中取出若干相邻的数,求其中的最大值,不能够同时取所有的数。

本来想着要扩张出一个圈出来,再通过限定长度来转化为线段上的问题,但是这样明显就复杂很多了,有一个结论就是当这个区间在[1-N]之间的话,那么直接输出最大值,否则一定是sum - min[1-N],也就是说如果区间跨越了1,N两个点,那么[1-N]中就一定隐藏了最小子串和。很好理解,因为整个循环串就是由一个最大子串和一个最小子串组成的。

线段树的节点中要存储较多的值:

lmax -- 从左边开始至少取一个节点的最大值
lmin  -- 从左边开始至少取一个节点的最小值
rmax -- 从右边开始至少取一个节点的最大值
rmin  -- 从右边开始至少取一个节点的最小值
mmax -- 整段区间的最大值
mmin  -- 整段区间的最小值
sum   --  整段区间的和

维护好这些值即可

代码如下:

#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define MAXN 100005
using namespace std;

struct Node {
    int l, r, lmin, rmin, mmin;
    int lmax, rmax, mmax, sum;
}s[MAXN<<2];

int seq[MAXN];

void push_up(int p) {
    s[p].sum = s[p<<1].sum + s[p<<1|1].sum;
    s[p].lmax = max(s[p<<1].lmax, s[p<<1].sum+s[p<<1|1].lmax);
    s[p].lmin = min(s[p<<1].lmin, s[p<<1].sum+s[p<<1|1].lmin);
    s[p].rmax = max(s[p<<1|1].rmax, s[p<<1|1].sum+s[p<<1].rmax);
    s[p].rmin = min(s[p<<1|1].rmin, s[p<<1|1].sum+s[p<<1].rmin);
    // 从五个部分推过来    
s[p].mmax = max(s[p].lmax, s[p].rmax); s[p].mmax = max(s[p].mmax, s[p<<1].mmax); s[p].mmax = max(s[p].mmax, s[p<<1|1].mmax); s[p].mmax = max(s[p].mmax, s[p<<1].rmax+s[p<<1|1].lmax); s[p].mmin = min(s[p].lmin, s[p].rmin); s[p].mmin = min(s[p].mmin, s[p<<1].mmin); s[p].mmin = min(s[p].mmin, s[p<<1|1].mmin); s[p].mmin = min(s[p].mmin, s[p<<1].rmin+s[p<<1|1].lmin); } void build(int p, int l, int r) { s[p].l = l, s[p].r = r; if (l != r) { int mid = (l + r) >> 1; build(p<<1, l, mid); build(p<<1|1, mid+1, r); push_up(p); } else { s[p].mmax = s[p].mmin = s[p].sum = seq[r]; s[p].lmin = s[p].rmin = s[p].lmax = s[p].rmax = seq[r]; } } void modify(int p, int x) { if (s[p].l == s[p].r) { s[p].mmax = s[p].mmin = s[p].sum = seq[s[p].r]; s[p].lmin = s[p].rmin = s[p].lmax = s[p].rmax = seq[s[p].r]; } else { int mid = (s[p].l + s[p].r) >> 1; if (x <= mid) modify(p<<1, x); else modify(p<<1|1, x); push_up(p); } } int main() { int N, M, x, y; while (scanf("%d", &N) == 1) { for (int i = 1; i <= N; ++i) { scanf("%d", &seq[i]); } build(1, 1, N); scanf("%d", &M); while (M--) { scanf("%d %d", &x, &y); seq[x] = y; modify(1, x); if (s[1].mmax == s[1].sum) { printf("%d\n", s[1].sum-s[1].mmin); } else { printf("%d\n", max(s[1].mmax, s[1].sum-s[1].mmin)); } } } return 0; }
原文地址:https://www.cnblogs.com/Lyush/p/2694971.html