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;
}