最优配餐（多源DFS/BFS） Java解法

问题描述

  栋栋最近开了一家餐饮连锁店，提供外卖服务。随着连锁店越来越多，怎么合理的给客户送餐成为了一个急需解决的问题。
  栋栋的连锁店所在的区域可以看成是一个n×n的方格图（如下图所示），方格的格点上的位置上可能包含栋栋的分店（绿色标注）或者客户（蓝色标注），有一些格点是不能经过的（红色标注）。
　　
  方格图中的线表示可以行走的道路，相邻两个格点的距离为1。栋栋要送餐必须走可以行走的道路，而且不能经过红色标注的点。
  送餐的主要成本体现在路上所花的时间，每一份餐每走一个单位的距离需要花费1块钱。每个客户的需求都可以由栋栋的任意分店配送，每个分店没有配送总量的限制。
  现在你得到了栋栋的客户的需求，请问在最优的送餐方式下，送这些餐需要花费多大的成本。

【输入形式】

  输入的第一行包含四个整数n, m, k, d，分别表示方格图的大小、栋栋的分店数量、客户的数量，以及不能经过的点的数量。
  接下来m行，每行两个整数xi, yi，表示栋栋的一个分店在方格图中的横坐标和纵坐标。
  接下来k行，每行三个整数xi, yi, ci，分别表示每个客户在方格图中的横坐标、纵坐标和订餐的量。（注意，可能有多个客户在方格图中的同一个位置）
  接下来d行，每行两个整数，分别表示每个不能经过的点的横坐标和纵坐标。

【输出形式】

输出一个整数，表示最优送餐方式下所需要花费的成本。

【样例输入】

10 2 3 3
1 1
8 8
1 5 1
2 3 3
6 7 2
1 2
2 2
6 8

【样例输出】

29

【评分标准】

评测用例规模与约定
  前30%的评测用例满足：$1<=n<=20$。
  前60%的评测用例满足：$1<=n<=100$。
  所有评测用例都满足：$1<=n<=1000，1<=m,k,d<=n^2$。可能有多个客户在同一个格点上。每个客户的订餐量不超过1000，每个客户所需要的餐都能被送到。

解题思路

  用dis数组保存任意位置到任意饭店的最短距离，通过dfs或bfs以所有饭店为起点，进行遍历，不断更新dis数组，得出最优配送方案，然后计算得出结果。

代码

bfs

public class Main {
	public static int n,m,k,d;
	public static int dis[][],map[][],consumer[][],store[][]; 
	public static Queue<int[]>qu;
	public static int move[][]= {{0,1},{1,0},{0,-1},{-1,0}};
	public static void bfs() {
		while(!qu.isEmpty()) {
			int node[]=qu.poll();
			for(int i=0;i<4;i++) {
				int newnode[]=new int[2];
				newnode[0]=node[0]+move[i][0];
				newnode[1]=node[1]+move[i][1];
				if(newnode[0]>0&&newnode[0]<=n&&newnode[1]>0&&newnode[1]<=n&&map[newnode[0]][newnode[1]]!=-1) {
					if(dis[newnode[0]][newnode[1]]>dis[node[0]][node[1]]+1) {
						dis[newnode[0]][newnode[1]]=dis[node[0]][node[1]]+1;
						qu.add(newnode);
					}
				}
			}
		}
	}
	public static void main(String[] args) {	
		Scanner in=new Scanner(System.in);
		n=in.nextInt();
		m=in.nextInt();
		k=in.nextInt();
		d=in.nextInt();
		map=new int[n+1][n+1];
		dis=new int[n+1][n+1];//表示任意位置到任意店铺的最短距离
		consumer=new int[k][3];
		store=new int[m][2];
		qu=new LinkedList<>();
		int x,y;
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				dis[i][j]=(int) 1e6;
			}
		}
		for(int i=0;i<m;i++) {
			store[i][0]=in.nextInt();
			store[i][1]=in.nextInt();
			dis[store[i][0]][store[i][1]]=0;
			qu.add(store[i]);
		}
		for(int i=0;i<k;i++) {
			consumer[i][0]=in.nextInt();
			consumer[i][1]=in.nextInt();
			consumer[i][2]=in.nextInt();
		}
		for(int i=0;i<d;i++) {
			x=in.nextInt();
			y=in.nextInt();
			map[x][y]=-1;
		}
		bfs();
		int res=0;
		for(int i=0;i<k;i++) {
			res+=dis[consumer[i][0]][consumer[i][1]]*consumer[i][2];
		}
		System.out.println(res);
	}   
}

dfs

public class Main {
	public static int n,m,k,d;
	public static int dis[][],map[][],consumer[][],store[][]; 
	public static int move[][]= {{0,1},{1,0},{0,-1},{-1,0}};
	public static void dfs(int node[]) {
		for(int i=0;i<4;i++) {
			int newnode[]=new int[2];
			newnode[0]=node[0]+move[i][0];
			newnode[1]=node[1]+move[i][1];
			if(newnode[0]>0&&newnode[0]<=n&&newnode[1]>0&&newnode[1]<=n&&map[newnode[0]][newnode[1]]!=-1) {
				if(dis[newnode[0]][newnode[1]]>dis[node[0]][node[1]]+1) {
					dis[newnode[0]][newnode[1]]=dis[node[0]][node[1]]+1;
					dfs(newnode);
				}
			}
		}
	}
	public static void main(String[] args) {	
		Scanner in=new Scanner(System.in);
		n=in.nextInt();
		m=in.nextInt();
		k=in.nextInt();
		d=in.nextInt();
		map=new int[n+1][n+1];
		dis=new int[n+1][n+1];//表示任意位置到任意店铺的最短距离
		consumer=new int[k][3];
		store=new int[m][2];
		qu=new LinkedList<>();
		int x,y;
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				dis[i][j]=(int) 1e6;
			}
		}
		for(int i=0;i<m;i++) {
			store[i][0]=in.nextInt();
			store[i][1]=in.nextInt();
			dis[store[i][0]][store[i][1]]=0;
			qu.add(store[i]);
		}
		for(int i=0;i<k;i++) {
			consumer[i][0]=in.nextInt();
			consumer[i][1]=in.nextInt();
			consumer[i][2]=in.nextInt();
		}
		for(int i=0;i<d;i++) {
			x=in.nextInt();
			y=in.nextInt();
			map[x][y]=-1;
		}
		for(int i=0;i<m;i++) {
			dfs(store[i]);
		}
		int res=0;
		for(int i=0;i<k;i++) {
			res+=dis[consumer[i][0]][consumer[i][1]]*consumer[i][2];
		}
		System.out.println(res);
	}   
}