통계러버

문제 링크 

• 더 맵게

문제 설명

섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)

입출력 예 설명

1. 스코빌 지수가 1인 음식과 2인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
   새로운 음식의 스코빌 지수 = 1 + (2 * 2) = 5
   가진 음식의 스코빌 지수 = [5, 3, 9, 10, 12]
2. 스코빌 지수가 3인 음식과 5인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
   새로운 음식의 스코빌 지수 = 3 + (5 * 2) = 13
   가진 음식의 스코빌 지수 = [13, 9, 10, 12]

모든 음식의 스코빌 지수가 7 이상이 되었고 이때 섞은 횟수는 2회입니다.

힙

힙(heaps)

• 성질 : 최대/최소 원소를 빠르게 찾을 수 있음.
• 연산
     *   힙 구성(heapify) : O(NlogN)      # 삽입을 N번 하는 꼴로 이해하자.
     *   삽입(insert) : O(logN) : 삽입후에도 log에 비례하는 시간복잡도를 사용해서 min heap(or max heap)구조를 유지하도록 함.
    *   삭제(remove) : O(logN) : 삭제후에도 log에 비례하는 시간복잡도를 사용해서 min heap(or max heap)구조를 유지하도록 함.

코드

sol1)  min_heap을 이용한 우선순위 큐

def solution(scoville, K):
    import heapq as hq
    min_heap = []
    for i in scoville:
        hq.heappush(min_heap, i)                 #O(NlogN)
        
    cnt = 0    
    while min_heap[0] < K and len(min_heap)>1 :  # 힙구조가 되면 가장 처음값은 최솟값이 됨
        fist_min = hq.heappop(min_heap)
        second_min = hq.heappop(min_heap)
        new_sco = fist_min + second_min*2
        hq.heappush(min_heap, new_sco)
        cnt += 1
    return -1 if min_heap[0] < K else cnt

sol2) 

hq.heapify()를 이용하여 리스트를 min heap 형태(맨 앞이 항상 최솟값이 오게 됨)로 바꿔서 푸는 풀이

• hq.heapify() : 리스트를 min heap형태로 바꿈. 시간복잡도: O(logN)
• 주의! min_heap = hq.heapify(scoville) 이렇게 객체로 받아주면 안됨! 오류남

def solution(scoville, K):
    import heapq as hq
    hq.heapify(scoville)  # 리스트 scoville를 min heap으로 바꿈 
        
    cnt = 0    
    while scoville[0] < K and len(scoville)>1 :  # 힙구조가 되면 가장 처음 나오는 값은 최솟값이 됨
        fist_min = hq.heappop(scoville)
        second_min = hq.heappop(scoville)
        new_sco = fist_min + second_min*2
        hq.heappush(scoville, new_sco)
        cnt += 1
    return -1 if scoville[0] < K else cnt

sol3)

def solution(scoville, K):
    import heapq as hq
    hq.heapify(scoville)  # 리스트 scoville를 min heap으로 바꿈   O(logN)
     
    cnt = 0    
    while True:  # O(nlogn)
        fist_min = hq.heappop(scoville)  #O(logn)
        if fist_min >= K:
            break
        elif len(scoville) == 0:
            cnt = -1
            break      
        second_min = hq.heappop(scoville)  #O(logn)
        new_sco = fist_min + second_min*2
        hq.heappush(scoville, new_sco)
        cnt += 1
    return cnt

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/42883

문제 설명

어떤 숫자에서 k개의 수를 제거했을 때 얻을 수 있는 가장 큰 숫자를 구하려 합니다.

탐욕법인 이유

• 앞 단계에서의 선택(앞자리에 큰수)이 이후 단계에서의 동작에 의한 해(solution)의 최적성(optimality)에 영향을 주지 않음

솔루션

주어진 숫자(number)에서 왼쪽부터 하나씩 꺼내어 모으되,

1. 이때, 이미 모아둔 것 중 지금 등장하는 것보다 작은 것들은 빼낸다.
2. 단, 이때 유효한 숫자를 모두 소진할 때 까지만 뺀다.
3. 아직 뺄 개수(k)를 채우지 못한 경우 뒤에서 부터 남은 자리수만큼 뺀다.

코드

sol1) 스택을 이용한 풀이

