알고리즘/프로그래머스
[프로그래머스]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
문제 회고
오히려 앞선 공원산책 문제가 더 어렵게 느껴져서 .. 상대적으로 쉽게 접근