지우너

[코드트리] 겹치는 원소가 적은 부분 수열 C++ 본문

Problem Solving

[코드트리] 겹치는 원소가 적은 부분 수열 C++

지옹 2024. 8. 26. 19:33

문제

https://www.codetree.ai/missions/8/problems/subsequences-with-few-overlapping-elements?&utm_source=clipboard&utm_medium=text

 

코드

#include <iostream>
#include <unordered_map>
using namespace std;

const int MAX_N = 100001;

int n, k;
int num[MAX_N];
unordered_map<int,int> um;

int main() {
    cin >> n >> k;
    for(int i=0; i<n; ++i){
        cin >> num[i];
    }

    int r=0, answer=1;
    for(int l=0; l<n; ++l){
        // um에 num[r]이 없거나 k개보다 적을 경우
        while(r<n && (um.find(num[r])==um.end()|| um[num[r]]<k)){
            um[num[r]]++;
            r++;
        }
        answer=max(answer, r-l);
        um[num[l]]--;
    }
    cout << answer << '\n';
    return 0;
}