문제 설명

[본 문제는 정확성과 효율성 테스트 각각 점수가 있는 문제입니다.]

카카오 초등학교의 “니니즈 친구들”이 “라이언” 선생님과 함께 가을 소풍을 가는 중에 징검다리가 있는 개울을 만나서 건너편으로 건너려고 합니다. “라이언” 선생님은 “니니즈 친구들”이 무사히 징검다리를 건널 수 있도록 다음과 같이 규칙을 만들었습니다.

• 징검다리는 일렬로 놓여 있고 각 징검다리의 디딤돌에는 모두 숫자가 적혀 있으며 디딤돌의 숫자는 한 번 밟을 때마다 1씩 줄어듭니다.
• 디딤돌의 숫자가 0이 되면 더 이상 밟을 수 없으며 이때는 그 다음 디딤돌로 한번에 여러 칸을 건너 뛸 수 있습니다.
• 단, 다음으로 밟을 수 있는 디딤돌이 여러 개인 경우 무조건 가장 가까운 디딤돌로만 건너뛸 수 있습니다.

“니니즈 친구들”은 개울의 왼쪽에 있으며, 개울의 오른쪽 건너편에 도착해야 징검다리를 건넌 것으로 인정합니다. “니니즈 친구들”은 한 번에 한 명씩 징검다리를 건너야 하며, 한 친구가 징검다리를 모두 건넌 후에 그 다음 친구가 건너기 시작합니다.

디딤돌에 적힌 숫자가 순서대로 담긴 배열 stones와 한 번에 건너뛸 수 있는 디딤돌의 최대 칸수 k가 매개변수로 주어질 때, 최대 몇 명까지 징검다리를 건널 수 있는지 return 하도록 solution 함수를 완성해주세요.

[제한사항]

• 징검다리를 건너야 하는 니니즈 친구들의 수는 무제한 이라고 간주합니다.
• stones 배열의 크기는 1 이상 200,000 이하입니다.
• stones 배열 각 원소들의 값은 1 이상 200,000,000 이하인 자연수입니다.
• k는 1 이상 stones의 길이 이하인 자연수입니다.

[입출력 예]

입출력 예에 대한 설명

입출력 예 #1

첫 번째 친구는 다음과 같이 징검다리를 건널 수 있습니다.

첫 번째 친구가 징검다리를 건넌 후 디딤돌에 적힌 숫자는 아래 그림과 같습니다. 두 번째 친구도 아래 그림과 같이 징검다리를 건널 수 있습니다.

두 번째 친구가 징검다리를 건넌 후 디딤돌에 적힌 숫자는 아래 그림과 같습니다. 세 번째 친구도 아래 그림과 같이 징검다리를 건널 수 있습니다.

세 번째 친구가 징검다리를 건넌 후 디딤돌에 적힌 숫자는 아래 그림과 같습니다. 네 번째 친구가 징검다리를 건너려면, 세 번째 디딤돌에서 일곱 번째 디딤돌로 네 칸을 건너뛰어야 합니다. 하지만 k = 3 이므로 건너뛸 수 없습니다.

따라서 최대 3명이 디딤돌을 모두 건널 수 있습니다.

문제 풀이

이분 탐색 문제였다… 앞으로 계산해야하는 수가 너무 크면 이분탐색이 아닌지부터 먼저 의심해야 될 것 같다. 처음 시도는 나름 효율성을 만족시키기 위해 머리를 굴려서 슬라이딩 윈도우 방식을 채용하였는데 정답과 효율성 7번까지는 만족하였지만 그 이후를 코드를 아무리 개선해도 통과하지 못하였다. 이 이상 좋은 방법이 생각나지 않아 결국 해답을 찾아봤는데 이분탐색으로 푸는 문제였다.
이분탐색을 알고는 있었지만 특정한 값을 찾을 때만 사용하는줄 알았고 이분탐색 문제를 별로 풀어보지 않아 익숙하지 않았는데 이렇게 특정한 조건을 만족하는 값을 찾을 때도 사용할 수 있다는걸 처음 알았다.

코드#1(슬라이딩 윈도우)

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int nini=200000000;

int solution(vector<int> stones, int k) {
    int answer = 0; int count=0;
    for(int i=0; i<=stones.size()-k; i++) {
        vector<int> tmp; tmp.clear();
        tmp.assign(stones.begin()+i, stones.begin()+i+k);
        int max=*max_element(tmp.begin(), tmp.end());
        if(max<nini) nini=max; //k개의 윈도우를 stones위에 놓고 1씩 슬라이드하여
        //윈도우의 최대값을 찾으면서 가장 작은 최대값을 구한다.
    }
    answer=nini;
    return answer;
}

코드#2(이분 탐색)

 
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> stones, int k) {
    int answer = 0; int left=1; int right=*max_element(stones.begin(), stones.end()); int mid; //left는 1, right는 stones의 최대값으로 시작
    while(left<=right) {
        int count=0;
        int mid=(right+left)/2;
        for(int i : stones) {
            i=i-mid;
            if(i<=0) count++;
            else count=0;
            if(count==k) break; //0 이하의 돌이 k이상 연속되면 break
        }
        if(count==k) right=mid-1; //mid만큼의 사람이 건널 수 없다면 right=mid-1
        else left=mid+1; //mid만큼이 건널 수 있다면 mid=left+1;
    }
    answer=left;
    return answer;
}