def solution(number, k):
    stack = [number[0]]
    for num in number[1:]:
        while len(stack) > 0 and stack[-1] < num and k > 0: # 파이썬에서 문자열의 대소관계는 아스키 코드를 따라 간다. 이를 이용하면 문자열로 된 숫자를 대소비교히면 숫자일때의 대소 비교때와 같아진다.
            k -= 1
            stack.pop()
        stack.append(num)
    if k != 0:   # while문을 다 돌았는데 k가 0이 아니면 뒤에서부터 pop한다
        stack = stack[:-k]
    return ''.join(stack)

sol2) 스택을 이용한 풀이2

def solution(number, k):
    stack = []
    for x in number:
        while stack and k >0 and stack[-1] < x:  # 파이썬에서 문자열의 대소관계는 아스키 코드를 따라 간다. 이를 이용하면 문자열로 된 숫자를 대소비교히면 숫자일때의 대소 비교때와 같아진다.
            stack.pop()
            k -= 1
        stack.append(x) # 맨 처음에는 stack이 비어있으므로 while문 돌지 않고 x부터 append함
    if k!=0:
        stack = stack[:-k]
    res = ''.join(map(str, stack))  
    return res

sol3) 스택과 큐를 이용한 풀이

from collections import deque
def solution(number, k):
    cnt = 0
    q= deque(number)
    sol = []
    sol.append(q.popleft()) #number의 첫글자를 stack에 넣고 시작
    while q:
        x = q.popleft()
        while sol and x > sol[-1] and cnt !=k:  # 스택의 맨위 글자와 큐의 왼쪽 첫번째 글자를 비교해서 스택의 맨위 글자가 더 작으면 크거나 같아 질때까지 스택의 맨 위 글자를 pop한다. 이 때 pop을하다 pop의 숫자가 k가 되면 멈춘다
                                                # 또한, 스택이 비어있으면 pop을 못하므로 비게 되면 멈춘다
                                                #  파이썬에서 문자열의 대소관계는 아스키 코드를 따라 간다. 이를 이용하면 문자열로 된 숫자를 대소비교히면 숫자일때의 대소 비교때와 같아진다.
            sol.pop()
            cnt +=1 
        sol.append(x)                           
        while (not q) and (cnt != k) :         #만약 q의 원소를 모두 썼는데 cnt에 도달 못했으면 도달할때까지 뒤에서부터 빼냄
            sol.pop()
            cnt +=1
    return ''.join(sol)

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/42862

문제 설명

제한사항

• 전체 학생의 수는 2명 이상 30명 이하입니다.
• 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

입출력 예

입출력 예 설명

예제 #1
1번 학생이 2번 학생에게 체육복을 빌려주고, 3번 학생이나 5번 학생이 4번 학생에게 체육복을 빌려주면 학생 5명이 체육수업을 들을 수 있습니다.

예제 #2
3번 학생이 2번 학생이나 4번 학생에게 체육복을 빌려주면 학생 4명이 체육수업을 들을 수 있습니다.

솔루션

위에 처럼 여벌의 체육복을 지닌 학생들이 오름차순으로 정렬 되어 있을때, 나보다 번호가 1번 높은 체육복이 없는 학생에게 체육복을 주면 5명이 체육복을 지니게 됨.

반면, 나보다 번호가 1번 낮은 체육복이 없는 학생에게 체육복을 주면 6명이 체육복을 지니게 됨.

따라서 여벌의 체육복을 지닌 학생이 오름차순으로 정렬되어있을때는 먼저 자신보다 번호가 1 낮은 체육복이 없는 학생에게 체육복을 주고, 번호가 1 낮은 체육복이 없는 학생이 없으면 번호가 1 높은 체육복이 없는 학생에게 체육복을 주어야 한다. 이 순서가 바뀌게 되면 체육복을 지닌 학생 수의 최댓값을 구할 수 없다.

코드

sol1)

