Algorithm/inflearn

★매출액의 종류(Hash, sliding Window)

마닐라 2021. 9. 15. 17:42

 

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

public class Main {
    public ArrayList<Integer> solution(int n, int k, int[] arr) {
        ArrayList<Integer> answer = new ArrayList<>();
        HashMap<Integer, Integer> HM = new HashMap<>();
        //3일 세팅 -> 4일 매출액 구하고 왼쪽 제거 오른쪽 추가하여 다음날부터 4일 매출액 구함
        //7 4
        //20 12 20 10 23 17 10
        //k가 4라면 3일까지 세팅(20, 12, 20)
        for(int i=0; i<k-1; i++) {
            HM.put(arr[i], HM.getOrDefault(arr[i], 0)+1);
        }


        int lt = 0;
        //4일부터 시작
        for(int rt=k-1; rt<n; rt++) {
            HM.put(arr[rt], HM.getOrDefault(arr[rt], 0)+1);

            answer.add(HM.size());//4

            HM.put(arr[lt], HM.get(arr[lt])-1);

            if(HM.get(arr[lt])==0) HM.remove(arr[lt]);
            lt++;
        }

        //키가 있으면 1증가 없으면 0 (동일 매출액은 하나로 본다)
        //처음 HashMap에는 20 12 20 10 이 들어간다.
        //맨 처음 4일 구간의 매출액의 종류(사이즈)를 ArrayList에 add
        //맨 처음 구간 매출액 종류 add했으니 첫번째 인덱스 매출액 수 -1로 감소
        //매출액 수가 0이라면 remove까지 시켜줘야 size로 인식 안함★★

        return answer;
    }

    public static void main(String[] args){
        Main T = new Main();
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        int m = kb.nextInt();

        int[] arr = new int[n];
        for(int i =0 ; i<n; i++) {
            arr[i] = kb.nextInt();
        }

        for(int x : T.solution(n, m, arr)) {
            System.out.print(x + " ");
        }


    }
}