2016蓝桥杯假期任务之《邮局》

问题描述

　　C村住着n户村民，由于交通闭塞，C村的村民只能通过信件与外界交流。为了方便村民们发信，C村打算在C村建设k个邮局，这样每户村民可以去离自己家最近的邮局发信。

　　现在给出了m个备选的邮局，请从中选出k个来，使得村民到自己家最近的邮局的距离和最小。其中两点之间的距离定义为两点之间的直线距离。

输入格式

　　输入的第一行包含三个整数n, m, k，分别表示村民的户数、备选的邮局数和要建的邮局数。
　　接下来n行，每行两个整数x, y，依次表示每户村民家的坐标。
　　接下来m行，每行包含两个整数x, y，依次表示每个备选邮局的坐标。
　　在输入中，村民和村民、村民和邮局、邮局和邮局的坐标可能相同，但你应把它们看成不同的村民或邮局。

输出格式

　　输出一行，包含k个整数，从小到大依次表示你选择的备选邮局编号。（备选邮局按输入顺序由1到m编号）

样例输入

5 4 2
0 0
2 0
3 1
3 3
1 1
0 1
1 0
2 1
3 2

样例输出

2 4

数据规模和约定

　　对于30%的数据，1<=n<=10，1<=m<=10，1<=k<=5；
　　对于60%的数据，1<=m<=20；
　　对于100%的数据，1<=n<=50，1<=m<=25，1<=k<=10。

代码如下：

import java.util.Scanner;

public class Main {

	static int n, m, k, j, f1, f2;
	static int[][] a = new int[55][2];
	static int[][] b = new int[27][2];
	static int[] c = new int[12];
	static int[] d = new int[55];
	static float[][] yc = new float[27][55];
	static float s = 1000000000;

	public static void main(String[] args) {
		Scanner input = new Scanner(System.in);
		int i, j;
		int[] o = new int[12];
		float[] w = new float[55];
		n = input.nextInt();
		m = input.nextInt();
		k = input.nextInt();
		for (i = 1; i <= n; i++) {
			a[i][0] = input.nextInt();
			a[i][1] = input.nextInt();
		}
		for (i = 1; i <= m; i++) {
			b[i][0] = input.nextInt();
			b[i][1] = input.nextInt();
			for (j = 1; j <= n; j++)
				yc[i][j] = (float) Math.sqrt((a[j][0] - b[i][0])
						* (a[j][0] - b[i][0]) + (a[j][1] - b[i][1])
						* (a[j][1] - b[i][1]));
		}
		dfs(0, 1, o, w, 0);
		for (i = 0; i < k; i++)
			System.out.print(c[i] + " ");

	}

	private static void dfs(int t, int i, int o[], float w[], float sum) {
		if (i <= m + 1) {
			if (t == k) {
				if (sum < s) {
					s = sum;
					for (j = 0; j < k; j++)
						c[j] = o[j];
				}
			} else if (i <= m && t < k) {
				float[] ww = new float[55];
				for (j = 1; j <= n; j++)
					ww[j] = w[j];
				dfs(t, i + 1, o, w, sum);
				f1 = 1;
				f2 = 0;
				if (d[i] == 0) {
					o[t] = i;
					if (t > 0) {
						f2 = 1;
						for (j = 1; j <= n; j++) {
							if (ww[j] > yc[i][j]) {
								sum = sum - ww[j] + yc[i][j];
								ww[j] = yc[i][j];
								f1 = 0;
							}
						}
					} else {
						for (j = 1; j <= n; j++) {
							sum += yc[i][j];
							ww[j] = w[j] = yc[i][j];
						}
					}
					if (f1 == 1 && f2 == 1) {
						d[i] = 1;
						dfs(t, i + 1, o, w, sum);
					} else
						dfs(t + 1, i + 1, o, ww, sum);
				}
			}
		}

	}

}

运行结果：

5 4 2
0 0
2 0
3 1
3 3
1 1
0 1
1 0
2 1
3 2
2 4