[프로그래머스] 혼자 놀기의 달인 - 파이썬

문제

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

프로그래머스

 

설계

- 각 상자를 모두 묶음으로 분류한 후 묶음별 개수로 정렬한다.

def solution(cards):
    visit = [0] * len(cards)
    cnt = 0  # 상자의 개수
    box = [[]]

    for i in range(len(cards)):
        index = i
        while sum(visit) < len(cards):      # 모든 상자를 분류 할때까지 확인
            if visit[index] == 0:
                visit[index] = 1
                box[cnt].append(cards[index])
                index = cards[index] - 1

            else:         # index 값이 끝났다면
                box.append([])
                cnt += 1  # 각 상자를 채웠다고 표시
                break

    box.sort(key=len, reverse=True)      # 상자가 많은 개수대로 정렬

    if len(box) >= 2:
        return len(box[0]) * len(box[1])
    else:
        return 0    # 박스가 1그룹만 있으면 0 반환