def solution(n, lost, reserve):
    u = [1]* (n+2) # 학생수보다 2 큰 리스트를 만든다. 사람수는 n명이지만 for문을 돌면서 앞에 혹은 뒷 사람에게 체육복을 빌려주는 행위(-1)을 해야 하므로 n+2개를 만든다
    for i in reserve:  # O(n)
        u[i] +=1 # 여벌의 옷을 가지고 있는 학생들은 체육복의 수가 2임 
    for i in lost:     # O(n)
        u[i] -= 1 # 옷을 가지고 있지 않은 학생들은 체육복의 수가 0임
    
    for j in range(1,n+1): # for문은 1번부터 n번의 학생들만 돈다  # O(n)
                           # 사람이 오름차순으로 정렬되어있으므로 탐욕법을 적용하려면 나보다 번호가 1 낮은 사람이 체육복을 가지고 있는지를 먼저 살펴야 함   
        if u[ j - 1] == 0 and u[j] == 2:  # 내가 2벌을 가지고 있고 나보다 번호가 1 작은 사람이 체육복이 없다면 빌려준다
            u[j-1 : j+1] = [1,1]
        elif  u[j] == 2 and u[j + 1] ==0:  # 내가 2벌을 가지고 있고 나보다 번호가 1 큰 사람이 체육복이 없다면 빌려준다
            u[j : j+2] = [1, 1]
            
    return len([x for x in u[1:-1] if x > 0]) # 1번 부터 n번의 학생들 중에서 숫자가 1인 사람들만 모은 리스트의 길이를 구한다  # O(n)

sol2) n은 매우 큰데 여벌의 체육복을 가져온 학생수가 매우 작으면 해당 풀이가 더 적합하다.

def solution(n, lost, reserve):
    
    s = set(lost) & set(reserve)  # 여벌의 체육복을 가지고 왔지만 도난당해서 다른사람에게 빌려줄 수 없는 사람들
    lost_only = set(lost) - s # 체육복을 빌려야 하는 사람들
    reserve_only = set(reserve) - s # 여벌의 체육복을 가진 사람들
    
    for x in sorted(reserve_only):   # 여벌의 체육복을 지닌 사람들을 오름차순으로 정렬했으므로 탐욕법을 적용하려면 나보다 번호가 1 낮은 사람이 체육복을 가지고 있는지를 먼저 살펴야 함   # O(KlogK)
        if  x-1 in lost_only:
            lost_only.remove(x-1)
        elif x+1 in lost_only:
            lost_only.remove(x+1)
    return n - len(lost_only)

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

• 완주하지 못한 선수

문제 설명

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한사항

• 마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.
• completion의 길이는 participant의 길이보다 1 작습니다.
• 참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.
• 참가자 중에는 동명이인이 있을 수 있습니다.

입출력 예

입출력 예 설명

예제 #1
"leo"는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #2
"vinko"는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #3
"mislav"는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

코드

sol1) 딕셔너리를 이용한 풀이

def solution(participant, completion):
    temp = {} # key,value를 가진 딕셔너리 이용

    #딕셔너리에 참가자 이름(key)과 해당 이름 사람수(value) 넣기 
    for i in participant :
        if i in temp : # 딕셔너리에서 in을 쓰면 key값이 있는지를 묻는 것
            temp[i]+=1
        else: 
            temp[i] =1
    
    #완주자 이름(key)의 value가 1이면 지우기, 동명이인이면 -1  
    for i in completion :
        if temp[i]==1 : 
            del temp[i]
        else : 
            temp[i] -=1

    #딕셔너리를 리스트로 바꾸고 가장 첫번째꺼 리턴(어차피 하나뿐)
    return list(temp.keys())[0]

sol2) 딕셔너리의 get을 이용한 풀이

• dic.get(key) : key값에 대응하는 value값을 출력한다. 이는 dic[key]와 같으나 만약 없는 key값을 출력하라고 하면 dic[key]는 error를 출력하고, dic.get(key)는 None을 출력함. 만약 해당 key가 없을 때 반환되는 값을 바꾸 고 싶다면, get() 메소드의 두번째 인자로 주면 됨.

def solution(participant, completion):
    dic = {}
    for x in participant:
        dic[x] = dic.get(x,0) + 1 # dic에 key값으로 x가 있으면 x에 대응하는 value를 출력하고 없으면 0을 출력 #O(n)
    for x in completion:
        dic[x] -= 1                # O(n)
    sol = [k for k, v in dic.items() if v > 0]  #O(n)
    answer = sol[0]
    return answer

