chyam

[프로그래머스 Lv2, python] - 혼자 놀기의 달인 본문

프로그래머스/LV2

[프로그래머스 Lv2, python] - 혼자 놀기의 달인

chyam_eun 2025. 5. 8. 13:00

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

def play(cards, i): # 열려야하는 상자가 이미 열려있을때까지
    li = [] # 연 상자
    while cards[i] >0:
        tmp = cards[i]
        li.append(tmp)
        cards[i] = 0 # 열면 0으로 바꿔주기
        i = tmp -1
    
    return li

def solution(cards):
    tmp = []
    for i in range(len(cards)):
        answer = play(cards,i)
        tmp.append(answer)
   
    tmp.sort(key = lambda x:-len(x)) # 길이 긴 순서대로 나열
    
    return len(tmp[0]) * len(tmp[1]) # 큰 상자 두개