线段树——通过维护区间前缀和后缀的到最终结果类的题目

P6492 [COCI2010-2011#6] STEP

第一次看这道题的时候，我根本就没有想到该如何用线段树去解决这道问题，想用动态规划，但是一直在修改如果每次都进行一次动态规划，那么时间必然会超限的，然后我这个蒟蒻就在题解区里看到了大犇的写法，只能用巧妙来形容

思路：我们在线段树里，用sum去统计这个区间内01序列的最长长度，lsum去表示前缀01序列的最长长度，rsum表示后缀01序列的最长长度 

然后我们需要分析，对于sum来说

1.如果左右两个区间相邻的部分相同，即a[mid]==a[mid+1]，那么sum必然就是左区间的最长前缀或者右区间的最长后缀

2.如果左右两个区间相邻部分不同，即a[mid]!=a[mid+1]，那么我们就要考虑相连的这一部分，我们sum也应该和左区间后缀和+右区间前缀和比较大小

对于lsum来说，如果区间可以合并，并且我的左区间前缀长度就等于左区间的长度，那么

lsum=L（左区间长度）+tree[i*2+1].lsum（右区间的最长前缀）

对于rsum同理

因此我们就可以写出代码：

#include<bits/stdc++.h>  
using namespace std;  
struct node 
{
	int l,r;
    int sum;   // 最大01连续子串长度  
    int lsum;  // 前缀最大01连续子串长度  
    int rsum;  // 后缀最大01连续子串长度  
}tree[200005*4];  
int n, m, dis, a[200005];  

void push_up(int i)
{
	int mid=(tree[i].l+tree[i].r)/2;
	int llen=(mid-tree[i].l+1);
	int rlen=(tree[i].r-mid);
	tree[i].sum=max(tree[i*2].sum,tree[i*2+1].sum);
	tree[i].lsum=tree[i*2].lsum;  
    tree[i].rsum=tree[i*2+1].rsum;  
	if(a[mid]!=a[mid+1])
	{
		tree[i].sum=max(tree[i].sum,tree[i*2].rsum+tree[i*2+1].lsum);
		if(tree[i*2].lsum==llen)
		{
			tree[i].lsum=llen+tree[i*2+1].lsum;
		}
		if(tree[i*2+1].rsum==rlen)
		{
			tree[i].rsum=rlen+tree[i*2].rsum;
		}
	}
}

void build(int i,int l,int r)
{
	tree[i].l=l;
	tree[i].r=r;
	tree[i].sum=1;
	tree[i].lsum=1;
	tree[i].rsum=1;
	if(l==r)
	{
		return ;
	}
	int mid=(l+r)/2;
	build(i*2,l,mid);
	build(i*2+1,mid+1,r);
	push_up(i);
}

void update(int i,int dis)
{
	if(tree[i].l==tree[i].r)
	{
		a[tree[i].l]^=1;
		return ;
	}
	int mid=(tree[i].l+tree[i].r)/2;
	if(dis<=mid)
	{
		update(i*2,dis);
	}
	if(dis>mid)
	{
		update(i*2+1,dis);
	}
	push_up(i);
	return ;
}

signed main()
{
	cin>>n>>m;
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		cin>>dis;
		update(1,dis);
		cout<<tree[1].sum<<"\n";
	}
	return 0;
}