输油管问题

#include<iostream>
#include<cmath>
using namespace std;
#define NUM 1001
int a[NUM];
int select(int left, int right, int k)//采用分治算法计算中位数。
{
    int i = left;
    int j = right;
    int pivot = a[left];
    while (true)
    {
        while (a[i] < pivot)
        {
            i++;
        }while (a[j] > pivot)
        {
            j--;
        }
        if (i == j)break;
        else
        {
            swap(a[i], a[j]);
        }
        if (a[i] == a[j])j--;
    }
    if (j - left == k - 1)
    {
        return pivot;
    }
    if (j - left < k - 1)
    {
        return select(j + 1, right, k - (j - left + 1));
    }
    else
    {
        return select(left, j - 1, k);
    }
}
//根据中位数定理：如果有一个数轴，数轴上有若干个点。要在数轴上找一点，使得它到各个点的距离之和最短。
//中位数就是最优解。中位数有这样的性质 ：所有数与中位数的绝对差之和最小。=n/2.
int mindis(int a[],int n, int k)//k就是中位数。
{
    int yy = 0;
    for (int i = 0; i < n; i++)
    {
        yy += abs(a[i] - k);//绝对值。
    }
    return yy;
}

int main()
{
    int n;
    cout << "请输入有多少个油井" << endl;
    cin >> n; int x,y; ;
    for (int i = 0; i < n; i++)
    {
        cout << "请输入第"<<i+1<<"个油井的横纵坐标 "<< endl;
        cin >> x >> y;
        a[i] = y;//因为主管道是一条与x轴平行的管道，所以只保存y值就行。
    }
    int k = select(0, n - 1, n / 2);
    cout <<"输油管道最小长度总和为" << mindis(a, n, k);
}