codevs 1690 开关灯 线段树 解题报告

彩蛋

题目描述 Description

YYX家门前的街上有N(2<=N<=100000)盏路灯，在晚上六点之前，这些路灯全是关着的，六点之后，会有M(2<=m<=100000)个人陆续按下开关，这些开关可以改变从第i盏灯到第j盏灯的状态，现在YYX想知道，从第x盏灯到第y盏灯中有多少是亮着的(1<=i,j,x,y<=N)

输入描述 Input Description

第 1 行: 用空格隔开的两个整数N和M

第 2..M+1 行: 每行表示一个操作, 有三个用空格分开的整数: 指令号(0代表按下开关，1代表询问状态), x 和 y

输出描述 Output Description

第 1..询问总次数 行:对于每一次询问，输出询问的结果

样例输入 Sample Input

4 5

0 1 2

0 2 4

1 2 3

0 2 4

1 1 4

样例输出 Sample Output

1

2

数据范围及提示 Data Size & Hint

一共4盏灯，5个操作，下面是每次操作的状态(X代表关上的，O代表开着的)：

XXXX -> OOXX -> OXOO -> 询问1~3 -> OOXX -> 询问1~4

思路

这种“区间复制修改查询”类题目，肯定是线段树。

打好修改标记就好了。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
struct os
{
int yes,total;
bool flag;
}tree[400010];
int n,m;
int k,head,tail;
void f(int root)
{
tree[root].yes=tree[root].total-tree[root].yes;
}
void build(int root,int begin,int end)
{
if (begin>end) return;
tree[root].total=end-begin+1;
if (begin==end) return;
int mid=(begin+end)/2;
build(root*2,begin,mid);
build(root*2+1,mid+1,end);
}
void pushdown(int root)
{
if (!tree[root].flag) return;
tree[root*2].flag=!tree[root*2].flag;
tree[root*2+1].flag=!tree[root*2+1].flag;
f(root*2);
f(root*2+1);
tree[root].flag=0;
}
void update(int root,int begin,int end)
{
if (begin>=head&&end<=tail) {f(root);tree[root].flag=!tree[root].flag;return;}
pushdown(root);
int mid=(begin+end)/2;
if (head<=mid) update(root*2,begin,mid);
if (tail>mid)  update(root*2+1,mid+1,end);
tree[root].yes=tree[root*2].yes+tree[root*2+1].yes;
}
int get(int root,int begin,int end)
{
if (begin>=head&&end<=tail) return tree[root].yes;
pushdown(root);
int mid=(begin+end)/2,ans=0;
if (head<=mid) ans+=get(root*2,begin,mid);
if (tail>mid)  ans+=get(root*2+1,mid+1,end);
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
build(1,1,n);
for (int i=1;i<=m;i++)
{
scanf("%d%d%d",&k,&head,&tail);
if (k==0) update(1,1,n);
else printf("%d\n",get(1,1,n));
}
return 0;
}