sol3) 딕셔너리를 사용한 후 sorted를 이용한 풀이

def solution(participant, completion):
    dic = {}
    for i in participant:
        if i not in dic:
            dic[i] = 1
        else:
            dic[i] += 1
    for j in completion:
        dic[j] -= 1
        
    sol = sorted(dic.items(), key = lambda x: x[1], reverse = True)
    return sol[0][0]

sol4) zip을 이용한 풀이

• zip : 배열을 같은 인덱스끼리 짝지어준다. 만약 배열의 길이가 다를 경우 같은 인덱스끼리만 짝지어주고, zip 객체에서 나머지 인덱스는 제외된다.

def solution(participant, completion):
    participant.sort()  # O(nlogn)
    completion.sort()   # O(nlogn)
    for p,c in zip(participant, completion):
        if p != c:
            return p
    return participant[-1] # 모두 일치하면 사람수가 하나 더 많은 participant의 마지막 번호의 사람이 완주못한 것

sol5) Counter()를 이용한 풀이

• collections.Counter: 원소의 수를 딕셔너리 형태로 출력함. 딕셔너리는 차집합이 불가능하지만 Counter내의 딕셔너리는 차집합이 가능함.

import collections
def solution(participant, completion):
    sol=collections.Counter(participant) - collections.Counter(completion)
    return list(sol.keys())[0]

sol6) 해시함수 이용 2번째 방법

• hash('a') # 해당 value에 해당하는 hash값을 출력해줌

       >>> 3338796687179829302

def solution(participant , completion):
    Dic ={}
    sum = 0
    for part in participant:
        Dic[hash(part)] = part  # {숫자:'참가자'} 형태의 딕셔너리를 만듦
        sum += hash(part)       # 해당 참가자에 해당하는 key겂인 숫자를 더해준다
        
    for comp in completion:
        sum -=hash(comp)       # 완주한 사람에서 해당하는 key값인 숫자를 빼준다
    return Dic[sum]          # 남은 숫자의 key에 대응하는 값이 완주 못한 사람임

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

• 최솟값 만들기

문제 설명

길이가 같은 배열 A, B 두개가 있습니다. 각 배열은 자연수로 이루어져 있습니다.
배열 A, B에서 각각 한 개의 숫자를 뽑아 두 수를 곱합니다. 이러한 과정을 배열의 길이만큼 반복하며, 두 수를 곱한 값을 누적하여 더합니다. 이때 최종적으로 누적된 값이 최소가 되도록 만드는 것이 목표입니다. (단, 각 배열에서 k번째 숫자를 뽑았다면 다음에 k번째 숫자는 다시 뽑을 수 없습니다.)

예를 들어 A = [1, 4, 2] , B = [5, 4, 4] 라면

• A에서 첫번째 숫자인 1, B에서 첫번째 숫자인 5를 뽑아 곱하여 더합니다. (누적된 값 : 0 + 5(1x5) = 5)
• A에서 두번째 숫자인 4, B에서 세번째 숫자인 4를 뽑아 곱하여 더합니다. (누적된 값 : 5 + 16(4x4) = 21)
• A에서 세번째 숫자인 2, B에서 두번째 숫자인 4를 뽑아 곱하여 더합니다. (누적된 값 : 21 + 8(2x4) = 29)

즉, 이 경우가 최소가 되므로 29를 return 합니다.

배열 A, B가 주어질 때 최종적으로 누적된 최솟값을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 배열 A, B의 크기 : 1,000 이하의 자연수
• 배열 A, B의 원소의 크기 : 1,000 이하의 자연수

입출력 예

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

입출력 예 #2
A에서 첫번째 숫자인 1, B에서 두번째 숫자인 4를 뽑아 곱하여 더합니다. (누적된 값 : 4) 다음, A에서 두번째 숫자인 2, B에서 첫번째 숫자인 3을 뽑아 곱하여 더합니다. (누적된 값 : 4 + 6 = 10)
이 경우가 최소이므로 10을 return 합니다.

솔루션

⇒ A의 가장 작은 값과 B의 가장 큰 값을 순서대로 곱해서 더하는 형태

⇒ A와 B를 하나는 오름차순, 하나는 내림차순으로 정렬한 후 각 원소끼리 곱해서 더하자

