考点:
矩阵的广度优先遍历。
思路:
-
求出分店到各个点的最短距离
-
根据最短距离求出最少成本
-
输出成本
代码:
#include <iostream>
#include <queue>//队列头文件
#include <cstring>//memste 的头文件
using namespace std;
typedef pair<int, int> PII;
const int N = 1010;
bool g[N][N];// 地图:0:不可走,1:可走
int dist[N][N];//距离
int consumer[N * N][3];//用户地址和订餐的量
queue<PII> q;
int n, m, k, d;
void bfs()//广度优先遍历
{
int dx[] = {-1, 0 , 1, 0}, dy[] = {0, 1, 0,-1};//上下左右四个方向
while(q.size())
{
PII f = q.front();//取队头
q.pop();//出队
for(int i = 0; i < 4 ; i++)//向四个方向进行遍历
{
int x = f.first + dx[i], y = f.second + dy[i];
if(g[x][y] == 1)//如果能走过去
{
if(dist[x][y] > dist[f.first][f.second] + 1)//如果距离变小
{
dist[x][y] = dist[f.first][f.second] + 1;//更新距离
q.push({x, y});//并且入队
}
}
}
}
}
int main()
{
ios::sync_with_stdio(false);//输入量较大,此语句可以加快输出
memset(dist, 0x3f, sizeof(dist));//初始化dist,各个元素为int最大值
cin >> n >> m >> k >> d;
for(int i = 1; i <= n; i++)//初始化地图,全部为 1。地图从 g[1,1] 到 g[n,n],四周被 0 包围。这样再bfs时,可以不用做边界处理
{
for(int j = 1; j <= n; j++)
g[i][j] = 1;
}
while(m--)//分店
{
int x, y;
cin >> x >> y;
q.push({x, y});//分店入队列
dist[x][y] = 0;//分店的送餐距离为 0
}
for(int i = 0; i < k; i++)//保存客户信息
{
cin >> consumer[i][0] >> consumer[i][1] >> consumer[i][2];
}
while(d--)//不能走的地方为 0
{
int x, y;
cin >> x >> y;
g[x][y] = 0;
}
bfs();//广度优先遍历
long long res = 0;//保存结果
while(k > 0)//求出各个客户的成本
{
k--;
res += dist[consumer[k][0]][consumer[k][1]] * consumer[k][2];
}
cout << res;
}