20922 - 겹치는 건 싫어

문제

https://www.acmicpc.net/problem/20922

느낀점

기본적인 투 포인터 문제, while문으로도 접근할 수 있고 그냥 for 문으로도 접근할 수 있다. 풀이 후 엣지 케이스를 검토하여 풀이를 보완하는 작업을 먼저 갖자.

풀이

#include <iostream>
#include <vector>

using namespace std;

int main()
{
    int N, K;
    cin >> N >> K;
    vector<int> numList(N);
    vector<int> numCount(100001, 0);
    for (int i = 0; i < N; i++)
    {
        cin >> numList[i];
    }
    int cnt = 0;
    int maxLength = 0;
    int left = 0;
    for (int right = 0; right < N; right++)
    {
        numCount[numList[right]]++;
        cnt++;
        if (numCount[numList[right]] > K)
        {
            maxLength = max(maxLength, cnt - 1);
            while (true)
            {
                numCount[numList[left++]]--;
                cnt--;
                if (numCount[numList[right]] <= K)
                {
                    break;
                }
            }
        }
    }
    cout << max(maxLength, cnt);

    return 0;
}