写在前面
一个混子研究生写最烂的代码,写思路的时候像碎碎念,展现自己的思考过程。
思路
读完题目我在想,有没有什么事实/直觉能告诉我,信号最强的点在哪?
想了许久,没有灵机一动,被迫暴力。
我一开始的想法是,以每个塔为中心,在半径 radius 范围内用 BFS 或 DFS 搜索所有的点,累加它们的信号强度。后来实现过程中发现有点麻烦,还得开个记录数组记录哪些点已经访问过了。
后来在纸上随便点了几个点,画了几个圆,发现可能的点在一个矩形范围内,就换了个暴力思路。
为什么不用考虑矩形外却在圆形里的点呢?因为这样的点的信号强度必然不超过矩形里也在圆形里的点的信号强度。
代码
class Solution {
public:
// 二维空间内欧几里得距离
// 即两点之间的直线距离
double distance(int x1, int y1, int x2, int y2) {
return sqrt(pow(x1 - x2, 2.0) + pow(y1 - y2, 2.0));
}
vector<int> bestCoordinate(vector<vector<int>>& towers, int radius) {
// 求出所有可能点的边界
int xLimit = INT_MIN, yLimit = INT_MIN;
for (vector<int> &tower : towers) {
int x = tower[0], y = tower[1];
xLimit = max(xLimit, x);
yLimit = max(yLimit, y);
}
// 遍历范围内所有点
int maxSignal = 0, xMax = 0, yMax = 0;
for (int i = 0; i <= xLimit; ++i) {
for (int j = 0; j <= yLimit; ++j) {
int signal = 0; // 点(i, j)的信号强度
// 计算所有塔对该点信号强度的贡献
for (vector<int> &tower : towers) {
int x = tower[0], y = tower[1], q = tower[2];
double d = distance(i, j, x, y);
if (d <= (double)radius) {
signal += floor(q / (1.0 + d));
}
}
// 取最大
// 因为按照字典序升序遍历所有点
// 所以不用考虑相等的情况
if (maxSignal < signal) {
xMax = i;
yMax = j;
maxSignal = signal;
}
}
}
return vector<int>{xMax, yMax};
}
};
最后
如果觉得有帮助,点个赞吧!
想看看我究竟有多混?公众号搜索“醉翁的零一世界”。
刚刚开始建设,像个树洞吧,时刻提醒自己本科太混了,研究生应该做点事情。
目标是聚集更多“天涯人”,然后建个群唠嗑。