코딩테스트 -- 징검다리 건너기 - (프로그래머스 / C++)

프로그래머스

 

문제!

---

문제 설명

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

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

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

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

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

[제한사항]

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

 

---

요약하면 징검다리를 건널껀데,

한 번 건널 때 마다 밟을 수 있는 횟수가 줄어들고,

밟을 수 있는 횟수가 0인 돌을 최대 k개 만큼만 뛰어 넘을수 있을때,

최대 인원을 구해라!

 

처음 보고 어? 금방 풀겠는데? 생각했었는데...

 

 

간단하게 풀었는데 정확성은 100퍼 나오는데 시간초과가 나네요...?

#include <string>
#include <vector>

#include<iostream>
using namespace std;

int solution(vector<int> stones, int k) {
    int answer = 0;
    int count=0;
    
    while(true){
        count=0;
        for(int i=0;i<stones.size();i++){
            
            if(count>=k){//넘겼을때
                break;
            }
            
            if(stones[i]!=0){//0이아니면 위치 돌 하나씩줄이기
                count=0;
                stones[i]-=1;
            }
            else{//0이면 conunt+=1
                count+=1;
            }
        }
        
        if(count>=k){
            break;
        }
        
        answer+=1;
    }
    return answer;
}

모든 사람을 하나 씩 이동시켜보며 건널수 없을때를 찾아 보았는데 조건에 신경을 못썻네요....

• stones 배열 각 원소들의 값은 1 이상 200,000,000 이하인 자연수입니다.

네... 최대 인원수가 200,000,000입니다....

 

이러한 문제처럼 가짓수가 너무 많아지거나 너무 큰 숫자의 연산이 필요할 때에는

연산 시간을 줄여주기 위해 이분법을 사용합니다!

 

이분법이란 중간값과 찾아야하는 답간의 관계를 비교해보고 begin 값과 end값을 변경해주며 범위를 줄여가는 방법입니다!

 

 

 

다시 시작!

 

먼저 사람수에 대한 begin(최소인원)값과 end(최대인원) 값을 찾아 선언해줍시다. 

그 후 사람수를 (begin + end) /2 로 잡고 건널수 있는지 판단!  

int solution(vector<int> stones,int k) {
    int begin=1;
    int end=0;
    
    for(int i : stones){ // 최댓값 찾기
        end=max(end,i);
    }
    
    while(begin<=end){//둘이 같아지는 시점이 답!
        int mid=(begin+end)/2;
        
        if(find(stones, mid, k)){//건널수 없을때.. --> 좀 더 사람을 줄인다.
            end=mid-1;
        }
        
        else{//건널수 있을때! --> 좀 더 사람을 더 늘려본다 
            begin=mid+1;
        }
    }
    
    return begin;
}

건널수 있다면? 사람을 좀더 늘려야되므로 begin = mid+1 

건널수 없다면? 사람을 좀더 늘려야되므로 end = mid-1

 

이렇게 범위를 조금씩 줄여나가며 탐색을 하면 됩니다!

 

 

 

건널수 있는지 판단하는 함수는 아래처럼 간단합니다....

bool find(vector<int>stones,int mid, int k){
    int count=0;
    for(int i=0;i<stones.size();i++){
        if(stones[i]<= mid){//현재 돌을 밟을수 있는 횟수<=사람수  --> 지나갈수 있는지
            count++;
        }
        else{
            count = 0;
        }
        
        if(count >= k){
            return true;
        }
    }
    return false;
}

 

 

사람 수에 대해서 이분법을 할 생각을 빨리 못해서 늦어졌네요;;

오늘도 하나 배웠습니다!

 

 

ALL

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

using namespace std;

bool find(vector<int>stones,int mid, int k){
    int count=0;
    for(int i=0;i<stones.size();i++){
        if(stones[i]<= mid){//현재 돌을 밟을수 있는 횟수<=사람수  --> 지나갈수 있는지
            count++;
        }
        else{
            count = 0;
        }
        
        if(count >= k){
            return true;
        }
    }
    return false;
}

int solution(vector<int> stones,int k) {
    int begin=1;
    int end=0;
    
    for(int i : stones){
        end=max(end,i);
    }
    
    while(begin<=end){//둘이 같아지는 시점이 답!
        int mid=(begin+end)/2;
        
        if(find(stones, mid, k)){//건널수 없을때.. --> 좀 더 사람을 줄인다.
            end=mid-1;
        }
        
        else{//건널수 있을때! --> 좀 더 사람을 더 늘려본다 
            begin=mid+1;
        }
    }
    
    return begin;
}

틀린점이 있다면 댓 달아주세요!