프로그래머스

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

컴공코딩러 2020. 11. 22. 15:43

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

 

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

 

programmers.co.kr

문제 설명

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

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

입출력 예

aresult

[0] 0
[5,2,3,3,5,3] 4
[0,3,3,0,7,2,0,2,2,0] 8

입출력 예 설명

입출력 예 #1

  • a의 부분 수열 중에서 주어진 조건을 모두 만족하는 스타 수열이 없으므로, 0을 return 해야 합니다.

입출력 예 #2

  • [5,2,5,3], [5,3,3,5] 는 a의 부분 수열인 동시에 스타 수열입니다. a의 부분 수열 중 이보다 더 긴 스타 수열은 없으므로, 4를 return 해야 합니다.

입출력 예 #3

  • [0,3,3,0,7,0,2,0] 는 a의 부분 수열인 동시에 스타 수열입니다. a의 부분 수열 중 이보다 더 긴 스타 수열은 없으므로, 8을 return 해야 합니다.

 

map을 사용하여 있는 숫자의 index값을 넣은후에 index값을 비교하면서 

왼쪽값과 오른쪽값중 하나를 선택하여 비교 다르면 cnt++

처음엔 visited 배열만들어서 했는데 시간초과나서 바꿨음

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;
unordered_map<int,vector<int>>m;
int solution(vector<int> a)
{
    int answer = 0;
    int len = a.size();
    int i, j;
    for (i = 0; i < len; i++)m[a[i]].push_back(i);
    for (auto num : m)
    {
        int len2 = num.second.size();
        int left = -1;
        int cnt = 0;
        if (len2 <= answer)continue;
        for (i = 0; i < len2; i++)
        {
            int idx = num.second[i];
            int back = idx - 1;
            int right = idx + 1;
            if (back >= 0 && left < back&&a[idx]!=a[back])
            {
                left = back;
                cnt++;
            }
            else if (right < len && left < right && a[idx] != a[right])
            {
                left = right;
                cnt++;
            }
        }
        answer = max(cnt, answer);
    }
    return answer * 2;
}
cs