[프로그래머스] 스타 수열 / C++

문제 설명

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

• 어떤 수열 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

 

 

코드

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

int solution(vector<int> a)
{
    int answer = 0;
    int i, len;
    unordered_map<int, int> um;
    
    // a의 모든 원소의 개수를 구함
    for (i = 0; i < a.size(); i++)
    {
        um[a[i]]++;
    }
    // 각 원소를 교집합의 원소라고 가정함
    for (auto u : um)
    {
        // 이 원소가 교집합의 원소일 때 만들 수 있는 스타 수열의 최대 길이는
        // 원소의 개수 * 2
        // 최대 길이가 answer보다 작다면 고려할 필요가 없음
        if (u.second * 2 <= answer)
            continue;
        // 스타 수열의 길이 초기화
        len = 0;
        for (i = 0; i < a.size() - 1; i++)
        {
            // 같은 집합의 값이 같으면 안 됨
            if (a[i] == a[i + 1])
                continue;
            // 집합의 원소 중 교집합의 원소가 없으면 안 됨
            if (a[i] != u.first && a[i + 1] != u.first)
                continue;
            // 조건을 만족하는 짝을 찾은 경우 길이를 갱신
            len += 2;
            // a[i + 1]을 이미 사용했으므로 i를 1만큼 증가
            i++;
        }
        // 스타 수열의 최대 길이를 갱신
        answer = max(answer, len);
    }
    return answer;
}

 

세 번째 입출력 예의 경우 원소의 개수를 세면 um에 다음과 같이 저장된다.

um

0부터 순서대로 교집합의 원소로 가정하여 스타 수열의 길이를 구한다. a[0]과 a[1]은 값이 같지 않으면서 0이 포함되기 때문에 스타 수열에 추가한다. a[4]와 a[5]는 값은 다르지만 0이 없으므로 스타 수열에 추가될 수 없다. a[7]과 a[8]은 값이 같기 때문에 스타 수열에 추가될 수 없다. 이렇게 스타 수열을 구하면 교집합의 원소가 0일 때의 최대 길이는 8이 된다.

이후, 3, 7, 2는 원소의 개수 * 2가 answer보다 작으므로 스타 수열의 최대 길이를 구하지 않고 넘어간다.

 

링크

programmers.co.kr/learn/courses/30/lessons/70130

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

github.com/inbdni/Programmers/blob/master/level03/%EC%8A%A4%ED%83%80%EC%88%98%EC%97%B4.cpp

inbdni/Programmers