C++：分治算法之输油管道问题

目录

描述

输入

输出

输入样例

输出样例

分析

代码

运行结果

---

描述

¢ 某石油公司计划建造一条 由东向西 的主输油管道。该管道要穿过一个有n口油井的油田。从每口油井都要有一条输油管道沿最短路经（或南或北）与主管道相连。

¢ 如果给定 n 口油井的位置，即它们的 x 坐标（东西向）和 y 坐标（南北向），应如何确定主管道的最优位置，即使 各油井到主管道之间 的输油管道长度总和最小的位置？

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

输入

第1行是一个整数n，表示油井的数量（1≤n≤10 000）。
接下来n行是油井的位置，每行两个整数x和y
（﹣10 000≤x，y≤10 000）。

输出

各油井到主管道之间的输油管道最小长度总和。

输入样例

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

输出样例

6

分析

设n口油井的位置分别为 Pi=(xi,yi)，i＝1～n。由于主输油管道是东西向的，因此可用其主轴线的y坐标唯一确定其位置。主管道的最优位置y应该满足：

由中位数定理可知，y是中位数。

代码

方法一：对数组a排序（一般是升序），取中间的元素

算法1：对数组a排序（一般是升序），取中间的元素

int n;					//油井的数量
int x;					//x坐标，读取后丢弃
int a[1000];				//y坐标
cin>>n;
for(int k=0;k<n;k++)
	cin>>x>>a[k];
sort(a,a+n);				//按升序排序
//计算各油井到主管道之间的输油管道最小长度总和
int min=0;
for(int i=0;i<n;i++)
	min += (int)fabs(a[i]-a[n/2]);
cout<<min<<endl;

/*
* 输油管问题
*/

#include<iostream>
#include<algorithm>
using namespace std;

int main() {
	int n;//油井数量
	int x;//横坐标
	int a[1000];//纵坐标

	cin >> n;

	for (int i = 0; i < n; i++) {
		cin >> x >> a[i];//输入每个油井的坐标
	}
	sort(a, a + n);//n个油井的y轴按大小升序排列（a-》0，a+n-》a[n]）

	//计算各油井到主管道之间的输油管道最小长度总和
	int min=0;//初始化最小长度
	for (int k = 0; k < n; k++) {
		min += (int)fabs(a[k] - a[k / 2]);
	}
	cout << "各油井到主管道之间的输油管道最小长度总和为：";
	cout << min << endl;

	return 0;
}

运行结果

 方法二：采用分治策略求中位数

算法2：采用分治策略求中位数

int n;					//油井的数量
int x;					//x坐标，读取后丢弃
int a[1000];				//y坐标
cin>>n;
for (int i=0; i<n; i++)
	cin>>x>>a[i];
int y = select(0, n-1, n/2);		//采用分治算法计算中位数
//计算各油井到主管道之间的输油管道最小长度总和
int min=0;
for(int i=0;i<n;i++)
	min += (int)fabs(a[i]-y);
cout<<min<<endl;

---

/*
* 输油管问题--分治算法计算中位数
*/

#include<iostream>
#include<algorithm>
using namespace std;

const int  NUM=1001;
int a[NUM];

//在a[left：right]中选择第k小的元素
int select(int left,int right,int k) {
	if (left >= right)
		return a[left];
	int low = left;//从左到右的指针
	int hight = right+1;//从右到左的指针

	//把最左边的元素作为分界数据
	int pivot = a[left];

	//把左侧>=pivot的和右侧<=pivot的元素交换
	while (true) {

		//在左侧找出>=pivot的元素
		do {
			low= low+1;
		} while (a[low]<pivot);

		//在右侧找出<=pivot的元素
		do {
			hight= hight-1;
		} while (a[hight]>pivot);
		if (low > hight)
			break;
		swap(a[low], a[hight]);
	}
	if ((hight - left + 1 )== k)
		return pivot;

	a[left] = a[hight];
	a[hight] = pivot;//存储pivot

	if ((hight - left + 1 )< k)
		//对一个段进行递归调用
		return select(hight + 1, right, k-hight+left-1);
	else
		return select(left, hight - 1, k);
}

int main() {
	int n;//油井数量
	int x;//横坐标
	int a[1000];//纵坐标

	cin >> n;

	for (int i = 0; i < n; i++) {
		cin >> x >> a[i];//输入每个油井的坐标
	}
	int y = select(0, n - 1, n / 2);//采用分治法计算中位数

	//计算各油井到主管道之间的输油管道最小长度总和
	int min = 0;
	for (int i = 0; i < n; i++) {
		min += (int)fabs(a[i] - y);
	}
	cout << min<<endl;
	return 0;
}