1. 程式人生 > >4301 Can you answer on these queries III 0x40「資料結構進階」例題(線段樹)

4301 Can you answer on these queries III 0x40「資料結構進階」例題(線段樹)

4301 Can you answer on these queries III 0x40「資料結構進階」例題

描述

給定長度為N的數列A,以及M條指令 (N≤500000, M≤100000),每條指令可能是以下兩種之一:
2 x y”,把 A[x] 改成 y。
1 x y”,查詢區間 [x,y] 中的最大連續子段和,即 max(x≤l≤r≤y)⁡ { ∑(i=l~r) A[i] }。
對於每個詢問,輸出一個整數表示答案。

輸入格式

第一行兩個整數N,M

第二行N個整數Ai

接下來M行每行3個整數k,x,y,k=1表示查詢(此時如果x>y,請交換x,y),k=2表示修改

輸出格式

對於每個詢問輸出一個整數表示答案。

樣例輸入

5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 3 2

樣例輸出

2
-1

資料範圍與約定

  • 對於100%的資料: N≤500000, M≤100000, |Ai|<=1000

線段樹板子啊。線段樹程式碼風格要改~ 結構體有他的好處。。。。

 

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define oo cout<<"!!!"<<endl;
typedef long long ll;
typedef unsigned long long ull;
#define ms(s) memset(s, 0, sizeof(s))
const int inf = 0x3f3f3f3f;
//head

const int maxn = 2e6+1111;

struct node
{
	int l,r,dat,sum,dl,dr;
}t[maxn];
int a[maxn];

void pushup(int i)
{
	t[i].sum = t[i<<1].sum + t[i<<1|1].sum;
	t[i].dl = max(t[i<<1].dl,t[i<<1].sum + t[i<<1|1].dl);
	t[i].dr = max(t[i<<1|1].dr,t[i<<1|1].sum + t[i<<1].dr);
	t[i].dat = max(t[i<<1].dat,t[i<<1|1].dat);
	t[i].dat = max(t[i].dat,t[i<<1].dr + t[i<<1|1].dl);
}
void build(int i,int l,int r)
{
	t[i].l = l;t[i].r = r;
	if(l == r){int x;scanf("%d",&x);t[i].sum = t[i].dat = t[i].dl = t[i].dr = x;return;} 
	int mid = (l+r)>>1;
	build(i<<1,l,mid);
	build(i<<1|1,mid+1,r);
	pushup(i);
}

void update(int x,int v,int i)
{
	if(t[i].l == t[i].r){t[i].sum = t[i].dat = t[i].dl = t[i].dr = v;return;}
	int mid = (t[i].l + t[i].r)>>1;
	if(x <= mid)update(x,v,i<<1);
	else update(x,v,i<<1|1);
	pushup(i);
}


node query(int ql,int qr,int i)	
{
	if(ql <= t[i].l && qr >= t[i].r)return t[i];
	int mid = (t[i].l + t[i].r)>>1;
	if(qr <= mid)return query(ql,qr,i<<1);
	if(ql > mid)return query(ql,qr,i<<1|1);
	node a = query(ql,qr,i<<1);
	node b = query(ql,qr,i<<1|1);
	node c;
	c.sum = a.sum + b.sum;
	c.dl = max(a.dl,a.sum + b.dl);
	c.dr = max(b.dr,b.sum + a.dr);
	c.dat = max(max(a.dat,b.dat),a.dr + b.dl);
	return c;
}

int main() 
{
	int n,q;
	cin>>n>>q;
	build(1,1,n);
	rep(i,1,q+1)
	{
		int k,x,y;
		scanf("%d%d%d",&k,&x,&y);
		if(k == 2)
			update(x,y,1);
		else 
		{
			if(y<x)swap(x,y);
			printf("%d\n",query(x,y,1).dat);
		}
	}
    return 0;
}