[프로그래머스] 스타 수열 / 파이썬

https://programmers.co.kr/learn/courses/30/lessons/70130#

코딩테스트 연습 - 스타 수열

최근에 일을 시작해서(사실은 귀찮아서.. 일수도..) 알고리즘 문제를 풀지 못했습니다.
오늘은 설연휴라 푹 쉬고 밤에 자기 전에, 이대로 자면 안될 것 같다..는 생각에 한 문제 풀었습니다.
내일은 꼭 두 개 풀어야지..

풀이 방법은 모든 경우의 수를 생각합니다.

당연히 최빈값부터 계산하면 좋습니다!

그리고, 도달할 수 없는 케이스는 배제하면 시간 내에 풀 수 있습니다. :)

from collections import Counter

def solution(a):
    answer = 0

    # 최빈값
    common = Counter(a).most_common()
    n = len(a)

    for (fval, count) in common:
        ret = 0
        idx = 0

        # 도달할 수 없는 케이스. 종료.
        if count * 2 < answer:
            break

        while idx < n:
            if a[idx] != fval:
                if idx + 1 < n and a[idx+1] == fval:
                    ret += 2
                    idx += 2
                else:
                    idx += 1
            else:
                if idx + 1 < n:
                    if a[idx+1] != fval:
                        ret += 2
                        idx += 2
                    else:
                        idx += 1
                else:
                    idx += 1
        answer = max(answer, ret)
    return answer