swe 4014 활주로 건설

1. 설계 완벽하게 하기 ( 예제 3개 돌리기) 2. 초기화 조건 확인하기 3. 가지치기 4. 경우의 수 나열 하기.(세세하게)

/*
1. 설계 완벽하게 하기
2. 초기화 변수 확인하기
3. 가지치기.
4. 경우의 수 나열하기

1533
1635
한시간 컷 입니다.

가로에 대해서 다리 놓기
세로에 대해서

가로에 대해서 갈 수 있는지 확인하기
세로에 대해서

*/
 
#include <iostream>
#include <algorithm>
using namespace std;
#define SIZE 22
int map[SIZE][SIZE];
int visit[SIZE][SIZE];
int visit2[SIZE][SIZE];
int N, L;
int ans;
int curH;
int isFine;
int same_H;
 
void problemIn() {
    cin >> N >> L;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            cin >> map[i][j];
        }
    }
}
 
void init() {
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            visit[i][j] = 0;
            visit2[i][j] = 0;
        }
    }
    ans = 0;
}
 
void solve() {
 
    // 가로에 대해서 다리 놓기
    for (int i = 0; i < N; i++) {
        curH = map[i][0];
        for (int j = 1; j < N; j++) {
            if (map[i][j] == curH) {
                continue;
            }
            else if (map[i][j] != curH) {
                if (map[i][j] - curH == 1) { // 오르막이면, 올라가기 위해서 L 길이를 둠.
                    same_H = 0;
                    for (int k = 1; k <= L; k++) {
                        if (curH == map[i][j - k]) {
                            same_H++;
                        }
                        else {
                            break;
                        }
                    }
                    if (same_H >= L) { // 다리를 둡니다.
                        for (int k = j - L; k < j; k++) {
                            if (visit[i][k] == 1) break;
                            visit[i][k] = 1;
                        }
 
                        curH = map[i][j];
                    }
                }
                else if (curH - map[i][j] == 1) { // 내리막이면,
                    same_H = 0;
                    for (int k = 0; k < L; k++) {
                        if (map[i][j + k] == (curH - 1)) {
                            same_H++;
                        }
                        else {
                            break;
                        }
                    }
                    if (same_H >= L) {
                        for (int k = 0; k < L; k++) {
                            if (visit[i][j + k] == 1) break;
                            visit[i][j + k] = 1;
                        }
                        curH = map[i][j];
                    }
                }
                else {
                    break; // 이 행은 다리를 둘 수 없음.
                }
            } // 높이가 다르면 end
        } // j end
    } // i end
 
 
    // 세로에 대해서 다리 놓기
    for (int j = 0; j < N; j++) {
        curH = map[0][j];
        for (int i = 1; i < N; i++) {
            if (map[i][j] == curH) {
                continue;
            }
            else if (map[i][j] != curH) {
                if (map[i][j] - curH == 1) { // 오르막이면, 올라가기 위해서 L 길이를 둠.
                    same_H = 0;
                    for (int k = 1; k <= L; k++) {
                        if (curH == map[i - k][j]) {
                            same_H++;
                        }
                        else {
                            break;
                        }
                    }
                    if (same_H >= L) { // 다리를 둡니다.
                        for (int k = i - L; k < i; k++) {
                            if (visit2[k][j] == 1) break;
                            visit2[k][j] = 1;
                        }
                        curH = map[i][j];
                    }
                }
                else if (curH - map[i][j] == 1) { // 내리막이면,
                    same_H = 0;
                    for (int k = 0; k < L; k++) {
                        if (map[i + k][j] == (curH - 1)) {
                            same_H++;
                        }
                        else {
                            break;
                        }
                    }
                    if (same_H >= L) {
                        for (int k = 0; k < L; k++) {
                            if (visit2[i + k][j] == 1) break;
                            visit2[i + k][j] = 1;
                        }
                        curH = map[i][j];
                    }
                }
                else {
                    break; // 이 행은 다리를 둘 수 없음.
                }
            } // 높이가 다르면 end
        } // i end
    } // j end
 
 
    // 갈 수 있는 길인지 체크를 합니다. 가로
    for (int i = 0; i < N; i++) {
        curH = map[i][0];
        for (int j = 1; j < N; j++) {
            if (map[i][j] == curH) {
                isFine = 1;
                continue;
            }
            else if (map[i][j] != curH) {
                if (map[i][j] - curH == 1) {
                    if (visit[i][j-1] == 1) {
                        isFine = 1;
                        curH = map[i][j];
                    }
                    else {
                        isFine = 0;
                        break;
                    }                
                }
                else if (curH - map[i][j] == 1) {
                    if (visit[i][j] == 1) {
                        isFine = 1;
                        curH = map[i][j];
                        visit[i][j + L - 1] = 0;
                    }
                    else {
                        isFine = 0;
                        break;
                    }
                }
                else {
                    isFine = 0;
                    break;
                }
            }
        }
        if (isFine) ans++;
    }
 
    // 세로
    for (int j = 0; j < N; j++) {
        curH = map[0][j];
        for (int i = 1; i < N; i++) {
            if (map[i][j] == curH) {
                isFine = 1;
                continue;
            }
            else if (map[i][j] != curH) {
                if (map[i][j] - curH == 1) {
                    if (visit2[i - 1][j] == 1) {
                        isFine = 1;
                        curH = map[i][j];
                    }
                    else {
                        isFine = 0;
                        break;
                    }
                }
                else if (curH - map[i][j] == 1) {
                    if (visit2[i][j] == 1) {
                        isFine = 1;
                        curH = map[i][j];
                        visit2[i + L - 1][j] = 0;
                    }
                    else {
                        isFine = 0;
                        break;
                    }
                }
                else {
                    isFine = 0;
                    break;
                }
            }
        }
        if (isFine) ans++;
    }
}
 
int main() {
    int T;
    cin >> T;
    for (int tc = 1; tc <= T; tc++) {
        problemIn();
        solve();
        cout << "#" << tc << " " << ans << endl;
        init();
    }
    return 0;
}