LOI 3801 红色的幻想乡

题目描述

经过上次失败后，蕾米莉亚决定再次发动红雾异变，但为了防止被灵梦退治，她决定将红雾以奇怪的阵势释放。
我们将幻想乡看做是一个n*m的方格地区，一开始没有任何一个地区被红雾遮盖。蕾米莉亚每次站在某一个地区上，向东南西北四个方向各发出一条无限长的红雾，可以影响到整行/整列，但不会影响到她所站的那个地区。如果两阵红雾碰撞，则会因为密度过大而沉降消失。灵梦察觉到了这次异变，决定去解决它。但在解决之前，灵梦想要了解一片范围红雾的密度。可以简述为两种操作:
1 x y 蕾米莉亚站在坐标(x,y)的位置向四个方向释放无限长的红雾。
2 x1 y1 x2 y2 询问左上点为(x1,y1)，右下点为(x2,y2)的矩形范围内，被红雾遮盖的地区的数量。

输入输出

输入格式：
第一行三个整数n,m,q，表示幻想乡大小为n*m，有q个询问。
接下来q行，每行3个或5个整数,用空格隔开，含义见题目描述。
输出格式：
对于每一个操作2，输出一行一个整数，表示对应询问的答案。

数据范围

对于20%的数据，1<=n,m,q<=200
对于 40%的数据，1<=n,m,q<=1000
对于100%的数据，1<=n,m,q<=100000

分析

相当于一个二维的RMQ问题，因为是单点修改，所以显然树状数组要好写一点，计算答案的时候不要忘记容斥一下。

#include <bits/stdc++.h>
#define rep( i , l , r ) for( int i = (l) ; i <= (r) ; ++i )
#define per( i , r , l ) for( int i = (r) ; i >= (l) ; --i )
#define erep( i , u ) for( int i = head[(u)] ; ~i ; i = e[i].nxt )
using namespace std;
int _read(){
    char ch = getchar();
    int x = 0 , f = 1 ;
    while( !isdigit( ch ) )
           if( ch == '-' ) f = -1 , ch = getchar();
           else ch = getchar();
    while( isdigit( ch ) )
           x = (ch  - '0') + x * 10 , ch =  getchar();
    return x * f;
}
const int maxn = 100000 + 5;
typedef long long ll;
struct BIT{
    int key[maxn] , N;
    inline int lowbit( int x ){ return x & (-x); }
    void modify( int x , int d ){ for( ; x <= N ; x += lowbit(x) ) key[x] += d; }
    int q( int x ){
        int t = 0;
        for( ; x ; x-= lowbit(x) ) t += key[x];
        return t; 
    }
    int query( int l , int r ){
        return q( r ) - q( l - 1 );
    }
} dx , dy ;
int _x[maxn] , _y[maxn];
void _modify( int x , int y ){
    if( _x[x] == 1 ){ _x[x] = 0 ; dx.modify( x , -1 ); }
    else if( _x[x] == 0 ){ _x[x] = 1 ; dx.modify( x , 1 ) ; }
    if( _y[y] == 1 ){ _y[y] = 0 ; dy.modify( y , -1 ); }
    else if( _y[y] == 0 ){ _y[y] = 1 ; dy.modify( y , 1 ) ; }
}
void calc( int x1 , int y1 , int x2 , int y2 ){
    int s1 = dx.query( x1 , x2 );
    int s2 = dy.query( y1 , y2 );
    ll ans = (ll)s1 * (y2 - y1 + 1) + (ll)s2  * (x2 - x1 + 1) - (ll)s1 * s2 * 2ll;
    printf("%lld\n" , ans);
}
int main(){
    int N = _read() , M = _read() , q = _read();
    dx.N = N , dy.N = M;
    int t , x1 , x2 , y1 , y2;
    while( q-- ){
        t = _read();
        if( t == 1 ){ x1 = _read() , y1 = _read(); _modify( x1 , y1 ); }
        else{
            x1 = _read() , y1 = _read() , x2 = _read() , y2 = _read();
            calc( x1 , y1 , x2 , y2 );
        }
    }
    return 0;
}