[JAVA/코딩테스트] 매출액의 종류

설명

현수의 아빠는 제과점을 운영합니다. 현수아빠는 현수에게 N일 동안의 매출기록을 주고 연속된 K일 동안의 매출액의 종류를

각 구간별로 구하라고 했습니다.

만약 N=7이고 7일 간의 매출기록이 아래와 같고, 이때 K=4이면

20 12 20 10 23 17 10

각 연속 4일간의 구간의 매출종류는

첫 번째 구간은 [20, 12, 20, 10]는 매출액의 종류가 20, 12, 10으로 3이다.

두 번째 구간은 [12, 20, 10, 23]는 매출액의 종류가 4이다.

세 번째 구간은 [20, 10, 23, 17]는 매출액의 종류가 4이다.

네 번째 구간은 [10, 23, 17, 10]는 매출액의 종류가 3이다.

N일간의 매출기록과 연속구간의 길이 K가 주어지면 첫 번째 구간부터 각 구간별

매출액의 종류를 출력하는 프로그램을 작성하세요.

입력

첫 줄에 N(5<=N<=100,000)과 K(2<=K<=N)가 주어집니다.

두 번째 줄에 N개의 숫자열이 주어집니다. 각 숫자는 500이하의 음이 아닌 정수입니다.

출력

첫 줄에 각 구간의 매출액 종류를 순서대로 출력합니다.

 

예시입력	출력
7 4 20 12 20 10 23 17 10	3 4 4 3

 

 

1. 내가 푼 답

import java.util.*;
import java.io.*;

public class Main {
	
	public void solution(int day, int k, int[] array) {
	
		HashMap<Integer, Integer> count = new HashMap<>();
		
		int lt = 0;
		int rt = k-1;
		
		for(int i=0; i<array.length; i++) {
			// key에 숫자와 횟수를 넣는다
			if(count.containsKey(array[i])) {
				count.put(array[i], count.get(array[i]).intValue() + 1);
			}else {
				count.put(array[i], 1);
			}
			
			if(i==rt) {
				System.out.print(count.size() + " ");
				count.put(array[lt], count.get(array[lt]).intValue() - 1);
                // 만약 횟수가 없다면 아예 키를 지워버리기
				if(count.get(array[lt]).intValue() == 0) {
					count.remove(array[lt]);
				}
				lt++;
				rt++;
			}
			
		}
		
						
	}

	public static void main(String args[]) {
        Main main = new Main();
		Scanner in= new Scanner(System.in);
        int day = in.nextInt();
        int k = in.nextInt();
		int[] array = new int[day];
        for(int i=0; i<day; i++) {
			array[i] = in.nextInt(); 
        }
		
        main.solution(day, k, array);
    }

}