【bzoj4094】[Usaco2013 Dec]Optimal Milking 线段树

好吧,承认我脑子有洞,竟然往dp上想。

线段树暴力维护一下四个量

只选左端点,只选右端点,两个都不选,两个都选


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#define maxn 40010 

using namespace std;

struct yts
{
	int l,r;
	long long lx,rx,x2,x0,ans;
}t[4*maxn];

int a[maxn];
int n,m,tot,T;
long long ans;

void update(int i)
{
	int lch=i*2,rch=i*2+1;
	t[i].lx=max(t[lch].x2+t[rch].x0,max(t[lch].lx+t[rch].lx,t[lch].lx+t[rch].x0));
	t[i].rx=max(t[rch].x2+t[lch].x0,max(t[rch].rx+t[lch].rx,t[rch].rx+t[lch].x0));
	t[i].x0=max(t[lch].x0+t[rch].x0,max(t[lch].x0+t[rch].lx,t[rch].x0+t[lch].rx));
	t[i].x2=max(t[lch].x2+t[rch].rx,max(t[lch].lx+t[rch].x2,t[lch].lx+t[rch].rx));
	t[i].ans=max(max(t[i].lx,t[i].rx),max(t[i].x0,t[i].x2));
}

void build(int i,int l,int r)
{
	t[i].l=l;t[i].r=r;
	if (l==r) {t[i].x2=a[l];return;}
	int mid=(l+r)/2;
	build(i*2,l,mid);build(i*2+1,mid+1,r);
	update(i);
}

void modify(int i,int x,int d)
{
	if (t[i].l==t[i].r) {t[i].x2=t[i].ans=d;return;}
	int mid=(t[i].l+t[i].r)/2;
	if (x<=mid) modify(i*2,x,d);
	if (mid<x) modify(i*2+1,x,d);
	update(i);
}

int main()
{
	scanf("%d%d",&n,&T);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	build(1,1,n);
	while (T--)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		modify(1,x,y);
		ans+=t[1].ans;
	}
	printf("%lld\n",ans);
	return 0;
}


你可能感兴趣的:(【bzoj4094】[Usaco2013 Dec]Optimal Milking 线段树)