\quad 这个题主要考察多源BFS,有几个坑点如下:
- 实现上尽可能高效,博主在使用cin,cout时最后两个测试点超时
- 队列中每个点信息存储时尽量用结构体,博主最开始用queue里面嵌套vector得的方式存储每个点信息,超时
- 最后的结果可能会爆int,最多有大约1000*1000个用户点,每个点最多参量为1000,距离供货点最远为2000,完全可以爆int。故用long long存储结果
#include <queue>
#include <cstdio>
using namespace std;
int g[1010][1010];
int dx[4] = {-1, 1, 0, 0}, dy[4] = {0, 0, 1, -1};
struct P{
int x, y, d;
};
int main()
{
int n, m, k, d; scanf("%d%d%d%d", &n, &m, &k, &d);
int x, y, num;
queue<P> q;
for(int i = 0; i < m; i++)
{
scanf("%d%d", &x, &y);
q.push(P{x, y, 0});
}
for(int i = 0; i < k; i++)
{
scanf("%d%d%d", &x, &y, &num);
g[x][y] += num;
}
for(int i = 0; i < d; i++)
{
scanf("%d%d", &x, &y);
g[x][y] = -1;
}
long long res = 0;
while(!q.empty())
{
P t = q.front();
int x = t.x, y = t.y, d = t.d;
q.pop();
for(int i = 0; i < 4; i++)
{
int nx = x + dx[i], ny = y + dy[i];
if(nx<1 || nx>n || ny<1 || ny>n) continue;
if(g[nx][ny] == -1) continue;
res += g[nx][ny] * (d + 1);
q.push(P{nx, ny, d + 1});
g[nx][ny] = -1;
}
}
printf("%lld\n", res);
return 0;
}