728x90
반응형

https://programmers.co.kr/learn/courses/30/lessons/42885/

 

코딩테스트 연습 - 구명보트

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다. 예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 5

programmers.co.kr

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다.

예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 50kg]이고 구명보트의 무게 제한이 100kg이라면 2번째 사람과 4번째 사람은 같이 탈 수 있지만 1번째 사람과 3번째 사람의 무게의 합은 150kg이므로 구명보트의 무게 제한을 초과하여 같이 탈 수 없습니다.

구명보트를 최대한 적게 사용하여 모든 사람을 구출하려고 합니다.

사람들의 몸무게를 담은 배열 people과 구명보트의 무게 제한 limit가 매개변수로 주어질 때, 모든 사람을 구출하기 위해 필요한 구명보트 개수의 최솟값을 return 하도록 solution 함수를 작성해주세요.

제한사항

무인도에 갇힌 사람은 1명 이상 50,000명 이하입니다.

각 사람의 몸무게는 40kg 이상 240kg 이하입니다.

구명보트의 무게 제한은 40kg 이상 240kg 이하입니다.

구명보트의 무게 제한은 항상 사람들의 몸무게 중 최댓값보다 크게 주어지므로 사람들을 구출할 수 없는 경우는 없습니다.

입출력 예

people limit return
[70, 50, 80, 50] 100 3
[70, 80, 50] 100 3

내 풀이.

태우고 나가야하니까 태우는 순간 계속 빠진다는 점에서 pop과popleft를 쓰고 싶어 우선순위 큐인 데크로 문제를 풀었다.

 

 

from collections import deque
def solution(people, limit):
    answer = 0
    people.sort(reverse=True)
    q = deque(people)
    while q:
        temp = q.popleft()
        if q and q[-1] <= limit-temp:
            q.pop()
        answer += 1
    return answer

 

런너기법처럼

리스트의 머리부터 뛰면서 체크하는a 와 꼬리부터 뛰면서 체크하는b의 합을 가지고 체크하는 방식의 풀이가 있어 가져왔다.

 

def solution(people, limit) :
    answer = 0
    people.sort()

    a = 0
    b = len(people) - 1
    while a < b :
        if people[b] + people[a] <= limit :
            a += 1
            answer += 1
        b -= 1
    return len(people) - answer
728x90
반응형
728x90
반응형

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/49993

 

코딩테스트 연습 - 스킬트리

 

programmers.co.kr

문제 설명

선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.

예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.

위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.

선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.

제한 조건

스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.

스킬 순서와 스킬트리는 문자열로 표기합니다.

예를 들어, C → B → D 라면 CBD로 표기합니다

선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.

skill_trees는 길이 1 이상 20 이하인 배열입니다.

skill_trees의 원소는 스킬을 나타내는 문자열입니다.

skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.

입출력 예

skill

skill_trees

return

"CBD"

["BACDE", "CBADF", "AECB", "BDA"]

2

정해진 스킬트리의 순서를 인식하고, 그 스킬트리의 순서에 맞게 스킬을 찍고 있는지를 확인하기 위해

skill문자열을 덱으로 바꿔서 popleft() 를 쓸 수 있게 했다.

저렇게 풀다가 막힌 부분이 CBD의 스킬트리에서 D까지 스킬을 다 찍었을 때 q에 더 이상 요소가 남아있지 않아 popleft()실행 시 IndexError가 발생한다는 점이었는데,

try except문을 사용함과 동시에 스킬트리 순서를 지킨 개수를 answer에 하나씩 더하는 것보다

시작과 동시에 주어진 skill_trees의 개수를 answer에 넣어두고 스킬트리 순서를 지키지 않은 개수를 answer에서 하나씩 빼는 방법으로 접근했다. 코드모양이 조금 길어진 것 같지만 다른 접근방법으로 해결할 수 있어서 기뻤다.

 

from collections import deque
def solution(skill, skill_trees):
    answer = len(skill_trees)
    for tree in skill_trees:        
        q = deque(skill)
        s = q.popleft()
        for v in tree:
            if v in q:
                answer -= 1
                break
            elif s == v:
                try:
                    s = q.popleft()
                except IndexError:
                    break
    return answer
728x90
반응형

+ Recent posts