알고리즘/프로그래머스

[프로그래머스]Lv2. 혼자 놀기의 달인(Python)

cha_eyoon 2024. 2. 29. 19:07

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

문제 접근

 

1. 앞에서부터 차례로 시작

2. cards[0]의 값의 인덱스로 점프

3. 해당 값을 0으로 세팅하기 전에 임의의 변수(tmp)를 선언해서 그 값을 담아 놓은 후에 0으로 세팅

4. 만약 점프한 인덱스에 담긴 카드의 값이 0이라면 멈춘 후 다시 시작

 

초반에 고민했던 부분이 while 문을 탈출하기 직전까지 해당 인덱스의 값을 빈리스트에 할당할지 고민했지만 만약 그렇게 하게 되면 마지막에 answer에 담긴 리스트들을 순회하며 그 리스트의 길이를 다시 저장하고 정렬해야 하므로 비효율적이라고 생각해서 cnt 변수를 사용해 하나의 사이클 원소의 개수를 세어주는 것이 낫다고 판단!

 

코드
def solution(cards):
    answer = []
    
    for i in range(len(cards)):
        if cards[i] == 0:
            continue
        cnt = 1 # 나
        jump = cards[i] - 1 # 위치 이동 
        cards[i] = 0    # 나 방문 체크
        while True:
            if cards[jump] == 0:
                answer.append(cnt)
                break
            tmp = cards[jump]
            cards[jump] = 0
            jump = tmp -1 
            cnt += 1 
            
    answer.sort(reverse = True)
    
    if len(answer) >= 2:
        max_score = answer[0] * answer[1]
    else:
        max_score = 0
    
    return max_score

 

문제 회고

 

오히려 앞선 공원산책 문제가 더 어렵게 느껴져서 .. 상대적으로 쉽게 접근