[파이썬] 프로그래머스 베스트앨범

맞출 줄 모르고 손가는대로 풀었더니 풀렸다. 차근차근 구현하면 생각보다 잘 풀리는 것 같다.

 

 

 

def solution(genres, plays):
    answer = []
 
    genre_check = [ ]
    genre_name_check = [ ]
 
    for i in range(0,len(genres)):
        total = 0
        inputs = [ ]
 
        if genres[i] not in genre_name_check:
 
            for j in range(0,len(genres)):
                if genres[i] == genres[j]:
                    total += plays[j]
 
            genre_name_check.append(genres[i])
            inputs.append(total)
            inputs.append(genres[i])
 
            genre_check.append(inputs)
 
 
    genre_check = sorted(genre_check,reverse=True)
 
    for i in range(0,len(genre_check)):
        check_name = genre_check[i][1]
        check_number = [ ]
 
 
        for j in range(0,len(genres)):
            if genres[j] == check_name:
                temp = [ ]
                temp.append(plays[j])
                temp.append(j)
                check_number.append(temp)       
 
        check_number = sorted(check_number,reverse=True)
 
 
 
 
 
        if len(check_number)>=2:
 
            if check_number[0][0]==check_number[1][0]:
                answer.append(check_number[1][1])
                answer.append(check_number[0][1])
            else:
                answer.append(check_number[0][1])
                answer.append(check_number[1][1])
 
        else:
            for z in range(0,len(check_number)):
                answer.append(check_number[z][1])
 
 
 
    return answer