알고리즘 3주차 - 숙제

반응형

Q1. 쓱 최대로 할인 적용하기

Q. 다음과 같이 숫자로 이루어진 배열이 두 개가 있다. 하나는 상품의 가격을 담은 배열이고, 하나는 쿠폰을 담은 배열이다. 쿠폰의 할인율에 따라 상품의 가격을 할인 받을 수 있다. 이 때, 최대한 할인을 많이 받는다면 얼마를 내야 하는가? 단, 할인쿠폰은 한 제품에 한 번씩만 적용 가능하다.
# 코드 스니펫
shop_prices = [30000, 2000, 1500000]
user_coupons = [20, 40]


def get_max_discounted_price(prices, coupons):
    # 이 곳을 채워보세요!
    return 0


print(get_max_discounted_price(shop_prices, user_coupons))  # 926000 이 나와야 합니다.
shop_prices = [30000, 2000, 1500000]
user_coupons = [20, 40]


def get_max_discounted_price(prices, coupons):
    result_dict = {}
    prices.sort(reverse=True)
    coupons.sort(reverse=True)

    # 두 배열의 크기가 다를 수 있게 때문에 while문 사용
    price_index = 0
    coupon_index = 0
    max_discounted_price = 0

    while price_index < len(prices) and coupon_index < len(coupons):
        max_discounted_price += prices[price_index] * (100 - coupons[coupon_index]) / 100
        price_index += 1
        coupon_index += 1

    while price_index < len(prices):
        max_discounted_price += prices[price_index]
        price_index += 1

    return max_discounted_price


print(get_max_discounted_price(shop_prices, user_coupons))  # 926000 이 나와야 합니다.
  • 여기서 핵심: 두 배열의 크기가 다를 수 있는 경우에는 while 문을 사용

Q2. 올바른 괄호

Q. 괄호가 바르게 짝지어졌다는 것은 '(' 문자로 열렸으면 반드시 짝지어서 ')' 문자로 닫혀야 한다는 뜻이다. 예를 들어 ()() 또는 (())() 는 올바르다. )()( 또는 (()( 는 올바르지 않다. 이 때, '(' 또는 ')' 로만 이루어진 문자열 s가 주어졌을 때, 문자열 s가 올바른 괄호이면 True 를 반환하고 아니라면 False 를 반환하시오.
# 코드 스니펫
s = "(())()"


def is_correct_parenthesis(string):
    # 구현해보세요!
    return


print(is_correct_parenthesis(s))  # True 를 반환해야 합니다!

s = "(())()("


def is_correct_parenthesis(string):
    stack = []

    for i in range(len(string)):
        if string[i] == "(":
            stack.append(i) # 어떤 값이 들어가도 상관 없음
        elif string[i] == ")":
            if len(stack) == 0:
                return False
            else:
                stack.pop()
    if len(stack) != 0:
        return False
    else:
        return True


print(is_correct_parenthesis(s))  # True 를 반환해야 합니다!
  • 순차적으로 들어가야 하는 구조같으면 스택을 사용

Q3. 멜론 베스트 앨범 뽑기

Q. 멜론에서 장르 별로 가장 많이 재생된 노래를 두 개씩 모아 베스트 앨범을 출시하려 한다. 노래는 인덱스 구분하며, 노래를 수록하는 기준은 다음과 같다. 속한 노래가 많이 재생된 장르를 먼저 수록한다. 장르 내에서 많이 재생된 노래를 먼저 수록한다. 장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록한다. 노래의 장르를 나타내는 문자열 배열 genres와 노래별 재생 횟수를 나타내는 정수 배열 plays가 주어질 때, 베스트 앨범에 들어갈 노래의 인덱스를 순서대로 반환하시오.
# 코드 스니펫

genres = ["classic", "pop", "classic", "classic", "pop"]
plays = [500, 600, 150, 800, 2500]


def get_melon_best_album(genre_array, play_array):
    # 구현해보세요!
    return []


print(get_melon_best_album(genres, plays))  # 결과로 [4, 1, 3, 0] 가 와야 합니다!
genres = ["classic", "pop", "classic", "classic", "pop"]
plays = [500, 600, 150, 800, 2500]

# 장르 별로 재생된 횟수를 저장해야함
# 장르 별로 곡의 정보(인덱스, 재생횟수)를 배열로 묶어 저장한다.
def get_melon_best_album(genre_array, play_array):
    dict = {}
    genre_index_play_array = {}
    for i in range(len(genre_array)):
        genre = genre_array[i]
        play = play_array[i]
        if genre not in dict:
            dict[genre] = play
            genre_index_play_array[genre] = [[i, play]]
        else:
            dict[genre] += play
            genre_index_play_array[genre].append([i, play])

    sorted_dict = sorted(dict.items(), key=lambda item: item[1], reverse=True)
    result = []
    for genre, _value in sorted_dict:
        index_play_array = genre_index_play_array[genre]
        sorted_index_play_array = sorted(index_play_array, key=lambda item:item[1], reverse=True)

        for i in range(len(sorted_index_play_array)):
            if i > 1:
                break
            result.append(sorted_index_play_array[i][0])
    return result


print(get_melon_best_album(genres, plays))  # 결과로 [4, 1, 3, 0] 가 와야 합니다!
반응형