[Baekjoon] 13732_Falling Apples

문제(출처: https://www.acmicpc.net/problem/13732)

---

< Falling Apples >

 

문제 풀이 

 

각 사과를 장애물이 없을 때까지 내린다.

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;
import java.io.OutputStreamWriter;

public class _13732_ { // Falling Apples

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st = new StringTokenizer(bf.readLine());

		int r = Integer.parseInt(st.nextToken());
		int c = Integer.parseInt(st.nextToken());

		char arr[][] = new char[r][c];
		Stack<int[]> stack = new Stack<>();

		for (int i = 0; i < r; i++) {
			String str = bf.readLine();
			for (int j = 0; j < c; j++) {
				if (str.charAt(j) == 'a') {
					stack.add(new int[] { i, j });
					arr[i][j]='.';
				}else {
					arr[i][j] = str.charAt(j);
				}
				
			}
		}

		while (!stack.isEmpty()) {
			int apple[] = stack.pop();
			int position = apple[0];
			for (int i = apple[0] + 1; i < r; i++) {
				if(arr[i][apple[1]]=='.') {
					position=i;
				}else {
					break;
				}
			}
			arr[position][apple[1]] = 'a';
		}

		for (int i = 0; i < r; i++) {
			for (int j = 0; j < c; j++) {
				bw.write(arr[i][j]);
			}
			bw.write("\n");
		}
		bw.flush();
	}
}

변수)
r, c : 그리드 크기
arr : 그리드 정보
stack : 사과 위치

 

그리드 크기를 입력받는다. 그리드 크기만큼 정보를 입력받으면서 사과의 위치를 stack에 저장한다. stack에 있는 사과 위치를 살펴보며 아래로 이동할 수 있는 위치를 찾아 이동한다. 최종 arr 배열을 출력한다.

---

---