이번에 풀어본 문제는
백준 20922번 겹치는 건 싫어 입니다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int N = Integer.parseInt(st.nextToken());
int K = Integer.parseInt(st.nextToken());
int [] numbers = new int[N];
st = new StringTokenizer(br.readLine());
for (int i = 0; i < N; i++) {
numbers[i] = Integer.parseInt(st.nextToken());
}
int [] count = new int[100001];
int start = 0, end = 0;
int answer = Integer.MIN_VALUE;
while (end < N) {
// 늘릴 수 있으면 end 증가
while (end < N && count[numbers[end]] + 1 <= K) {
count[numbers[end++]]++;
}
// 더이상 늘릴 수 없음
int len = end - start;
answer = Math.max(answer, len);
//start 이동
count[numbers[start++]]--;
}
System.out.print(answer);
}
}
K개 이하의 중복값을 가진 연속 수열 중 가장 긴 수열의 길이를 출력하는 문제입니다.
K값을 초과하는지 여부를 판단하기 위해 count 배열을 하나 두고, start/end를 이동시키는 투포인터 방식을 사용했습니다.
현재 start값에서 end값을 가능한 위치까지 옮긴 후, 길이를 answer과 비교합니다. 해당 반복은 끝났으므로, start값을 증가시켜, 앞줄의 길이를 줄이고 다시 end값을 비교해가며 길이를 늘려주면 최종적으로 가장 큰 값을 구해낼 수 있습니다.