刷题记录:牛客NC15172情人节的电灯泡

传送门:牛客

题目描述:

情人节到了，小芳和小明手牵手，打算过一个完美的情人节，但是小刚偏偏也来了，当了一
个明晃晃的电灯泡，小明很尴尬，就和小刚说，我交给你个任务，你完成了我俩就带你玩，
否则你就回家吧。小刚很有当单身狗的觉悟，他坚决不想让小明过好情人节，同为单身狗的
你能帮帮他吗？现在有一个n×n（1 <= n <= 1000）的格子，每一个格子都有一个电灯泡，可
能是亮的，也可能是灭的（1代表亮， 0代表灭），现在有两种操作，一种是给你一个坐标，
对于那个坐标上的灯泡，如果他是亮的，那么熄灭他，反之如果他是灭的，那么打开它。第
二种操作是给你两个坐标，第一个坐标代表一个子矩阵的左上角，另一个坐标是右下角，请
你求出当前子矩阵中有多少个灯泡是亮着的。燥起来吧！！！单身狗们！！！！
输入:
6 4
0 0 1 0 1 0
1 0 1 1 0 1
1 0 0 0 0 0
1 1 0 0 1 0
0 0 0 0 1 1
0 0 0 0 1 0
2 2 2 4 5
1 1 1
2 1 1 6 6
1 2 6
输出:
4
14

线段树维护单点修改+区间和

本题是二维上的修改+查询区间和.对于这种情况,我们可以直接建二维的线段树,也可以将二维拆成一维的,两种方法都是可以的.但是鉴于二维线段树写起来代码量会增多不少,所以本题博主使用的是拆成一维的方法.

那么对于$(x,y)$的单点修改来说,此时我们修改$(x-1)\ast n+y$即可.
对于$(x1,y1),(x2,y2)$的求和来说,此时我们可以进行一个for循环来遍历每一行然后对于每一行来说,我们可以使用线段树来返回连续的列的和.总体复杂度为$n\ast m\ast logn$,可以过此题

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
struct Segmrnt_tree{
	int l,r,sum;
}tree[maxn*4];
Segmrnt_tree operator + (Segmrnt_tree l,Segmrnt_tree r) {
	Segmrnt_tree u;
	u.l=l.l;u.r=r.r;
	u.sum=l.sum+r.sum;
	return u;
}
int n,m;
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;
	if(l==r) return ;
	int mid=(l+r)>>1;
	build(lson);build(rson);
	tree[rt]=tree[ls]+tree[rs];
}
void update(int pos,int rt) {
	if(tree[rt].l==pos&&tree[rt].r==pos) {
		tree[rt].sum^=1;
		return ;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(pos<=mid) update(pos,ls);
	else update(pos,rs);
	tree[rt]=tree[ls]+tree[rs];
}
int query(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		return tree[rt].sum;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) return query(l,r,ls);
	else if(l>mid) return query(l,r,rs);
	else return query(l,mid,ls)+query(mid+1,r,rs);
}
int main() {
	n=read();m=read();
	build(1,n*n,1);
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=n;j++) {
			int a=read();
			if(a==1) update((i-1)*n+j,1);
		}
	}
	for(int i=1;i<=m;i++) {
		int opt=read();
		if(opt==1) {
			int x=read(),y=read();
			update((x-1)*n+y,1);
		}
		else {
			int x1=read(),y1=read(),x2=read(),y2=read();
			int ans=0;
			for(int i=0;i<=x2-x1;i++) {
				ans+=query((x1+i-1)*n+y1,(x1+i-1)*n+y2,1);
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}