Hash表的应用(通过降维，优化DFS)

先看题目

小明最近迷上了一款名为《扫雷》的游戏。

其中有一个关卡的任务如下：

在一个二维平面上放置着 n
个炸雷，第 i
个炸雷 (xi,yi,ri)
表示在坐标 (xi,yi)
处存在一个炸雷，它的爆炸范围是以半径为 ri
的一个圆。

为了顺利通过这片土地，需要玩家进行排雷。

玩家可以发射 m
个排雷火箭，小明已经规划好了每个排雷火箭的发射方向，第 j
个排雷火箭 (xj,yj,rj)
表示这个排雷火箭将会在 (xj,yj)
处爆炸，它的爆炸范围是以半径为 rj
的一个圆，在其爆炸范围内的炸雷会被引爆。

同时，当炸雷被引爆时，在其爆炸范围内的炸雷也会被引爆。

现在小明想知道他这次共引爆了几颗炸雷？

你可以把炸雷和排雷火箭都视为平面上的一个点。

一个点处可以存在多个炸雷和排雷火箭。

当炸雷位于爆炸范围的边界上时也会被引爆。

输入格式

输入的第一行包含两个整数 n、m
。

接下来的 n
行，每行三个整数 xi,yi,ri
，表示一个炸雷的信息。

再接下来的 m
行，每行三个整数 xj,yj,rj
，表示一个排雷火箭的信息。

输出格式

输出一个整数表示答案。

数据范围

对于 40%
的评测用例：0≤x,y≤109,0≤n,m≤103,1≤r≤10
，
对于 100%
的评测用例：0≤x,y≤109,0≤n,m≤5×104,1≤r≤10
。

输入样例：

2 1
2 2 4
4 4 2
0 0 5

输出样例：

2

样例解释

示例图如下，排雷火箭 1
覆盖了炸雷 1
，所以炸雷 1
被排除；炸雷 1
又覆盖了炸雷 2
，所以炸雷 2
也被排除。

直接上代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int M = 999997;
const int Base = 1e9 + 1;
int res;
LL hash1[M]; //hash1用于维护find函数
int hash2[M], hash3[M], book[M]; //hash2:地雷在某点的出现次数 hash3:某点的地雷半径 book:标记该点是否被搜索
//Base进制编码降维
LL get_key(int x, int y) {
	return (LL)x * Base + y;
}
//手写哈希函数
int find(LL x) {
	int t = (x % M + M) % M;
	// 如果 hash1[t] == -1 表示hash表中没有改数据
	// 如果 hash1[t] == x  表示hash表中有改数据 
	while (hash1[t] != -1 && hash1[t] != x) {
		t++;
		if (t == M) t = 0;
	}
	return t;
}
//判断(x2,y2)是否位于(x1,y1,r1)的范围内
bool judge(int x1, int y1, int x2, int y2, int r) {
	int d = (x2 - x1) * (x2 - x1) + (y2 - y1) * (y2 - y1);
	return d <= r * r;
}
//dfs
void dfs(int x, int y, int r) {
	for (int i = -r; i <= r; i++)
		for (int j = -r; j <= r; j++) {
			int dx = x + i;
			int dy = y + j;
			LL t = get_key(dx, dy); // 降维得到的 val  
			// find(t) 表示从查找 hash 表中是否有该值的地雷 
			if (hash2[find(t)] && judge(x, y, dx, dy, r) && !book[find(t)]) {
				res += hash2[find(t)]; //答案加上该点地雷个数
				book[find(t)] = 1; //标记为已搜索
				dfs(dx, dy, hash3[find(t)]); //搜索下一个点
			}
		}
	return;
}

int main() {
	int n, m;
	cin >> n >> m;
	memset(hash1, -1, sizeof hash1); //hash1初始化为空
	for (int i = 0; i < n; i++) {
		int x, y, r;
		cin >> x >> y >> r;
		LL t = get_key(x, y);
		hash1[find(t)] = t; //维护哈希函数
		hash2[find(t)] ++; //统计该点地雷数量
		hash3[find(t)] = max(r, hash3[find(t)]); //记录该点地雷半径的最大值
	}
	for (int i = 0; i < m; i++) {
		int x, y, r;
		cin >> x >> y >> r;
		dfs(x, y, r); //从每一个排雷火箭引爆点开始 dfs
	}
	cout << res << endl;
	return 0;
}