BOJ 11054번 가장 긴 바이토닉 부분수열

BOJ : https://www.acmicpc.net/problem/11054

GitHub : https://github.com/junho0956/Algorithm/blob/master/11054/11054/%EC%86%8C%EC%8A%A4.cpp

 

수열 S가 어떤 수 Sk를 기준으로 S1 < S2 < ... Sk-1 < Sk > Sk+1 > ... SN-1 > SN을 만족한다면

그 수열을 바이토닉 수열이라고 한다.

1 5 2 1 4 3 4 5 2 1 을 예로들면 

길이 7의 1 2 3 4 5 2 1 이 정답이 된다. 5를 기준 왼쪽은 증가, 오른쪽은 감수하는 부분수열이다.

 

수열 인덱스 1~N 를 전부 반복문을 돌면서

임의의 한 부분을 최대값으로 잡고 왼쪽 오른쪽에 대한 부분수열을 구해주면 된다.

 

#include <iostream>
#include <vector>
using namespace std;
 
vector<int> v;
int N, start_num;
int arr[1001];
int Max, cnt;
 
int bound(int s, int e, int key) {
    int m;
    while (s < e) {
        m = (s + e) / 2;
        if (v[m] > key) s = m + 1;
        else e = m;
    }
    return s;
}
 
int setup(int i, int check) {
    if (check) {
        for (int k = i - 1; k >= 0; k--) {
            if (v[v.size() - 1] > arr[k]) {
                v.push_back(arr[k]);
            }
            else {
                int bnd = bound(0, v.size() - 1, arr[k]);
                if (v[bnd] != start_num) {
                    v[bnd] = arr[k];
                }
            }
        }
    }
    else {
        for (int k = i + 1; k < N; k++) {
            if (v[v.size() - 1] > arr[k]) {
                v.push_back(arr[k]);
            }
            else {
                int bnd = bound(0, v.size() - 1, arr[k]);
                if (v[bnd] != start_num) {
                    v[bnd] = arr[k];
                }
            }
        }
    }
    return v.size();
}
 
int main() {
    cin >> N;
    for (int i = 0; i < N; i++) cin >> arr[i];
 
    for (int i = 0; i < N; i++) {
        cnt = 0;
        v.clear();
        start_num = arr[i];
        v.push_back(arr[i]);
        cnt += setup(i,1);
 
        v.clear();
        v.push_back(arr[i]);
        cnt += setup(i,0) - 1;
 
        Max = Max < cnt ? cnt : Max;
    }
    cout << Max;
    return 0;
}

복습 20.02.09