Peter's codebook A blog full of codes

POJ 3579 Median

Binary Search

傳送門

POJ3579

題目概述

給定一堆數字,會有 個差。 問在這些差之中,中位數的值是多少?

解題想法

先將資料排序,並使用一個 deque ,以 的時間去檢查資料中相差 以內差的數量,是否可以達到不少於中位數的數量。在這裡定義上述結果為

之後對該 做二分搜,初始上界為

下界為 -1 ,尋找使得 為真的上界,該上界就是答案。

總體複雜度

注意:C++ STL 的 Deque 速度不是很理想,可以自己使用兩個指標實作。

程式碼:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#include <iostream>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#include <functional>
#include <ctime>
#include <stdexcept>
using namespace std;

class solver {
    private:
        inline int absll(int a) { return a<0?-a:a; }
        vector<int> data;
        int tar;
        inline bool io(void) {
            int n;
            if(scanf("%d",&n)==EOF) return false;
            data.clear();
            for(int i=0; i!=n; ++i) {
                int t;
                scanf("%d",&t);
                data.push_back(t);
            }
            return true;
        }
        inline void shu(vector<int> &in) {
            for(vector<int>::iterator v=in.begin(); v!=in.end(); ++v) {
                swap( *v, in.at(rand()%in.size()) );
            }
        }

        inline bool check( int m) {
            int counter=0;
            deque< int>que;
            int i=0;
            while (i-que.size()<data.size()) {
                for(; i<data.size() && (que.empty()||data.at(i)-que.front() <=m) ; ++i) que.push_back(data.at(i));
                if(!que.empty()) que.pop_front();
                counter += que.size();
                if(counter>=this->tar) return true;
            }
            return false;
        }
        int sol() {
            sort(data.begin(), data.end());
            int ub=*max_element(data.begin(), data.end()) - *min_element(data.begin(), data.end());
            int lb=-1;
            this->tar = ( ( data.size() * (data.size()-1) )/2 + 1 ) / 2;
            while(ub-lb>1) {
                int mid=((lb+ub)>>1);
                if(check(mid)) {
                    ub=mid;
                }
                else lb=mid;
            }
            return ub;
        }
    public:
        inline void solveit() {
            while(io()) {
                printf("%d\n",sol());
            }
        }
        solver() {
            srand(11);
        }
};

int main(void) {
    solver sol;
    sol.solveit();
    return 0;
}
comments powered by Disqus