[백준] S1 20055 컨베이어 벨트 위의 로봇 (java)

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

20055번: 컨베이어 벨트 위의 로봇

 

풀이

 

0) 컨베이어 벨트를 나타낼 이차원 배열 belt[][2]를 선언한다. belt[i][0]은 내구도, belt[i][1] 로봇의 유무를 나타낸다.

1) 한칸씩 이동

2) belt를 temp로 초기화

3) 로봇이 내리는 위치에 닿으면 빼야함

4) 로봇은 한칸 더 이동

5) 이동한 로봇이 내리는 위치이면 빼야함

6) 새로운 로봇을 올림

7) 내구도 검사

8) 1~7까지 반복하며 내구도가 0 인 칸의 개수가 K보다 커질 경우 종료.

9) 반복 횟수 출력.

 

주의사항

 

1) 없다.

 

package com.baekJoon;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StringReader;
import java.util.StringTokenizer;

public class BJ_S1_20055_컨베이어벨트위의로봇 {
	static BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
	static StringTokenizer tokens;
	static int N,K,belt[][], temp[][];
	public static void main(String[] args) throws NumberFormatException, IOException {
		input = new BufferedReader(new StringReader(src));
		tokens = new StringTokenizer(input.readLine());
		N = Integer.parseInt(tokens.nextToken());
		K = Integer.parseInt(tokens.nextToken());
		
		belt = new int [2*N][2];
		temp = new int [2*N][2];
		tokens = new StringTokenizer(input.readLine());
		for(int i=0; i<2*N; i++) {
			belt[i][0] = Integer.parseInt(tokens.nextToken());
		}
		int answer = 0;
		int count;
		while(true) {
			answer++;
			count = 0;

			// 1. 한칸씩 이동
			for(int i=0; i<2*N; i++) {
				if(i == 0) {
					for(int j=0; j<2; j++) {
						temp[i][j] = belt[2*N-1][j];
					}
				}else {
					for(int j=0; j<2; j++) {
						temp[i][j] = belt[i-1][j];
					}
				}
			}
			// 2. belt를 temp로 초기화
			for(int i=0; i<2*N; i++) {
				for(int j=0; j<2; j++) {
					belt[i][j] = temp[i][j];
				}
			}
			// 3. 로봇이 내리는 위치에 닿으면 빼야함
			if(belt[N-1][1] == 1) {
				belt[N-1][1] = 0;
			}
			// 4. 로봇은 한칸 더 이동
			for(int i=N-2; i>=1; i--) {
				if(belt[i][1] == 1 && belt[i+1][1] != 1 && belt[i+1][0] > 0) {
					belt[i][1] = 0;
					belt[i+1][1] = 1;
					belt[i+1][0]--;
				}
			}
			// 5. 이동한 로봇이 내리는 위치이면 빼야함
			if(belt[N-1][1] == 1) {
				belt[N-1][1] = 0;
			}
			// 6. 새로운 로봇을 올림
			if(belt[0][0] > 0) {
				belt[0][0]--;
				belt[0][1] = 1;
			}
			
			// 7. 내구도 검사
			for(int i=0; i<2*N; i++) {
				if(belt[i][0] <= 0) {
					count++;
				}
			}
			if(count >= K) {
				break;
			}
		}
		System.out.println(answer);
	}

	static String src =
			"5 8\r\n" + 
			"100 99 60 80 30 20 10 89 99 100";
}

 

후기

 

 지문에서 제시한 순서대로 하나씩 하다보면 어느새 정답이 나온다.