PTA:7-3 输油管道问题 (50 分)

一、题目

某石油公司计划建造一条由东向西的主输油管道。该管道要穿过一个有n 口油井的油田。从每口油井都要有一条输油管道沿最短路经(或南或北)与主管道相连。如果给定n口油井的位置,即它们的x 坐标（东西向）和y 坐标（南北向）,应如何确定主管道的最优位置,即使各油井到主管道之间的输油管道长度总和最小的位置? 证明可在线性时间内确定主管道的最优位置。

给定n口油井的位置, 计算各油井到主管道之间的输油管道最小长度总和。

输入格式:
输入的第1 行是油井数n，1<=n<=10000。接下来n 行是 油井的位置，每行2个用空格割开的整数 x和 y，-10000<=x，y<=10000。

输出格式:
输出油井到主管道之间的输油管道最小长度总和。

输入样例:
5
1 2
2 2
1 3
3 -2
3 3
输出样例:
6

二、详解

1.这个问题就是求中位数，关于中位数
2.中位数求法转换为第K小数问题；
n为奇数时，去中间数即可
n为偶数时，任取中间两个数中任意一个即可
3.求绝对值
#include<math.h>中有fabs()函数，对浮点数求绝对值
#include<stdlib.h>中有abs()函数，对整数求绝对值

三、代码

//输油管道问题
#include<iostream>
#include<stdlib.h>
using namespace std;
int partition(int a[],int first,int end)
{//快排 
	int x=a[first];
	while(first<end)
	{
		while(first<end&&a[end]>=x)
		end--;
		a[first]=a[end];
		while(first<end&&a[first]<=x)
		first++;
		a[end]=a[first];
	}
	a[first]=x;
	return first;
}
int find(int a[],int first,int end,int k)
{//找中位数即可 
	if(first==end)
	return a[first];
	int p=partition(a,first,end);
	if(p==k)
	return a[p];
	else if(k<p)
	return find(a,first,p-1,k);
	else find(a,p+1,end,k);
}
int main()
{
	int a[10005],b[10005];
	int n,mid,sum=0;
	cin>>n;
	for(int i=0;i<n;i++)
	cin>>b[i]>>a[i];
	
	mid=find(a,0,n-1,n/2);//找合适位置
	for(int i=0;i<n;i++)
	{
		sum=sum+abs(a[i]-mid);
	 }

	 cout<<sum;
	 return 0; 
	
 }