14719번 : 빗물 [Java]

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

14719번: 빗물

해결 방안

제일 아래 줄에서부터 블록이 아니라면 양옆으로 한 칸씩 전진하며 블록으로 막힐 수 있는지 찾는다. 양 쪽에 블록이 다 있다면 그 칸을 빗물로 채운다. 다 살핀 후 빗물 양을 센다.

 

아쉬운 점

풀고 다른 사람들 코드를 보니 같은 느낌인데 훨씬 간단하게 푼 분이 많았다... 너무 복잡하게 생각했나...

 

전체 코드

import java.util.Scanner;

public class Main {
	static int H, W;
	static int[][] map;
	static int ans = 0;

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		H = sc.nextInt(); // 세로
		W = sc.nextInt(); // 가로
		map = new int[H][W];
		for (int j = 0; j < W; j++) {
			int block = sc.nextInt();
			for (int i = H - 1; i >= H - block; i--) {
				map[i][j] = 1;
			}
		}

		for (int h = H - 1; h >= 0; h--) {
			rain(h);
		}

		calc();
		System.out.println(ans);
	}

	private static void calc() {
		for (int i = 0; i < H; i++) {
			for (int j = 0; j < W; j++) {
				if (map[i][j] == 2)
					ans++;
			}
		}
	}

	private static void rain(int h) {
		for (int j = 0; j < W; j++) {
			if (map[h][j] == 0) {
				boolean left = false;
				int temp = j;
				while (true) {
					temp -= 1;
					if (temp < 0)
						break;
					if (map[h][temp] == 1) {
						left = true;
					}
				}
				boolean right = false;
				temp = j;
				while (true) {
					temp += 1;
					if (temp >= W)
						break;
					if (map[h][temp] == 1) {
						right = true;
					}
				}
				if (left && right)
					map[h][j] = 2;
			}
		}
	}
}