와챠의 우당탕탕 코딩 일기장

[알고리즘] 스파르타 코딩 : 알고리즘 - 3주차(숙제) 본문

이런 저런 공부

[알고리즘] 스파르타 코딩 : 알고리즘 - 3주차(숙제)

minWachya 2021. 7. 10. 15:28
반응형

숙제

Q1. 다음과 같이 숫자로 이루어진 배열이 두 개가 있다.

하나는 상품의 가격을 담은 배열이고, 하나는 쿠폰을 담은 배열이다.

쿠폰의 할인율에 따라 상품의 가격을 할인 받을 수 있다.

이 때, 최대한 할인을 많이 받는다면 얼마를 내야 하는가?

단, 할인쿠폰은 한 제품에 한 번씩만 적용 가능하다.

 

A1.

shop_prices = [30000, 2000, 1500000]
user_coupons = [20, 40]


def get_max_discounted_price(prices, coupons):
    # 쿠폰 내림차순 정렬
    coupons.sort(reverse=True)
    # 가격 내림차순 정렬
    prices.sort(reverse=True)

    price_index = 0
    coupon_index = 0
    result = 0

    # 가장 높은 가격에 가장 높은 할인 쿠폰 적용
    while price_index < len(prices) and coupon_index < len(coupons):
        result += prices[price_index] * (100 - coupons[coupon_index]) / 100
        price_index += 1
        coupon_index += 1

    # 계산할 물건이 더 많을 때 그냥 더하기
    while price_index < len(prices):
        result += prices[price_index]
        price_index += 1

    return result


print(get_max_discounted_price(shop_prices, user_coupons))  # 926000 이 나와야 합니다.

Q2. 괄호가 바르게 짝지어졌다는 것은 '(' 문자로 열렸으면 반드시 짝지어서 ')' 문자로 닫혀야 한다는 뜻이다.

예를 들어 ()() 또는 (())() 는 올바르다.

)()( 또는 (()( 는 올바르지 않다.

이 때, '(' 또는 ')' 로만 이루어진 문자열 s가 주어졌을 때,

문자열 s가 올바른 괄호이면 True 를 반환하고 아니라면 False 를 반환하시오.

 

A2.

s = "(())()"


def is_correct_parenthesis(string):
    stack = []

    for i in range(len(string)):
        if string[i] == "(":
            stack.append("(")
        elif string[i] == ")":
            if len(stack) == 0:
                return False
            stack.pop()

    return stack is not None


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

Q3. 멜론에서 장르 별로 가장 많이 재생된 노래를 두 개씩 모아 베스트 앨범을 출시하려 한다.

노래는 인덱스 구분하며, 노래를 수록하는 기준은 다음과 같다.

 

1. 속한 노래가 많이 재생된 장르를 먼저 수록한다.

(단, 각 장르에 속한 노래의재생 수 총합은 모두 다르다.)

2. 장르 내에서 많이 재생된 노래를 먼저 수록한다.

3. 장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록한다.

 

노래의 장르를 나타내는 문자열 배열 genres와 노래별 재생 횟수를 나타내는 정수 배열 plays가 주어질 때,

베스트 앨범에 들어갈 노래의 인덱스를 순서대로 반환하시오.

 

 

A3.

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


def get_melon_best_album(genre_array, play_array):
    genre_total_play_dict = {}
    # 장르별 Key에, 재생 수와 인덱스를 배열로 묶어 배열에 저장
    genre_index_play_array_dict = {}

    for i in range(len(genre_array)):
        genre = genre_array[i]
        play = play_array[i]

        if genre not in genre_total_play_dict:
            genre_total_play_dict[genre] = play
            genre_index_play_array_dict[genre] = [[i, play]]
        else:
            genre_total_play_dict[genre] += play
            genre_index_play_array_dict[genre].append([i, play])

    # print(genre_total_play_dict) => {'classic': 1450, 'pop': 3100}
    # print(genre_index_play_array_dict)
    # => {'classic': [[0, 500], [[2, 150]], [[3, 800]]], 'pop': [[1, 600], [[4, 2500]]]}

    # item[1]인 play 순으로 정렬
    # reverse=True 내림차순 정렬
    sorted_genre_play_array = sorted(genre_total_play_dict.items(), key=lambda item: item[1], reverse=True)
    result = []
    for genre, _value in sorted_genre_play_array:
        index_play_array = genre_index_play_array_dict[genre]
        sorted_by_play_and_index_play_index_array = sorted(index_play_array, key=lambda item: item[1], reverse=True)
        for i in range(len(sorted_by_play_and_index_play_index_array)):
            if i > 1:
                break
            result.append(sorted_by_play_and_index_play_index_array[i][0])
    return result


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

...백준 풀다가 이거 푸니까...

쉽게 느껴지는 거 같음...

물론 다 잘 푸는 건 아님ㅋㅋㅋㅋ

 

그래도 2번같은 문제는 풀어봤던 문제라 바로 잘 풀 수 있었다.

1번도 파이썬 문법 잘 알았더라면 더 잘 풀 수 있었을 듯

3번은 생소해서 잘 못따라갔지만...

이해했으니 됐지...

 

잼썼다!!

반응형
Comments