백준 30804 과일 탕후루

 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
37
38
39
40
41
42
43
44
45
46

#include <iostream>
#include <unordered_map>

int gFruits[200'000] = {};

int main()
{
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);

    // 탕후루 정보 입력
    int N;
    std::cin >> N;
    for (int i = 0; i < N; ++i)
    {
        std::cin >> gFruits[i];
    }

    int maxLen = 1;
    std::unordered_map<int, int> fruitCount;
    int left = 0;

    // right을 한 칸씩 오른쪽으로 이동
    for (int right = 0; right < N; ++right)
    {
        ++fruitCount[gFruits[right]];

        // 과일 종류가 세 개가 되면, 두 종류가 될 때까지 left를 오른쪽으로 이동
        while (fruitCount.size() > 2)
        {
            --fruitCount[gFruits[left]];
            if (fruitCount[gFruits[left]] == 0)
            {
                fruitCount.erase(gFruits[left]);
            }
            ++left;
        }

        // 최대 길이 업데이트
        maxLen = std::max(maxLen, right - left + 1);
    }

    std::cout << maxLen;

    return 0;
}