Problem D: 新年彩灯Ⅰ(树状数组:区间更新,单点查询)

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

分析:

数据量超级大,如果挨个遍历一遍,铁定超时,所以需要树状数组优化。题意要输出一段区间,每次也是更新一段区间。所以是差分树状数组,区间更新,单点查询,套板子就行了,但是还有个小坑,具体看代码。

 ac代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include<string.h>
using namespace std;
int a[1000009];
int n,m;
void update(int p, int x)         //模板 更新 
{ 
    while (p <= n) a[p] += x, p += p & -p;
}
void range_update(int l, int r, int x)  //区间更新 
{
    update(l, x);
    update(r + 1, -x);
}
int sum(int p)  //单点查询 
{
    int res = 0;
    while (p) res += a[p], p -= p & -p;
    return res;
}
int main()
{
	 while(scanf("%d%d",&n,&m)!=EOF)
    {
    	int flag,l,r;
        memset(a, 0, sizeof(a));
        for (int i = 1; i <= m; i++)
        {
        	scanf("%d%d%d",&flag,&l,&r);
        	int t;
        	if(l>r) {t=l;l=r;r=t;}   //坑!!l和r大小不确定 
        	if(flag==1)
			range_update(l,r,1);        //变换状态,更新lr区间内所有的数加一
			else
			{
				for(int j=l;j<=r;j++)
				if(sum(j)%2==0) printf("0");//如果是偶数,说明灯是灭的 
				else printf("1");//奇数 亮的 
				printf("\n");
			} 
        	
}
}
return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_42079027/article/details/81292225
今日推荐