PTA 7 士兵排队问题 分治算法思想

在一个划分成网格的操场上，n个士兵散乱地站在网格点上。网格点用整数坐标(x,y)表示。士兵们可以沿网格边往上、下、左、右移动一步，但在同一时刻任一网格点上只能有一名士兵。按照军官的命令，士兵们要整齐地列成一个水平队列，即排列成(x,y),(x+1,y),…,(x+n-1,y)。如何选择x和y的值才能使士兵们以最少的总移动步数排成一行。

编程计算使所有士兵排成一行需要的最少移动步数。

题目引自POJ

输入格式:

第1行是士兵数n，1≤n≤10000。接下来n行是士兵的初始位置，每行有2个整数x和y，-10000≤x，y≤10000。

输出格式:

一个数据，即士兵排成一行需要的最少移动步数。

输入样例:

5
1  2
2  2
1  3
3  -2
3  3

结尾无空行

输出样例:

8

这道题很多人那道题可能一头雾水，自己C语言、java语言等语法学得还不错，但是这题却无从下手，不知道怎么开始写。与其说这道答题是编程题，还不如说是数学计算题。做过输油管道问题得同学可能有点思路，但是现在是二维坐标，不再是一维坐标了，多了一维，那么问题就要变得复杂了。现在看分析：

Y轴上：(假设现在有n个士兵)

可以把所有士兵得y坐标看成一个输油管道问题，即找一个坐标Y_mid，使得所有y坐标道该坐标的距离之和最小。显然sum1 = |y0-Y_mid| + |y1-Y_mid| + |y2-Y_mid|....|yn-1-Y_mid|。  经过证明Y_mid就是y坐标的中位数，注意中位数，不是数学意义上的中位数，是将n个数按序排序，第n/2大的数，即a[n/2]。

X轴上。这里就有点难以分析了。假设现在排序后的x坐标为x0,x1,x2,x3.....xn-1，而最终士兵站好的坐标为x'0,x'1,x'2.....x'n-1。那么需要移动的步数就是sum2 = |x0-x'0| + |x1-x'1| + |x2-x'2|....+| xn-1-x'n-1|。  而我们直到士兵排好序后是连续的，所以上面公式可以写成这样，sum2 = |x0-x'0| + |x1-(x'0+1)| + |x2-(x'0+2)|....+| xn-1-(x'0+n-1)|。  现在就和那个输油管道问题是一样的了。但是现在这个公式还不够直观，现在我们把公式再进行转化 sum2 = |x0-x'0| + |(x1-1)-x'0)| + |(x2-2)-x'0|....+| (xn-1-(n-1))-x'0)|。  很显然这个x'0就是数组x坐标的中位数。

则最小移动步数为 sum1+sum2。

代码如下：

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define Max 10000
int Compare(const void* e1, const void* e2)//定义函数比较
{
	return (int)*((int*)e1) - (int)*((int*)e2);
}

int add(int a[], int n,int mid)
{
	int sum = 0;
	for (int i = 0; i < n; i++)
	{
		sum += abs(a[i]-mid);
	}
	return sum;
}

int main()
{
	int x[Max] = { 0 }, y[Max] = { 0 };
	int n = 0;
	scanf("%d", &n);
	int i = 0;
	for (i = 0; i < n; i++)
	{
		scanf("%d%d", &x[i], &y[i]);
	}
	qsort(x, n, sizeof(x[0]), Compare);
	qsort(y, n, sizeof(y[0]), Compare);
	for (i = 0; i < n; i++)//这是x轴上的坐标移动的最终位置，是根据公式求得的
	{
		x[i] = x[i] - i;
	}
	qsort(x, n, sizeof(x[0]), Compare);
	int Y_mid = y[n / 2], X_mid = x[n / 2];
	int y_sum = 0, x_sum = 0;
	y_sum = add(y, n, Y_mid);	
	x_sum = add(x, n, X_mid);
	printf("%d\n", x_sum + y_sum);
	return 0;
}

运行结果：

提交情况：