◾ 스타 수열 : 프로그래머스 LEVEL 3
문제
다음과 같은 것들을 정의합니다.
- 어떤 수열 x의 부분 수열(Subsequence)이란, x의 몇몇 원소들을 제거하거나 그러지 않고 남은 원소들이 원래 순서를 유지하여 얻을 수 있는 새로운 수열을 말합니다.
- 예를 들어, [1,3]은 [1,2,3,4,5]의 부분수열입니다. 원래 수열에서 2, 4, 5를 제거해서 얻을 수 있기 때문입니다.
- 다음과 같은 조건을 모두 만족하는 수열 x를 스타 수열이라고 정의합니다.
- x의 길이가 2 이상의 짝수입니다. (빈 수열은 허용되지 않습니다.)
- x의 길이를 2n이라 할 때, 다음과 같은 n개의 집합 {x[0], x[1]}, {x[2], x[3]}, ..., {x[2n-2], x[2n-1]} 의 교집합의 원소의 개수가 1 이상입니다.
- x[0] != x[1], x[2] != x[3], ..., x[2n-2] != x[2n-1] 입니다.
- 예를 들어, [1,2,1,3,4,1,1,3]은 스타 수열입니다. {1,2}, {1,3}, {4,1}, {1,3} 의 교집합은 {1} 이고, 각 집합 내의 숫자들이 서로 다르기 때문입니다.
1차원 정수 배열 a가 매개변수로 주어집니다. a의 모든 부분 수열 중에서 가장 길이가 긴 스타 수열의 길이를 return 하도록 solution 함수를 완성해주세요. 이때, a의 모든 부분 수열 중에서 스타 수열이 없다면, 0을 return 해주세요.
입력
- a의 길이는 1 이상 500,000 이하입니다.
- a의 모든 수는 0 이상 (a의 길이) 미만입니다.
출력
- 가장 길이가 긴 스타 수열의 길이
입출력 예
| a | result |
|---|---|
| [0] | 0 |
| [5,2,3,3,5,3] | 4 |
| [0,3,3,0,7,2,0,2,2,0] | 8 |
◾ 풀이
1. 해설
- 공통 원소를 기준으로 만들 수 있는 최장 길이의 스타 수열을 탐색해간다.
- 길이 8, 임의의 원소의 개수가 4개라고 한다면 최대 길이 8인 스타 수열을 만들 수 있다.
- 공통 원소를 사용한 부분 수열의 개수를 확인한다.
- 인덱스를 i라고 할때, 아래와 같을 때 인덱스 +1
- a[i], a[i+1] 모두에 공통 원소가 없는 경우
- a[i], a[i+1]의 값이 동일한 경우
- 스타 수열로 가능한 부분 수열인 경우 인덱스 +2
- 인덱스를 i라고 할때, 아래와 같을 때 인덱스 +1
- 공통 원소를 사용한 부분 수열의 개수이므로 *2하여 값 반환
2. 프로그램
- 원소의 개수 확인
- 원소의 개수가 많은 경우부터 공통 원소 최고 개수 확인
- 공통 원소가 없는 경우, 값이 같은 경우 패스
- 공통 원소 최대값 선택
- 길이를 구하기 위해 answer * 2 반환
# 코드
from collections import Counter
def solution(a):
answer = 0
n = len(a)
# 각 원소의 카운트 확인
elements = Counter(a)
# 원소의 수가 2개인 경우부터 스타 수열이 가능하므로
# 1개인 경우는 0 반환
if n <= 1:
return answer
# 원소의 수가 가장 많은 경우부터 공통 원소 최고 개수 확인
for k, v in sorted(elements.items(), key=lambda x : -x[1]):
# 원소의 개수가 공통 원소의 개수보다 작을 경우 종료
if v <= answer:
break
# 공통 원소 개수
common_cnt = 0
# 인덱스
idx = 0
while idx < n - 1:
# a[idx], a[idx+1]에 공통 원소가 없는 경우
# a[idx] == a[idx+1]인 경우
# idx + 1
if (a[idx] != k and a [idx+1] != k) or (a[idx] == a[idx+1]):
idx += 1
continue
# 가능한 경우 다음 인덱스 확인
common_cnt += 1
idx += 2
# 공통 원소 최대값 선택
answer = max(common_cnt, answer)
# 공통 원소 개수이므로
# 전체 길이를 구하기 위해 *2 이후 반환
return answer * 2
후라이드 치킨