【算法實踐】1.1 油井問題——線性時間內完成(分治,中位數)

主油管道爲東西向,確定主油管道的南北位置,使南北向油井噴油管道和最小。要求線性時間完成。在這裏插入圖片描述

#include "stdio.h"

#include "vector"

using namespace std;

vector<int> map;

int part(int l, int r, int m)
{
    int i = l, j = r;
    int x = map[l];
    while (i < j)
    {
        while (i < j && map[j] >= x)
        {
            j--;
        }
        if (i < j)
        {
            map[i++] = map[j];
        }
        while (i < j && map[i] <= x)
        {
            i++;
        }
        if (i < j)
        {
            map[j--] = map[i];
        }
    }
    map[i] = x;
    return i;
}

void find(int m)
{
    int i = -1, l = 0, r = map.size() - 1;
    while (true)
    {
        i = part(l, r, m);
        if (i < m)
        {
            l = i + 1;
        }
        else if (i > m)
        {
            r = i - 1;
        }
        else
        {
            break;
        }
    }
}

int main(int argc, char const *argv[])

{
    int x, y;
    while (scanf("%d,%d", &x, &y) != EOF)
    {
        map.push_back(y);
    }
    int n = map.size();
    int m;
    if (n % 2 == 0)
    {
        m = n / 2 - 1;
    }
    else
    {
        m = n / 2;
    }

    find(m);
    printf("%d\n", map[m]);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章