[백준] BOJ - 19941 햄버거 분배 java 자바 풀이 (실버3)

https://www.acmicpc.net/problem/19941

19941번: 햄버거 분배

 

나의풀이

import java.io.*;
import java.util.*;
public class Main {
    public static void main(String args[]) throws Exception{
        int answer=0;
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st=new StringTokenizer(br.readLine());
        int n=Integer.parseInt(st.nextToken());
        int k=Integer.parseInt(st.nextToken());
        
        char[] arr=br.readLine().toCharArray();
        int[] arr2=new int[n];
        Arrays.fill(arr2,0);
        
        for(int i=0; i<n;i++){
            if(arr[i]=='H'){
                arr2[i]=1; // 햄버거가 있는 위치를 1로 초기화 
            }
        }
               
        for(int i=0;i<n;i++){
            if(arr[i]=='P'){//사람일때
                int temp= (i-k) < 0 ? 0 : (i-k) ;
                int temp2= (i+k) > n-1 ? n-1 : (i+k) ;
                for(int j=temp;j<=temp2;j++){//사람 위치에서 앞뒤 k 까지 안먹은 햄버거 있는지 살피기
                    if(arr2[j]==1){//안먹은 햄버거가 있으면 
                        arr2[j]=0; //먹었다고 바꾸기
                        answer++;//햄버거를 먹은 사람으로 추가 
                        break;
                    }
                }
            }
        }
        bw.write(Integer.toString(answer));
        bw.close();
    }
}