코드

sol1

def solution(A,B):
    n = len(A)
    A = sorted(A)
    B = sorted(B, reverse = True)
    answer = 0
    for i in range(n):
        answer += A[i]*B[i]
    return answer

sol2

def solution(A,B):
    return sum(a*b for a, b in zip(sorted(A), sorted(B, reverse = True)))

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

• 기능개발

문제 설명

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 100%일 때 서비스에 반영할 수 있습니다.

또, 각 기능의 개발속도는 모두 다르기 때문에 뒤에 있는 기능이 앞에 있는 기능보다 먼저 개발될 수 있고, 이때 뒤에 있는 기능은 앞에 있는 기능이 배포될 때 함께 배포됩니다.

먼저 배포되어야 하는 순서대로 작업의 진도가 적힌 정수 배열 progresses와 각 작업의 개발 속도가 적힌 정수 배열 speeds가 주어질 때 각 배포마다 몇 개의 기능이 배포되는지를 return 하도록 solution 함수를 완성하세요.

제한 사항

• 작업의 개수(progresses, speeds배열의 길이)는 100개 이하입니다.
• 작업 진도는 100 미만의 자연수입니다.
• 작업 속도는 100 이하의 자연수입니다.
• 배포는 하루에 한 번만 할 수 있으며, 하루의 끝에 이루어진다고 가정합니다. 예를 들어 진도율이 95%인 작업의 개발 속도가 하루에 4%라면 배포는 2일 뒤에 이루어집니다.

입출력 예

입출력 예 설명

입출력 예 #1
첫 번째 기능은 93% 완료되어 있고 하루에 1%씩 작업이 가능하므로 7일간 작업 후 배포가 가능합니다.
두 번째 기능은 30%가 완료되어 있고 하루에 30%씩 작업이 가능하므로 3일간 작업 후 배포가 가능합니다. 하지만 이전 첫 번째 기능이 아직 완성된 상태가 아니기 때문에 첫 번째 기능이 배포되는 7일째 배포됩니다.
세 번째 기능은 55%가 완료되어 있고 하루에 5%씩 작업이 가능하므로 9일간 작업 후 배포가 가능합니다.

따라서 7일째에 2개의 기능, 9일째에 1개의 기능이 배포됩니다.

입출력 예 #2
모든 기능이 하루에 1%씩 작업이 가능하므로, 작업이 끝나기까지 남은 일수는 각각 5일, 10일, 1일, 1일, 20일, 1일입니다. 어떤 기능이 먼저 완성되었더라도 앞에 있는 모든 기능이 완성되지 않으면 배포가 불가능합니다.

따라서 5일째에 1개의 기능, 10일째에 3개의 기능, 20일째에 2개의 기능이 배포됩니다.

코드

def solution(progress, speeds):
    import math
    from collections import deque
    q = deque()
    answer = []
    for i in range(len(progress)):
        q.append(math.ceil((100 - progress[i])/speeds[i]))

    while q:
        a = q.popleft()
        cnt = 1
        for _ in range(len(q)): 
            if a >= q[0]:
                q.popleft()
                cnt += 1
            else:
                break   
        answer.append(cnt)  
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제 링크

• 예산

문제 설명

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.

물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.

부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

제한 사항

• d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
• d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
• budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.

입출력 예

입출력 예 설명

입출력 예 #1
각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.

• 1원, 2원, 3원을 신청한 부서의 물품을 구매해주려면 6원이 필요합니다.
• 1원, 2원, 5원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
• 1원, 3원, 4원을 신청한 부서의 물품을 구매해주려면 8원이 필요합니다.
• 1원, 3원, 5원을 신청한 부서의 물품을 구매해주려면 9원이 필요합니다.

3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.

입출력 예 #2
모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.

코드

sol1)

# budget에서 d의 원소가 작은것부터 뺌. budget이 0보다 작아질때까지 뺌
def solution(d, budget):
    cnt = 0
    d.sort()
    for i in d:
        budget -= i
        if budget < 0:
            break
        cnt +=1
    return cnt

sol2) pop을 이용한 풀이

