hdu3911 Black And White--区间更新 & 懒惰标记

原题链接： http://acm.hdu.edu.cn/showproblem.php?pid=3911

一：分析

l0表示从left开始连续是0的个数

l1表示从left开始连续是1的个数

r0表示从right开始连续是0的个数

r1表示从right开始连续是1的个数

m0表示在[ left ,right ]区间内最长连续0的个数

m1表示在[ left ,right ]区间内最长连续1的个数

二：AC代码

#define _CRT_SECURE_NO_DEPRECATE 

#include<iostream>
#include<cmath>
#include<algorithm>

using namespace std;

struct Node
{
	int left, right, mid;
	int l0, l1;
	int r0, r1;
	int m0, m1;
	int len;
	int flag;
}p[100005 * 4];

int a[100005];

/* 向上更新 */
void pushUp(int root)
{
	//左
	p[root].l0 = p[root * 2].l0 + (p[root * 2].l0 == p[root * 2].len ? p[root * 2 + 1].l0 : 0);
	p[root].l1 = p[root * 2].l1 + (p[root * 2].l1 == p[root * 2].len ? p[root * 2 + 1].l1 : 0);

	//右
	p[root].r0 = p[root * 2 + 1].r0 + (p[root * 2 + 1].r0 == p[root * 2 + 1].len ? p[root * 2].r0 : 0);
	p[root].r1 = p[root * 2 + 1].r1 + (p[root * 2 + 1].r1 == p[root * 2 + 1].len ? p[root * 2].r1 : 0);

	//整个区间
	p[root].m0 = max(max(p[root * 2].m0, p[root * 2 + 1].m0), p[root * 2].r0 + p[root * 2 + 1].l0);
	p[root].m1 = max(max(p[root * 2].m1, p[root * 2 + 1].m1), p[root * 2].r1 + p[root * 2 + 1].l1);
}

/* 向下更新 */
void pushDown(int root)
{
	p[root * 2].flag ^= 1;//异或，相同为0，相异为1
	p[root * 2 + 1].flag ^= 1;

	swap(p[root * 2].l0, p[root * 2].l1);
	swap(p[root * 2].r0, p[root * 2].r1);
	swap(p[root * 2].m0, p[root * 2].m1);

	swap(p[root * 2 + 1].l0, p[root * 2 + 1].l1);
	swap(p[root * 2 + 1].r0, p[root * 2 + 1].r1);
	swap(p[root * 2 + 1].m0, p[root * 2 + 1].m1);

	p[root].flag = 0;//子节点已被更新，flag恢复
}

void build(int root, int l, int r)
{
	p[root].left = l;
	p[root].right = r;
	p[root].mid = (l + r) / 2;
	p[root].len = (r - l + 1);
	p[root].flag = 0;

	if (l == r)
	{
		if (a[l] == 1)
		{
			p[root].l1 = p[root].r1 = p[root].m1 = 1;
			p[root].l0 = p[root].r0 = p[root].m0 = 0;
		}
		else
		{
			p[root].l1 = p[root].r1 = p[root].m1 = 0;
			p[root].l0 = p[root].r0 = p[root].m0 = 1;
		}
		return;
	}

	build(root * 2, l, (l + r) / 2);
	build(root * 2 + 1, (l + r) / 2 + 1, r);

	pushUp(root);
}

void update(int root, int l, int r)
{
	if (p[root].left == l&&p[root].right == r)
	{
		swap(p[root].l0, p[root].l1);
		swap(p[root].r0, p[root].r1);
		swap(p[root].m0, p[root].m1);
		p[root].flag ^= 1;
		return;
	}

	if (p[root].flag == 1)//如果上一次没有更新,向下更新，因为下面的递归要用
		pushDown(root);

	if (r <= p[root].mid)
		update(root * 2, l, r);
	else if (l > p[root].mid)
		update(root * 2 + 1, l, r);
	else
	{
		update(root * 2, l, p[root].mid);//心好累啊，这里参数写反了，看了好久才看到
		update(root * 2 + 1, p[root].mid + 1, r);
	}

	pushUp(root);
}

int query(int root, int l, int r)
{
	if (p[root].left == l&&p[root].right == r)
		return p[root].m1;

	if (p[root].flag == 1)//注意判断，向下更新，这才是flag的用处，先不对子节点操作，只要你用到子节点，才去更新它
		pushDown(root);

	if (r <= p[root].mid)
		return query(root * 2, l, r);
	else if (l > p[root].mid)
		return query(root * 2 + 1, l, r);
	else
	{
		int lSum = query(root * 2, l, p[root].mid);
		int rSum = query(root * 2 + 1, p[root].mid + 1, r);
		int mSum = min(p[root * 2].r1, p[root].mid - l + 1) + min(p[root * 2 + 1].l1, r - p[root].mid);

		return max(max(lSum, rSum), mSum);
	}
}

int main()
{
	int n, m;
	int x, l, r;
	while (~scanf("%d", &n))
	{
		for (int i = 0; i < n; i++)
			scanf("%d", &a[i]);

		build(1, 0, n - 1);

		scanf("%d", &m);

		for (int i = 0; i < m; i++)
		{
			scanf("%d%d%d", &x, &l, &r);
			if (x == 1)
				update(1, l - 1, r - 1);
			else
				printf("%d\n", query(1, l - 1, r - 1));
		}
	}
	return 0;
}