思路:将每个店加入队列,bfs依次开始搜索,搜索到所有顾客所在地都被访问过为止,由于队列的性质,每个顾客被访问时,都是被离自己最近的店所访问,所以最多搜全图一次。注意答案可能会超int,要使用long long保存。
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<map>
#include<vector>
#include<cmath>
#include<queue>
using namespace std;
const int N = 1005;
typedef long long ll;
typedef pair<int,int> pa;
struct node
{
int x,y;
int c;
};
int vis[N][N],customer[N][N],dis[N][N];
int nex[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
int n;
queue<node> q;
ll bfs(int sum)
{
int num=0;
ll ans=0;
while(q.size())
{
node tem=q.front();
q.pop();
int x=tem.x,y=tem.y,t=tem.c;
for(int i=0;i<4;i++)
{
int xx=x+nex[i][0],yy=y+nex[i][1];
if(xx<=0 || xx>n || yy<=0 || yy>n || vis[xx][yy])
continue;
vis[xx][yy]=1;
q.push({xx,yy,t+1});
if(customer[xx][yy])
num++,ans+=customer[xx][yy]*(t+1);
if(num==sum) return ans;
}
}
return ans;
}
int main()
{
int m,k,d;
cin>>n>>m>>k>>d;
for(int i=0;i<m;i++)
{
int x,y;
cin>>x>>y;
vis[x][y]=1;
q.push({x,y,0});
}
for(int i=0;i<k;i++)
{
int x,y,c;
cin>>x>>y>>c;
customer[x][y]+=c; //同一个位置的顾客直接合并
}
for(int i=0;i<d;i++)
{
int x,y;
cin>>x>>y;
vis[x][y]=1;
}
cout<<bfs(k);
return 0;
}