인프런 커뮤니티 질문&답변

초기 window에 값을 넣을 때 k-1까지 넣는 것이 아니라 k-2로 해야하는거 아닌가요?

그래야 다음 반복문에서 처음 rt값이 중복되지 않을 것 같아서요.

신기한건, 저는 k-2로 풀었는데 강사님이 k-1까지로 해서 푸셔도 채점사이트에서는 둘다 정답으로 나오네요. 뭐가 맞는걸까요?

 

 

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;

public class Main{
	public ArrayList<Integer> solution(int n, int k, int[] input){
		ArrayList<Integer> answer = new ArrayList<>();
		HashMap<Integer,Integer> map = new HashMap<>();
		
		int lt =0;
		for(int i=0;i<=k-2;i++) { // k-1?
			map.put(input[i], map.getOrDefault(input[i], 0)+1);
		}
		
		for(int rt=k-1;rt<n;rt++) { //rt=k-1부터 시작
			map.put(input[rt], map.getOrDefault(input[rt],0)+1);
			answer.add(map.size());
			map.put(input[lt],map.get(input[lt])-1);
			if(map.get(input[lt])==0) {
				map.remove(input[lt]);
			}
			lt++;
		}
		
		return answer;
	}
	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int k = sc.nextInt();
		int[] input = new int[n];
		for(int i=0;i<n;i++) {
			input[i]=sc.nextInt();
		}
		for(int x : T.solution(n, k, input)) {
			System.out.print(x+" ");
		}
		
		sc.close();
	}

}