洛谷 3097 USACO13DEC 最优挤奶 Optimal Milking

题目：Optimal Milking

思路：
线段树的单点修改区间查询。
每个节点维护4个值，及左右端点选或不选的最大和，用二进制存储，更新时分类讨论。

注意：使用long long类型。

代码：#include <cstdio>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <cstring>
#include <map>
using namespace std;

#define maxn 40000
#define inf (1<<29)
#define ll long long

struct Node {
ll v[5];
Node (ll x=0,ll y=0,ll z=0,ll u=0) {
v[0]=x,v[1]=y,v[2]=z,v[3]=u;
}
};

int n,m;
int c[maxn+5];

Node a[maxn*4+5];
int p,q;

ll Max(ll x1=-inf,ll x2=-inf,ll x3=-inf,ll x4=-inf) {
return max(max(x1,x2),max(x3,x4));
}

ll gt(int o,int x,int y) {
int lc=o*2,rc=o*2+1;
return a[lc].v[x]+a[rc].v[y];
}

void st(int o) {
a[o].v[0]=Max(gt(o,0,0),gt(o,0,2),gt(o,1,0));
a[o].v[1]=Max(gt(o,0,1),gt(o,0,3),gt(o,1,1));
a[o].v[2]=Max(gt(o,2,2),gt(o,2,0),gt(o,3,0));
a[o].v[3]=Max(gt(o,2,1),gt(o,2,3),gt(o,3,1));
}

void make_tree(int o,int l,int r) {
if(l==r) {
a[o]=Node(0,-inf,-inf,c[l]);
return ;
}
int lc=o*2,rc=o*2+1,mid=l+(r-l)/2;
make_tree(lc,l,mid),make_tree(rc,mid+1,r);
st(o);
}

void update(int o,int l,int r) {
if(l>p||r<p) return ;
if(l==r) {
a[o]=Node(0,-inf,-inf,q);
return ;
}
int lc=o*2,rc=o*2+1,mid=l+(r-l)/2;
update(lc,l,mid),update(rc,mid+1,r);
st(o);
}

int main() {
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) scanf("%d",&c[i]);
make_tree(1,1,n);
ll ans=0;
for(int i=1; i<=m; i++) {
scanf("%d%d",&p,&q);
update(1,1,n);
ans+=Max(a[1].v[0],a[1].v[1],a[1].v[2],a[1].v[3]);
}
printf("%lld",ans);
return 0;
}