[Python] 알고리즘 연습 11 (프로그래머스)

푸드 파이트 대회 (연습문제)

• 수웅이는 매달 주어진 음식을 빨리 먹는 푸드 파이트 대회를 개최합니다. 이 대회에서 선수들은 1대 1로 대결하며, 매 대결마다 음식의 종류와 양이 바뀝니다. 대결은 준비된 음식들을 일렬로 배치한 뒤, 한 선수는 제일 왼쪽에 있는 음식부터 오른쪽으로, 다른 선수는 제일 오른쪽에 있는 음식부터 왼쪽으로 순서대로 먹는 방식으로 진행됩니다. 중앙에는 물을 배치하고, 물을 먼저 먹는 선수가 승리하게 됩니다.
  
  이때, 대회의 공정성을 위해 두 선수가 먹는 음식의 종류와 양이 같아야 하며, 음식을 먹는 순서도 같아야 합니다. 또한, 이번 대회부터는 칼로리가 낮은 음식을 먼저 먹을 수 있게 배치하여 선수들이 음식을 더 잘 먹을 수 있게 하려고 합니다. 이번 대회를 위해 수웅이는 음식을 주문했는데, 대회의 조건을 고려하지 않고 음식을 주문하여 몇 개의 음식은 대회에 사용하지 못하게 되었습니다.
  
  예를 들어, 3가지의 음식이 준비되어 있으며, 칼로리가 적은 순서대로 1번 음식을 3개, 2번 음식을 4개, 3번 음식을 6개 준비했으며, 물을 편의상 0번 음식이라고 칭한다면, 두 선수는 1번 음식 1개, 2번 음식 2개, 3번 음식 3개씩을 먹게 되므로 음식의 배치는 "1223330333221"이 됩니다. 따라서 1번 음식 1개는 대회에 사용하지 못합니다.
  
  수웅이가 준비한 음식의 양을 칼로리가 적은 순서대로 나타내는 정수 배열 food가 주어졌을 때, 대회를 위한 음식의 배치를 나타내는 문자열을 return 하는 solution 함수를 완성해주세요.
  
  
  • 제한사항
    • 2 ≤ food의 길이 ≤ 9
    • 1 ≤ food의 각 원소 ≤ 1,000
    • food에는 칼로리가 적은 순서대로 음식의 양이 담겨 있습니다.
    • food[i]는 i번 음식의 수입니다.
    • food[0]은 수웅이가 준비한 물의 양이며, 항상 1입니다.
    • 정답의 길이가 3 이상인 경우만 입력으로 주어집니다.

제출한 코드 :

