Codeforces Round #439 (Div. 2) E. The Untended Antiquity hash+二维树状数组或二维线段树

题目链接: E. The Untended Antiquity

题目大意

一个n*m的网格, 有三种操作, 一是将包含(x1, y1)(x2, y2)两点的最小矩形套上障碍, 二是将某个障碍取消, 三是查询点(x1, x2)(y1, y2)之间是否能直接到达(之间没有障碍)

思路

题解的思路非常6
利用二维线段树, 对于每个障碍对应一个随机的hash值, 然后操作一则让所有矩形内所有点都加上这个值, 操作二则减去这个值, 操作三则求出两点的值, 如果相同说明在同一个障碍内, 可以连通, 反之则反
也可以用异或(加一个值变成异或一个值, 写起来好一点)
如果只用一个unsigned long long 有很大几率会出现碰撞(WA), 用两个ull就好了
因为是区间修改, 单点查询, 所以用二维树状数组写起来方便

代码

GNU C++14 Accepted 187 ms 99000 KB

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2500 + 10, inf = 0x3f3f3f3f;
typedef unsigned long long ull;
typedef pair<ull, ull> pll;
pll bit[maxn][maxn];
int n;
inline int lowbit(int x) { return x&(-x); }
pll sum(int x, int y)
{
    pll s = pll((ull)0, (ull)0);
    for(int i=x; i>0; i-=lowbit(i))
        for(int j=y; j>0; j-=lowbit(j))
        {
            s.first ^= bit[i][j].first;
            s.second ^= bit[i][j].second;
        }
    return s;
}
void add(int x, int y, pll v)
{
    for(int i=x; i<=maxn; i+=lowbit(i))
        for(int j=y; j<=maxn; j+=lowbit(j))
        {
            bit[i][j].first ^= v.first;
            bit[i][j].second ^= v.second;
        }
}
void update(int x1, int y1, int x2, int y2, pll v)
{
    add(x1, y1, v);
    add(x2+1, y2+1, v);
    add(x1, y2+1, v);
    add(x2+1, y1, v);
}
int m, q, t, a, b, c, d;
pll v;
typedef pair<int, int> P;
map<pair<P, P>, pll> mp;

int main()
{
    srand(time(0));
    scanf("%d%d%d", &n, &m, &q);
    while(q--)
    {
        scanf("%d%d%d%d%d", &t, &a, &b, &c, &d);
        if(t==1)
        {
            v.first = (ull)rand()*rand()*rand();
            v.second = (ull)rand()*rand()*rand();
            mp[make_pair(P(a, b), P(c, d))] = v;
            update(a, b, c, d, v);
        }
        else if(t==2)
        {
            v = mp[make_pair(P(a, b), P(c, d))];
            update(a, b, c, d, v);
        }
        else 
        {
            if(sum(a, b) == sum(c, d)) puts("Yes");
            else puts("No");
        }
    }
    return 0;
}