输油管道问题(利用分治算法计算中位数)

【题目描述】
某石油公司计划建造一条由东向西的主输油管道。该管道要穿过一个有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

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

#define NUM 1001
int a[NUM];
int select(int left, int right, int k)//选择第k小的数
{
	if (left >= right) return a[left];
	int i = left;
	int j = right+1; 
	int pivot = a[left];
	while (true) 
	{
		do {
			i = i+1;
		} while (a[i] < pivot);
		do {
			j = j-1;
		} while (a[j] > pivot);
		if (i >= j) break;
		swap(a[i], a[j]);
	}
	if (j-left+1 == k) return pivot;
	a[left] = a[j];
	a[j] = pivot;
	if (j-left+1 < k)
	return select(j+1, right, k-j+left-1);
	else return select(left, j-1, k);
}

int main()
{
	int n;
	int x;
	int y;
	cin>>n;
	for (int i=0; i<n; i++)
		cin>>x>>a[i];//x没用,只要y的平均数
	y = select(0, n-1, n/2);//选择第n/2小的数
	int min=0;
	for(int i=0;i<n;i++)
		min += (int)fabs(a[i]-y);
	cout<<min<<endl;
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值