CCF201409-4 最优配餐（BFS多向搜索）

题目地址

• 本题易错点：

（1）网格坐标即输入数据是从1开始的，不要想当然为0
（2）可能有多个客户在同一位置，所以要先统计总的送餐地点与每个送餐地点的订餐量
（3）注意数据范围，结果用long long存储，否则最后两个测试点会溢出。

• 思路：

与基础广搜不同，本题有多个起始结点与多个目标结点，可以把所有的起始结点一次性全部入队执行BFS，要注意BFS的结束条件是所有的送餐地点都被扩展，剩下的操作与基础广搜相同。

• 代码实现

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

bool vis[1010][1010];
int client[1010][1010];
int dis[][2]={{1,0},{-1,0},{0,-1},{0,1}};
int n,m,k,d;

struct s{
	int x;
	int y;
	int steps;
	s(){}
	s(int xx,int yy,int ss):x(xx),y(yy),steps(ss){}
};

struct xy{
	int x;
	int y;
	xy(){}
	xy(int xx,int yy):x(xx),y(yy){}
	bool operator<(const xy &p) const{
		return x!=p.x?x<p.x:y<p.y;
	}
};

int main()
{
	ios::sync_with_stdio(false);
	while(cin>>n>>m>>k>>d){
		memset(vis,0,sizeof(vis));
		memset(client,0,sizeof(client));
		int count=0;
		long long sum=0;
		int xi,yi,ci;
		queue<s> q;
		for(int i=0;i<m;i++){
			cin>>xi>>yi;
			q.push(s(xi,yi,0));
			vis[xi][yi]=1;
		}
		for(int i=0;i<k;i++){
			cin>>xi>>yi>>ci;
			if(client[xi][yi]==0)
				count++;
			client[xi][yi]+=ci;
		}
		for(int i=0;i<d;i++){
			cin>>xi>>yi;
			vis[xi][yi]=1;
		}	
		while(!q.empty()){
			s t=q.front();
            if(client[t.x][t.y]>0){
                sum+=client[t.x][t.y]*t.steps; 
                if(--count==0)
                    break;
            }
			for(int i=0;i<4;i++){
				int dx=t.x+dis[i][0];
				int dy=t.y+dis[i][1];
				int dsteps=t.steps+1;
	            if(dx<1||dx>n||dy<1||dy>n||vis[dx][dy])
	                continue;	 	 
	            vis[dx][dy] = 1;
	            q.push(s(dx,dy,dsteps));
			}
			q.pop();
		}
		cout<<sum<<endl;
	}
	return 0;
}