编程之美2015资格赛#3

题目3 : 基站选址
时间限制:2000ms
单点时限:1000ms
内存限制:256MB
描述
需要在一个N × M的网格中建立一个通讯基站，通讯基站仅必须建立在格点上。

网格中有A个用户，每个用户的通讯代价是用户到基站欧几里得距离的平方。

网格中还有B个通讯公司，维护基站的代价是基站到最近的一个通讯公司的路程（路程定义为曼哈顿距离）。

在网格中建立基站的总代价是用户通讯代价的总和加上维护基站的代价，最小总代价。

输入
第一行为一个整数T，表示数据组数。

每组数据第一行为四个整数：N, M, A, B。

接下来的A+B行每行两个整数x, y，代表一个坐标，前A行表示各用户的坐标，后B行表示各通讯公司的坐标。

输出
对于每组数据输出一行"Case #X: Y"，X代表数据编号（从1开始），Y代表所求最小代价。

数据范围
1 ≤ T ≤ 20

1 ≤ x ≤ N

1 ≤ y ≤ M

1 ≤ B ≤ 100

小数据

1 ≤ N, M ≤ 100

1 ≤ A ≤ 100

大数据

1 ≤ N, M ≤ 107

1 ≤ A ≤ 1000

样例输入
2
3 3 4 1
1 2
2 1
2 3
3 2
2 2
4 4 4 2
1 2
2 4
3 1
4 3
1 4
1 3
样例输出
Case #1: 4

Case #2: 13

思路：由于只测试小数据，因此采用暴力方法，计算每个点求取最小值。

#include<iostream>
#include<cmath>
#include<map>
#include<vector>
using namespace std;
int ans[102][102];
vector<pair<int, int>> va;
vector<pair<int, int>> vb;
void getAns(int n, int m){
	for (int i = 0; i <= n; i++){
		for (int j = 0; j <= m; j++){
			vector<pair<int, int>>::iterator ita = va.begin();
			vector<pair<int, int>>::iterator itb = vb.begin();
			ans[i][j] = 0;
			int min = abs(itb->first - i) + abs(itb->second - j);
			itb++;
			for (; ita != va.end(); ita++){
				ans[i][j] += ((ita->first - i)*(ita->first - i) + (ita->second - j)*(ita->second - j));
			}
			for (; itb != vb.end(); itb++){
				if (abs(itb->first - i) + abs(itb->second - j) < min)
					min = abs(itb->first - i) + abs(itb->second - j);
			}
			ans[i][j] += min;
		}
	}
}
int searchAns(int n, int m){
	int min = ans[0][0];
	for (int i = 0; i <= n; i++){
		for (int j = 0; j <= m; j++){
			if (ans[i][j] < min){
				min = ans[i][j];
			}
		}
	}
	return min;
}
int main(){
	int t;
	while (cin >> t){
		for (int k = 1; k <= t; k++){
			int n, m, a, b;
			int x, y;
			cin >> n >> m >> a >> b;
			for (int i = 0; i < a; i++){
				cin >> x >> y;
				pair<int, int> p;
				p.first = x, p.second = y;
				va.push_back(p);
			}
			for (int i = 0; i < b; i++){
				cin >> x >> y;
				pair<int, int> p;
				p.first = x, p.second = y;
				vb.push_back(p);
			}
			getAns(n, m);
			cout <<"Case #"<<k<<": "<<searchAns(n, m) << endl;
			va.clear();
			vb.clear();
		}
	}
}