[이취코] 유형별 알고리즘 - 이진 탐색(C++)

1. 정렬된 배열에서 특정 수의 개수 구하기 

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

int n, num, x;
vector<int> v;

int solution(int x) {
    int low = lower_bound(v.begin(), v.end(), x) - v.begin(); //iter 반환 - 주소 = 해당 idx
    int upper = upper_bound(v.begin(), v.end(), x) - v.begin();
    return upper - low;
}

int main() {
    cin >> n >> x;
    for(int i = 0; i < n; i++){
        cin >> num;
        v.push_back(num);
    }
    sort(v.begin(), v.end());
    int result = solution(x); //2와 같은 것 몇개인지 구하기

    if(result == 0){
        cout << -1;
    } else{
        cout << result;
    }

    return 0;
}

 

2. 고정점 찾기

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

int n;
vector<int> v;

int binarySearch(int start, int end) {
    if (start > end)
        return -1;
    int mid = (start + end) / 2;
    if (v[mid] == mid)
        return mid;
    else if (v[mid] > mid) {
        return binarySearch(start, mid - 1);
    } else {
        return binarySearch(mid + 1, end);
    }
}

int main() {
    cin >> n;
    v.resize(n);
    for (int &i: v) {
        cin >> i;
    }
    sort(v.begin(), v.end());

    int idx = binarySearch(0, n - 1);
    cout << idx;
    return 0;
}

 

3. 공유기 설치

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

int n, c, num;

int main() {
    cin >> n >> c;
    vector<int> v;

    for (int i = 0; i < n; i++) {
        cin >> num;
        v.push_back(num);
    }
    sort(v.begin(), v.end());

    //start, end : 최소, 최대 거리
    //mid : 와이파이와 집 간 간격
    int start = 1;
    int end = v[n - 1] - v[0];
    int result = 0;

    while (start <= end) {
        int mid = (start + end) / 2;

        //모든 집을 돌면서 첫번째 집과의 거리와 mid 비교
        int pre_house = v[0]; //설치
        int cnt = 1;
        for (int i = 1; i < n; i++) {
            if (v[i] - pre_house >= mid) {
                cnt++;
                pre_house = v[i]; //설치했으니 바꿔주기
            }
        }
        // C개 이상의 공유기를 설치할 수 있는 경우, 거리를 증가시키기
        if (cnt >= c) {
            result = mid;
            start = mid + 1;
        } else { // C개 이상의 공유기를 설치할 수 없는 경우, 거리를 감소시키기
            end = mid - 1;
        }
    }

    cout << result;
    return 0;
}