最优配餐 多源bfs

最优配餐 多源bfs

//G[i][j] 为1表示可走，0表示不可走
//cus[i][0-2] 表示顾客i的横坐标，纵坐标以及送餐量
//dis[i][j]  表示最近分店到该位置的距离  

#include<iostream>
using namespace std;
#include<cstring>
#include<queue>

#define PII pair<int,int>

queue<PII > qq;

#define ll long long 

int n,m,k,d;

ll sum;
int GG[1005][1005];

int cus[1000005][3];
int dis[1005][1005];

int dir[4][2]={{0,1},{0,-1},{-1,0},{1,0}};

void bfs()
{
	int nextx;
	int nexty;
	while(!qq.empty())
	{
		PII cur=qq.front();
		qq.pop();
		
		for(int i=0;i<4;i++)
		{
			nextx=cur.first+dir[i][0];
			nexty=cur.second+dir[i][1];
			
			if(GG[nextx][nexty]==1 && dis[nextx][nexty]>(dis[cur.first][cur.second]+1))
			{
				dis[nextx][nexty]=dis[cur.first][cur.second]+1;
				qq.push({nextx,nexty});
			}
		}
		
	}		
}

int main()
{
	memset(dis,0x3f3f3f,sizeof(dis));
	cin>>n>>m>>k>>d;
	
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			GG[i][j]=1;
	
	int x,y;
	for(int i=0;i<m;i++)
	{
		scanf("%d %d",&x,&y);
		qq.push({x,y});
		dis[x][y]=0;
	}
	
	for(int i=0;i<k;i++)
	{
		scanf("%d %d %d",&cus[i][0],&cus[i][1],&cus[i][2]);
	}
	
	int xx,yy;
	
	for(int i=0;i<d;i++)
	{
		scanf("%d %d",&xx,&yy);
		GG[xx][yy]=0;
	}
	
	bfs();
	
	for(int i=0;i<k;i++)
	{
		sum+=(dis[cus[i][0]][cus[i][1]]*cus[i][2]);
	}
	
	
	cout<<sum;
	
	return 0;
 }