ZCMU-1156:新年彩灯I（树状数组）

1156: 新年彩灯Ⅰ

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 517  Solved: 72
[Submit][Status][Web Board]

Description

新年将至，YY准备挂一排彩灯，已知彩灯刚挂完的彩灯共有N盏（编号为1,2,3,……），并且都是灭的。彩灯的闪烁由一段程序控制。

每一秒钟程序会生成两个正整数a和b（1<=a,b<=N），然后将编号为a和b之间的所有灯的状态改变一次，即如果灯i是灭的，那么经过一次改变，灯i会亮，如果灯i是亮的，经过一次改变，灯i会灭。

 

当YY看着自己挂的彩灯不断闪烁的时候，问题来了，YY想知道任意时刻某一区间灯的状态。

 

Input

多组测试数据，每一组第一行是一个整数N（1<=N<=1000000）和一个整数M(1<=M<=3000)。

 

然后是M行数据，包括以下两种形式：

         1 a b 表示灯a和灯b之间的灯（含灯a和灯b）变换一次状态。

         0 x y 表示YY想知道此刻灯x到灯y（包含灯x和灯y）的状态.

 

Output

对于每次YY想知道结果的时候，输出一行灯的状态（编号小的灯优先），如果是亮的输出”1”，否则输出”0”;

 

Sample Input

3 3

1 1 2

1 2 3

0 1 3

Sample Output

101

HINT

Source

【解析】

很明显这题用普通的循环肯定会超时，所以可以采用树状数组的方法。

树状数组讲解

 

 

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1000005;
int tree[maxn];
int n, m;
void update(int p, int flag)
{
	while (p <= n)
	{
		tree[p] += flag;
		int temp = p & -p;//调试（即lowbit）
		p += temp;
	}
}
int query(int p)
{
	int res = 0;
	while (p)
	{
		res += tree[p];
		int temp = p & -p;
		p -= temp;
	}
	return res;
}

int main()
{
	while (~scanf("%d%d",&n,&m))
	{
		int flag, a, b;
		memset(tree, 0, sizeof(tree));
		for (int i = 0; i<m; i++)
		{
			scanf("%d%d%d",&flag, &a, &b);
			if (a>b)
				swap(a, b);
			if (flag)
			{
				update(a, flag);
				update(b + 1, -flag);
			}
			else
			{
				for (int j = a; j <= b; j++)
				{
					int t = query(j);
					if (t % 2)
						printf("1");
					else
						printf("0");

				}
				printf("\n");
			}
		}

	}
	return 0;
}