[Baekjoon] 13986_Gravity

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

---

< Gravity >

 

문제 풀이 

 

'o' 위치에서 아래로 탐색하며 빈칸인 곳으로 이동한다.

 

my solution (Java)

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

public class _13986_ { // Gravity

	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 n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());

		char[][] arr = new char[n][m];

		Stack<int[]> stack = new Stack<>();
		for (int i = 0; i < n; i++) {
			String str = bf.readLine();
			for (int j = 0; j < m; j++) {
				arr[i][j] = str.charAt(j);
				if (arr[i][j] == 'o') {
					stack.add(new int[] { i, j });
				}
			}
		}

		while (!stack.isEmpty()) {
			int position[] = stack.pop();
			int result = position[0];

			for (int i = position[0] + 1; i < n; i++) {
				if (arr[i][position[1]] == '.') {
					result = i;
				} else {
					break;
				}
			}
			arr[position[0]][position[1]] = '.';
			arr[result][position[1]] = 'o';
		}

		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				bw.write(arr[i][j]);
			}
			bw.write("\n");
		}
		bw.flush();
	}Add commentMore actions
}

변수)
n, m : 배열 크기
arr : 그리드
stack : 사과 위치 
position : 현재 사과 위치
result : 이동 위치

 

행, 열을 입력받는다. 행, 열 크기만큼 그리드 정보를 입력받아 arr에 저장하면서 사과가 있는 위치라면 stack에도 저장한다. stack이 빌 때까지 다음 과정을 반복한다.

 

1) stack pop

2) 현재 위치에서 아래로 탐색하면서 비어있다면 result 값 업데이트, 비어있지 않다면 더 이상 이동할 수 없으므로 종료

3) 현재 위치를 빈칸으로 저장, 이동할 수 있는 위치에 사과 저장

 

최종 배열 arr을 출력한다.  

---

---