상세 컨텐츠

본문 제목

징검다리 건너기 프로그래머스(c++)

알고리즘/프로그래머스

by 아리따운노을 2020. 3. 31. 12:33

본문

문제 설명

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

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

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

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

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

[제한사항]

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

 

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
#include <string>
#include <vector>
 
using namespace std;
 
int solution(vector<int> stones, int k) {
    int answer = 200000001,idx,len = stones.size();
    
    for(int i = -1;i<len-k;)
    {
        int max = 0;
        for(int j= i +1;j<=i+k;j++)
        {
            if(max<stones[j])
            {
                max = stones[j];
                idx = j;
            }
        }
        if(answer > max) answer = max;
        i = idx;
    }
    
    return answer;
}
cs

 

효율성 1개에서 시간초과 나는 코드다.

정확성 테케는 0.00ms, 나머지 효율성 테케도 1ms 내외로 나오는데 테스트 13만 시간 초과 나와서 모의고사 때도 생각하다 포기 했던 문제다.

알고리즘 자체는 코드를 알아먹기 쉽게 짰으니 보고도 이해 할 수 있다.

기본적으로 답은 모든 위치에서 k 만큼 검사하는데 그 중 가장 큰 값을 찾고

그 가장 큰 값들 중 제일 작은게 답이다. 그래서 모든 위치를 검사하기 위해 -1부터 시작한다.

0부터 시작하면 정확성 테스트케이스 1번이 틀릴거다.

 

1. i = -1(0,1,...,k)까지 검사해야하므로 i<len-k는 그 이후는 어차피 값이 같거나 의미가 없다. 증가문이 비어있는 이유는 아래에서 max 부분으로 한번에 jump 할거기 때문

2. max = 0, 현재 위치에서 k만큼 검사, 그 들중 가장 큰 값을 max로 기억하고 점프를 위해 idx도 기억

3. answer가 max보다 크다면 answer = max 조건에서 answer는 2억을 넘는다고 하지 않았음

4. i를 idx로 점프

 

O(n)도 아니고 O(logn)이라 생각하는데 왜 시간초과가 나는지 모르겠다.. 그렇다고 다른 테케도 아슬아슬 한것도 아니고 널널한데...

 

 

---------------------------------------------

추가 4.2 풀긴 풀었다.. 꼼수로 ㅋㅋㅋㅋㅋ

테케 13이 시간초과가 나길래 한번 배열 뒤쪽부터 탐색하는 경우로 짜봤다.

그러니까 13은 통과되고 14는 시간초과가 나더라

그래서 배열 첫번째 원소랑 마지막 원소를 비교해서 잘 하면 둘 다 만족할수 있겠지라고 생각해서

if else문을 두고 해봤는데 둘다 0.5ms정도로 통과했다...

쉬운 문제에서 꼼수로 몇번 통과해봤는데 level 4정도에서 통과해본건 처음이다 ㅋㅋ

 

 

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
#include <string>
#include <vector>
 
using namespace std;
 
int solution(vector<int> stones, int k) {
    int answer = 200000001,idx,len = stones.size();
    if(stones[0]>stones[len-1])
    {
        for(int i = len; i>=k;)
        {
            int max = 0;
            for(int j= i  - 1;j>=i-k;j--)
            {
                if(max<stones[j])
                {
                    max = stones[j];
                    idx = j;
                }
            }
            if(answer > max) answer = max;
            i = idx;
        }
    }
    else
    {
        for(int i = -1;i<len-k;)
        {
            int max = 0;
            for(int j= i +1;j<=i+k;j++)
            {
                if(max<stones[j])
                {
                    max = stones[j];
                    idx = j;
                }
            }
            if(answer > max) answer = max;
            i = idx;
        }
    
    }
    
 
    return answer;
}
cs

 

어찌 됐든 이것도 정답이긴 한데... 다른 분들 답을 보니 다들 이분탐색으로 풀었던데 이렇게 풀지 마세요...

 

----------------------------------------------------

추가 4.4 이분탐색으로 다시 풀었습니다.

 

 

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
#include <string>
#include <vector>
using namespace std;
bool calc(int n, const vector<int>& stones, int k) {
    int now = 0;
    for (int i = 0; i < stones.size(); i++) {
        if (stones[i] - n <= 0)
            now++;
        else
            now = 0;
        if (now >= k)
            return false;
    }
    return true;
}
int solution(vector<int> stones, int k) {
    int left = 1;
    int right = 200000001;
    while (left <= right) {
        int mid = (left + right) / 2;
        if (calc(mid, stones, k))
            left = mid + 1;
        else
            right = mid - 1;
    }
    return left;
}
cs

관련글 더보기

댓글 영역