<섹션4-HashMap, TreeSet> 3. 매출액의 종류

3. 매출액의 종류

<설명>

현수의 아빠는 제과점을 운영합니다. 현수아빠는 현수에게 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이하의 음이 아닌 정수입니다.

<출력>

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

===================================================

<코드>

먼저 k-1만큼의 요일의 매출종류를 map에 저장한다. 그 후 하루씩 추가하여 map.size()를 활용하여 종류를 구하고 가장 빠른 날을 삭제한다. 삭제하는 날의 매출값이 map에서 0이라면 map에서도 그 키와 값을 삭제한다.

import java.util.*;

class Main {	
	public ArrayList<Integer> solution(int n,int k,int[] array){
		ArrayList<Integer> answer=new ArrayList<>();
		HashMap<Integer, Integer> map=new HashMap<>();
		int lt=0;
		for(int i=0;i<k-1;i++) {
			map.put(array[i], map.getOrDefault(array[i], 0)+1);
		}
		for(int j=k-1;j<n;j++) {
			map.put(array[j], map.getOrDefault(array[j], 0)+1);
			answer.add(map.size());
			map.put(array[lt], map.get(array[lt])-1);
			if(map.get(array[lt])==0)map.remove(array[lt]); 
			lt++;
		}
		return answer;
	}
		
		
	public static void main(String[] args) {
		Main main = new Main();
		Scanner scan = new Scanner(System.in);
		
		int n=scan.nextInt();
		int k=scan.nextInt();
		int[] array=new int[n];
		for(int i=0;i<n;i++) {
			array[i]=scan.nextInt();
		}
		for(int x:main.solution(n,k,array)) {
		System.out.print(x+" ");
		}
		}
}

<중요>

1) HashMap의 크기

• map.size()

2) HashMap에서 삭제

• map.remove(x)