[프로그래머스] 스타 수열 [월간 코드 챌린지 시즌1] [python]

[프로그래머스] 스타 수열 [월간 코드 챌린지 시즌1] [python] Level3 파이썬

 

문제 설명

다음과 같은 것들을 정의합니다.

• 어떤 수열 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의 길이) 미만입니다.

 

 

설계 및 구현

 

접근 방법

문제의 조건은 총 4개이다.

1. x의 길이가 2 이상의 짝수입니다. (빈 수열은 허용되지 않습니다.)

2. x의 길이를 2n이라 할 때, 다음과 같은 n개의 집합 {x[0], x[1]}, {x[2], x[3]}, ..., {x[2n-2], x[2n-1]} 의 교집합의 원소의 개수가 1 이상입니다.

3. x[0] != x[1], x[2] != x[3], ..., x[2n-2] != x[2n-1] 입니다.

4. 예를 들어, [1,2,1,3,4,1,1,3]은 스타 수열입니다. {1,2}, {1,3}, {4,1}, {1,3} 의 교집합은 {1} 이고, 각 집합 내의 숫자들이 서로 다르기 때문입니다.

 

1. 가장 많이 출현한 숫자=> max_num을 통해 조건 2를 맞출 수 있다.

     => 가장 많이 출현한 숫자가 교집합의 원소가 된다면 가장 긴 스타수열을 구성 할 수 있다.

2. 그리고 3,4의 조건은 a에서 max_num과 인접한 원소들을 검사하여 맞출 수 있다.

 

my_solution

# 1. 리스트 a를 count한 리스트 cnt, 가장 많이 출현한 숫자 max_num, 가장 많이 출현한 횟수 cnt 
def count_array(a):
    max_num = 0
    max_cnt = 0
    cnt = [0]*len(a)
    for i in range(len(a)):
        cnt[a[i]] += 1
        if cnt[a[i]] > max_cnt:
            max_cnt = cnt[a[i]]
            max_num = a[i]
    return cnt,max_cnt,max_num

# 2. 최빈 횟수 cnt와 최빈 숫자 num을 이용해 조건3,4에 맞는 스타수열의 길이를 찾는 함수
def long_star(a, max_cnt, max_num):
    i = -1
    answer = 0
    while i < len(a)-1:
        i += 1
        try:
            if (a[i] != max_num and a[i+1] != max_num):
                continue
            if (a[i] == max_num and a[i+1] == max_num):
                continue
            answer += 2
            i   += 1
        except:
            pass
    return answer
    

def solution(a):
    answer = 0
    cnt,max_cnt,max_num = count_array(a)
    return long_star(a,max_cnt,max_num)

 

 

흠... 상당히 어려워서 다른분의 코드를 참고하였다.

 

[ 프로그래머스 [ 월간코드챌린지 ] 스타 수열 ] (C++)