C. Rooks Defenders #791 div2

Problem - C - Codeforces

题意就是给你一个n*n的矩阵，有很多询问：

①在（x,y）里面加入一个车

②在（x,y）里面删掉一个车（保证原来就有）

③是否子矩阵里面所有的单元格都可以被攻击到

分析：

这是一个有点板的线段树吧，加了一点点不同好像

明确一点，在子矩阵里面如果所有的单元格都可以被攻击到的充要条件是：

至少那个子矩阵的行或者列，都必须有一个车在不然不可以

所以就转化为了求一段区间的车的数量，如果行或者列的车的数量 == 子矩阵的行或者列

‘就yes，否则no

要看行还是列直接写两个线段树来维护就可以了

线段树相当于是提前算好了区间里面你想要的东西，比如这题就是区间里面车的数量（重合的不算）。还可以支持增加或者删掉操作（单点修改）

下面看代码：

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#define IOS ios::sync_with_stdio(false), cin.tie(0);
#include<iostream>
#include<map>
#include<set> 
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
#include<algorithm>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int,int> PAII;
const int N=1e5+10,M=5050,INF=1e18,mod=998244353;
int n,q;
struct Node{
	int l,r,sum,v;
}tr[N*4],trr[N*4];
void pushup(Node tr[],int u)
{
	tr[u].sum = tr[u<<1].sum + tr[u<<1|1].sum;
}
void build(Node tr[],int u,int l,int r)
{
	tr[u] = {l,r};
	if(l == r) return ;
	int mid = l + r >> 1;
	build(tr, u << 1, l, mid), build(tr, u <<1|1, mid + 1, r);
}
void modify(Node tr[], int u, int x, int v)
{
	if(tr[u].l == tr[u].r)
	{
		tr[u].v += v;
		tr[u].sum = (tr[u].v > 0);
		return ;
	}
	else
	{
		int mid = tr[u].l + tr[u].r >> 1;
		if(x <= mid) modify(tr, u<<1, x, v);
		else modify(tr, u<<1|1, x, v);
		pushup(tr, u);
	}
}
int query(Node tr[], int u, int l, int r)
{
	if(l <= tr[u].l && r >= tr[u].r) return tr[u].sum;
	int mid = tr[u].l + tr[u].r >> 1;
	int sum = 0;
	if(l <= mid) sum += query(tr, u << 1, l, r);
	if(r > mid) sum += query(tr, u<<1|1, l, r);
	return sum;
}
signed main(){
    IOS;
    int T;
    T=1;
   // cin>>T;
    while(T--)
    {
		cin>>n>>q;
		build(tr, 1, 1, n), build(trr, 1, 1, n);
		while(q --)
		{
			int t;
			cin >> t;
			if(t == 1)
			{
				int x,y;
				cin >> x >> y;
				modify(tr, 1, x, 1);
				modify(trr, 1, y, 1);
			}
			if(t == 2)
			{
				int x,y;
				cin >> x >> y;
				modify(tr, 1, x, -1);
				modify(trr, 1, y, -1);
			}
			if(t == 3)
			{
				int x1,y1,x2,y2;
				cin >> x1 >> y1 >> x2 >> y2;
				if(query(tr, 1, x1, x2) == x2 - x1 + 1 || query(trr, 1, y1, y2) == y2 - y1 + 1)
					cout<<"YES\n";
				else cout<<"NO\n";
			}
		}
    }
    return 0;
} 
/* 



*/