def solution(food):
    left_side = []
    
    # 왼쪽 부분 음식 배치
    for i in range(1, len(food)):
        left_side.extend([str(i)] * (food[i] // 2))
    
    # 중앙의 물 배치
    center = '0'
    
    # 오른쪽 부분 음식 배치 (왼쪽 부분을 반대로)
    right_side = left_side[::-1]
    
    answer = ''.join(left_side) + center + ''.join(right_side)
    
    return answer

---

콜라 문제 (연습문제)

• 오래전 유행했던 콜라 문제가 있습니다. 콜라 문제의 지문은 다음과 같습니다.
  

  정답은 아무에게도 말하지 마세요.
  
  콜라 빈 병 2개를 가져다주면 콜라 1병을 주는 마트가 있다. 빈 병 20개를 가져다주면 몇 병을 받을 수 있는가?
  
  단, 보유 중인 빈 병이 2개 미만이면, 콜라를 받을 수 없다.
  

  문제를 풀던 상빈이는 콜라 문제의 완벽한 해답을 찾았습니다. 상빈이가 푼 방법은 아래 그림과 같습니다. 우선 콜라 빈 병 20병을 가져가서 10병을 받습니다. 받은 10병을 모두 마신 뒤, 가져가서 5병을 받습니다. 5병 중 4병을 모두 마신 뒤 가져가서 2병을 받고, 또 2병을 모두 마신 뒤 가져가서 1병을 받습니다. 받은 1병과 5병을 받았을 때 남은 1병을 모두 마신 뒤 가져가면 1병을 또 받을 수 있습니다. 이 경우 상빈이는 총 10 + 5 + 2 + 1 + 1 = 19병의 콜라를 받을 수 있습니다.
  

  

  
  문제를 열심히 풀던 상빈이는 일반화된 콜라 문제를 생각했습니다. 이 문제는 빈 병 a개를 가져다주면 콜라 b병을 주는 마트가 있을 때, 빈 병 n개를 가져다주면 몇 병을 받을 수 있는지 계산하는 문제입니다. 기존 콜라 문제와 마찬가지로, 보유 중인 빈 병이 a개 미만이면, 추가적으로 빈 병을 받을 순 없습니다. 상빈이는 열심히 고심했지만, 일반화된 콜라 문제의 답을 찾을 수 없었습니다. 상빈이를 도와, 일반화된 콜라 문제를 해결하는 프로그램을 만들어 주세요.
  
  콜라를 받기 위해 마트에 주어야 하는 병 수 a, 빈 병 a개를 가져다 주면 마트가 주는 콜라 병 수 b, 상빈이가 가지고 있는 빈 병의 개수 n이 매개변수로 주어집니다. 상빈이가 받을 수 있는 콜라의 병 수를 return 하도록 solution 함수를 작성해주세요.
  • 제한사항
    • 1 ≤ b < a ≤ n ≤ 1,000,000
    • 정답은 항상 int 범위를 넘지 않게 주어집니다.

제출한 코드 :

def solution(a, b, n):
    total_coke = 0
    while n >= a:
        new_coke = (n // a) * b
        total_coke = total_coke + new_coke
        n = (n % a) + new_coke
    return total_coke

---

명예의 전당(1) (연습문제)

• "명예의 전당"이라는 TV 프로그램에서는 매일 1명의 가수가 노래를 부르고, 시청자들의 문자 투표수로 가수에게 점수를 부여합니다. 매일 출연한 가수의 점수가 지금까지 출연 가수들의 점수 중 상위 k번째 이내이면 해당 가수의 점수를 명예의 전당이라는 목록에 올려 기념합니다. 즉 프로그램 시작 이후 초기에 k일까지는 모든 출연 가수의 점수가 명예의 전당에 오르게 됩니다. k일 다음부터는 출연 가수의 점수가 기존의 명예의 전당 목록의 k번째 순위의 가수 점수보다 더 높으면, 출연 가수의 점수가 명예의 전당에 오르게 되고 기존의 k번째 순위의 점수는 명예의 전당에서 내려오게 됩니다.
  
  이 프로그램에서는 매일 "명예의 전당"의 최하위 점수를 발표합니다. 예를 들어, k = 3이고, 7일 동안 진행된 가수의 점수가 [10, 100, 20, 150, 1, 100, 200]이라면, 명예의 전당에서 발표된 점수는 아래의 그림과 같이 [10, 10, 10, 20, 20, 100, 100]입니다.
  명예의 전당 목록의 점수의 개수 k, 1일부터 마지막 날까지 출연한 가수들의 점수인 score가 주어졌을 때, 매일 발표된 명예의 전당의 최하위 점수를 return하는 solution 함수를 완성해주세요.

  • 제한사항
    - 3 ≤ k ≤ 100
    - 7 ≤ score의 길이 ≤ 1,000
    • 0 ≤ score[i] ≤ 2,000
      

제출한 코드 :

def solution(k, score):
    hall_of_fame = []  # 명예의 전당 리스트
    result = []  # 매일 발표되는 명예의 전당의 최하위 점수 리스트
    
    for s in score:
        hall_of_fame.append(s)
        hall_of_fame.sort(reverse=True)
        
        # 상위 k번째까지의 점수를 유지
        if len(hall_of_fame) > k:
            hall_of_fame.pop()
        
        # 현재 명예의 전당의 최하위 점수를 결과 리스트에 추가
        result.append(hall_of_fame[-1])
    
    return result

---

2016년 (연습문제)

• 2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT

입니다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 반환하세요.

• 제한사항
  • 2016년은 윤년입니다.
  • 2016년 a월 b일은 실제로 있는 날입니다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)

제출한 코드 :

def solution(a, b):
    days_in_month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    days_of_week = ["FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"]
    
    # 해당 날짜까지의 총 일수 계산
    total_days = sum(days_in_month[:a-1]) + b - 1
    
    # 요일 계산 (2016년 1월 1일이 금요일이므로 총 일수에서 7로 나눈 나머지를 이용)
    day_of_week = days_of_week[total_days % 7]
    
    return day_of_week

---

카드 뭉치 (연습문제)

• 코니는 영어 단어가 적힌 카드 뭉치 두 개를 선물로 받았습니다. 코니는 다음과 같은 규칙으로 카드에 적힌 단어들을 사용해 원하는 순서의 단어 배열을 만들 수 있는지 알고 싶습니다.
  
  - 원하는 카드 뭉치에서 카드를 순서대로 한 장씩 사용합니다.
  - 한 번 사용한 카드는 다시 사용할 수 없습니다.
  - 카드를 사용하지 않고 다음 카드로 넘어갈 수 없습니다.
  - 기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없습니다.
  
  예를 들어 첫 번째 카드 뭉치에 순서대로 ["i", "drink", "water"], 두 번째 카드 뭉치에 순서대로 ["want", "to"]가 적혀있을 때 ["i", "want", "to", "drink", "water"] 순서의 단어 배열을 만들려고 한다면 첫 번째 카드 뭉치에서 "i"를 사용한 후 두 번째 카드 뭉치에서 "want"와 "to"를 사용하고 첫 번째 카드뭉치에 "drink"와 "water"를 차례대로 사용하면 원하는 순서의 단어 배열을 만들 수 있습니다.
  문자열로 이루어진 배열 cards1, cards2와 원하는 단어 배열 goal이 매개변수로 주어질 때, cards1과 cards2에 적힌 단어들로 goal를 만들 있다면 "Yes"를, 만들 수 없다면 "No"를 return하는 solution 함수를 완성해주세요.
  
  
  • 제한사항
    • 1 ≤ cards1의 길이, cards2의 길이 ≤ 10
       # 1 ≤ cards1[i]의 길이, cards2[i]의 길이 ≤ 10
       # cards1과 cards2에는 서로 다른 단어만 존재합니다.
    • 2 ≤ goal의 길이 ≤ cards1의 길이 + cards2의 길이
       # 1 ≤ goal[i]의 길이 ≤ 10
       # goal의 원소는 cards1과 cards2의 원소들로만 이루어져 있습니다.
    • cards1, cards2, goal의 문자열들은 모두 알파벳 소문자로만 이루어져 있습니다.

제출한 코드 :

def solution(cards1, cards2, goal):
    index1, index2 = 0, 0
    
    for word in goal:
        if index1 < len(cards1) and cards1[index1] == word:
            index1 += 1
        elif index2 < len(cards2) and cards2[index2] == word:
            index2 += 1
        else:
            return "No"
    
    return "Yes"

---

오늘의 TMI


밤샘은 할게 못 된다. 정말로...