[백준 JAVA] 14890번: 경사로

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

 

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

public class Main {
	public static int T, N, X, ans;
	public static int[][] map;

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		// 1. 입력받기
		st = new StringTokenizer(br.readLine());
		N = Integer.parseInt(st.nextToken());
		X = Integer.parseInt(st.nextToken());
		map = new int[N][N];
		ans = 0;

		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine());
			for (int j = 0; j < N; j++) {
				map[i][j] = Integer.parseInt(st.nextToken());
			}
		}

		// 2. 지형 확인
		// 2-1. 가로 지형 확인
		int cnt = 0;
		
		for (int i = 0; i < N; i++) {
			cnt = 1;
			for (int j = 1; j < N; j++) {
				if (map[i][j] - map[i][j - 1] == 1) { // 이후 값이 1 큰경우
					if (cnt < X) {
						cnt = -1;
						break;
					}
					cnt = 1;
				} else if (map[i][j] - map[i][j - 1] == -1) { // 이후 값이 1 작은 경우
					if (cnt < 0) {
						break;
					}
					cnt = 1 - X;
				} else if (map[i][j] == map[i][j - 1]) { // 이후 값이 같은 경우
					cnt += 1;
				} else {
					cnt = -1;
					break;
				}
			}
			if (cnt >= 0) {
				ans++;
			}
		}
		
		// 2-2. 세로 지형 확인
		for (int j = 0; j < N; j++) {
			cnt = 1;
			for (int i = 1; i < N; i++) {
				if (map[i][j] - map[i - 1][j] == 1) { // 이후 값이 1 큰경우
					if (cnt < X) {
						cnt = -1;
						break;
					}
					cnt = 1;
				} else if (map[i][j] - map[i - 1][j] == -1) { // 이후 값이 1 작은 경우
					if (cnt < 0) {
						break;
					}
					cnt = 1 - X;
				} else if (map[i][j] == map[i - 1][j]) { // 이후 값이 같은 경우
					cnt += 1;
				} else {
					cnt = -1;
					break;
				}
			}
			if (cnt >= 0) {
				ans++;
			}
		}
		
		// 3. 출력하기
		System.out.println(ans);
	}
}