def solution(d, budget):
    d.sort()  # 오름차순 정렬
    while budget < sum(d): # 위와 반대로 가장 큰 것들을 빼면서 남은 d의 원소들의 길이를 return함 
        d.pop()  # 가장 왼쪽의 원소를 삭제함
    return len(d)

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges

문제링크

• 게임 맵 최단거리

문제설명

ROR 게임은 두 팀으로 나누어서 진행하며, 상대 팀 진영을 먼저 파괴하면 이기는 게임입니다. 따라서, 각 팀은 상대 팀 진영에 최대한 빨리 도착하는 것이 유리합니다.

지금부터 당신은 한 팀의 팀원이 되어 게임을 진행하려고 합니다. 다음은 5 x 5 크기의 맵에, 당신의 캐릭터가 (행: 1, 열: 1) 위치에 있고, 상대 팀 진영은 (행: 5, 열: 5) 위치에 있는 경우의 예시입니다.

위 그림에서 검은색 부분은 벽으로 막혀있어 갈 수 없는 길이며, 흰색 부분은 갈 수 있는 길입니다. 캐릭터가 움직일 때는 동, 서, 남, 북 방향으로 한 칸씩 이동하며, 게임 맵을 벗어난 길은 갈 수 없습니다. 아래 예시는 캐릭터가 상대 팀 진영으로 가는 두 가지 방법을 나타내고 있습니다.

첫 번째 방법은 11개의 칸을 지나서 상대 팀 진영에 도착했습니다.

두 번째 방법은 15개의 칸을 지나서 상대팀 진영에 도착했습니다.

위 예시에서는 첫 번째 방법보다 더 빠르게 상대팀 진영에 도착하는 방법은 없으므로, 이 방법이 상대 팀 진영으로 가는 가장 빠른 방법입니다.

만약, 상대 팀이 자신의 팀 진영 주위에 벽을 세워두었다면 상대 팀 진영에 도착하지 못할 수도 있습니다. 예를 들어, 다음과 같은 경우에 당신의 캐릭터는 상대 팀 진영에 도착할 수 없습니다.

게임 맵의 상태 maps가 매개변수로 주어질 때, 캐릭터가 상대 팀 진영에 도착하기 위해서 지나가야 하는 칸의 개수의 최솟값을 return 하도록 solution 함수를 완성해주세요. 단, 상대 팀 진영에 도착할 수 없을 때는 -1을 return 해주세요.

제한사항

• maps는 n x m 크기의 게임 맵의 상태가 들어있는 2차원 배열로, n과 m은 각각 1 이상 100 이하의 자연수입니다.
  • n과 m은 서로 같을 수도, 다를 수도 있지만, n과 m이 모두 1인 경우는 입력으로 주어지지 않습니다.
• maps는 0과 1로만 이루어져 있으며, 0은 벽이 있는 자리, 1은 벽이 없는 자리를 나타냅니다.
• 처음에 캐릭터는 게임 맵의 좌측 상단인 (1, 1) 위치에 있으며, 상대방 진영은 게임 맵의 우측 하단인 (n, m) 위치에 있습니다.

입출력 예

입출력 예 설명

예제 #1
주어진 데이터는 다음과 같습니다.

캐릭터가 적 팀의 진영까지 이동하는 가장 빠른 길은 다음 그림과 같습니다.

따라서 총 11칸을 캐릭터가 지나갔으므로 11을 return 하면 됩니다.

예제 #2

문제의 예시와 같으며, 상대 팀 진영에 도달할 방법이 없습니다. 따라서 -1을 return 합니다.

코드

def bfs(maps):
    from collections import deque
    n=len(maps)
    m =len(maps[0])
    visit = [[False]*m for _ in range(n)]
    
    dx = [-1,1,0,0]
    dy = [0,0,-1,1]
    
    
    q = deque()
    q.append((0,0,1))
    visit[0][0] = True
    while q:
        y, x, d =q.popleft()
        if y == n-1 and x == m-1:
            return d
        else:
            for k in range(4):
                nx = x+dx[k]
                ny = y+dy[k]
                if (0 <= nx < m) and (0 <= ny <n) and (visit[ny][nx] == False) and (maps[ny][nx] ==1):
                    q.append((ny,nx,d+1))
                    visit[ny][nx] = True
    return -1


def solution(maps):
    sol = bfs(maps)
    return sol

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges