CCF认证 2014-09 最优配餐

从所有分店为起点开始BFS计算到所有点的最短路，每个点的花费就是：距离*送餐数量，全部加起来就可以

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int N=1000+10;
const int dx[]={0,0,-1,1};
const int dy[]={-1,1,0,0};
struct node
{
	int x,y;
	node(int a,int b)
	{
		x=a;y=b;
	}
};
bool g[N][N];
int c[N][N],dist[N][N];
int main()
{
	int n,m,k,d;
	int x,y;
	memset(g,false,sizeof(g));
	memset(c,0,sizeof(c));
	memset(dist,-1,sizeof(dist));
	queue<node> q;
	scanf("%d%d%d%d",&n,&m,&k,&d);
	while(m--)
	{
		scanf("%d%d",&x,&y);
		q.push(node(x,y)); //分店入队 
		dist[x][y]=0;
	}
	while(k--)
	{
		scanf("%d%d",&x,&y);
		scanf("%d",&c[x][y]); //记录每个点送餐数量 
	}
	while(d--)
	{
		scanf("%d%d",&x,&y);
		g[x][y]=true; //标记不能走的点 
	}
	long long ans=0; //这里需要开long long
	while(!q.empty())
	{
		node u=q.front();
		q.pop();
		int ux=u.x,uy=u.y;
		ans+=(long long)dist[ux][uy]*c[ux][uy]; //计算每个点送餐花费 
		for(int i=0;i<4;i++)
		{
			int nx=ux+dx[i],ny=uy+dy[i];
			if(nx>n||nx<1||ny>n||ny<1||g[nx][ny]) continue;
			if(dist[nx][ny]!=-1) continue;
			q.push(node(nx,ny));
			dist[nx][ny]=dist[ux][uy]+1;
		}
	}
	printf("%lld\n",ans);
    return 0;
}