[백준]15961 회전 초밥.java

전영서·2021년 10월 5일
0

Algorithm

목록 보기
70/89

1.문제

2.코드

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

public class Main{
    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int d = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());
        int c = Integer.parseInt(st.nextToken());

        int[] arr = new int[N];
        int[] in = new int[d+1];
        for(int i=0; i<N; i++) {
            arr[i] = Integer.parseInt(br.readLine());
        }

        int result = 0;
        int[] window = new int[k];
        int cnt = 0;

        for(int i=0; i<k-1; i++){
            window[i] = arr[i];

            if(in[arr[i]]==0){
                cnt++;
                in[arr[i]]++;
            }else{
                in[arr[i]]++;
            }
        }

        for(int i=k-1; i<N+k; i++) {
            in[window[i%k]]--;
            if(in[window[i%k]]==0){
                cnt--;
            }
            window[i%k] = arr[i%N];

            if(in[window[i%k]]==0) {
                in[window[i % k]]++;
                cnt++;
            }else{
                in[window[i%k]]++;
            }

            if(in[c]==0){
                result = Math.max(result, cnt+1);
            }else{
                result = Math.max(result, cnt);
            }

            if(result==k+1) break;
        }


        System.out.println(result);
    }
}

3.Reveiw

시간초과가 나서 어떻게 하면 값에 접근을 덜할까 하다가 슬라이딩 윈도우 개념을 적용하였다. 앞에껀 추가되고 뒤에껀 추가된다는 느낌으로 배열을 조절하였다.

profile
꾸준히 성실하게

0개의 댓글