[BOJ 14890] 경사로

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

#include <iostream>

using namespace std;

int i, j;

int N;

int L;

int arr[105][105];

int arr2[105][105];

int arr3[105][105];

int result;

void get_info()

{

cin >> N >> L;

for (i = 0;i < N;i++)

{

for (j = 0; j < N; j++)

{

cin >> arr[i][j];

arr2[i][j] = 10;

arr3[i][j] = 10;

}

}

}

void print_info()

{

cout << result;

}

int main()

{

int count; // 이전의 0의 갯수

get_info();

result = 2*N;

// 가로

for (i = 0;i < N;i++) 

{

for (j = 1; j < N; j++)

{

arr2[i][j - 1] = arr[i][j] - arr[i][j - 1]; // 차이 값 arr2에 저장

}

}

for (i = 0; i < N;i++)

{

count = 0;

for (j = 0; j < N-1; j++)

{

int temp = result;

if (arr2[i][j] > 1 || arr2[i][j] < -1) // 1초과 -1 미만 경우

{

result--;

}

if (arr2[i][j] == 0) // 0인 경우

{

count++;

}

if (arr2[i][j] == 1) // 1인 경우

{

if (count < (L - 1))

result--;

count = 0;

}

if (arr2[i][j] == -1) // -1인 경우

{

if ( (L == 1) && (arr2[i][j + 1] == 1))

{

result--;

}

for (int k = j + 1; k < (L - 1) + j+1; k++)

{

if (arr2[i][k] != 0)

{

result--;

k = (L - 1) + j + 1;

}

}

if (arr2[i][(L - 1) + j + 1] == 0)

j = (L - 1) + j + 1;

else

j = (L - 1) + j + 1 - 1;

count = 0;

}

if (temp != result)

j = N - 1;

}

} 

count = 0;

// 세로

for (j = 0;j < N;j++) 

{

for (i = 1; i < N; i++)

{

arr3[j][i - 1] = arr[i][j] - arr[i-1][j]; // 차이 값 arr2에 저장

}

}

for (i = 0;i < N;i++)

{

count = 0;

for (j = 0; j < N-1; j++)

{

int temp2 = result;

if (arr3[i][j] > 1 || arr3[i][j] < -1) // 1초과 -1 미만 경우

{

result--;

}

if (arr3[i][j] == 0) // 0인 경우

{

count++;

}

if (arr3[i][j] == 1) // 1인 경우

{

if (count < (L - 1))

result--;

count = 0;

}

if (arr3[i][j] == -1) // -1인 경우

{

if ((L == 1) && (arr3[i][j + 1] == 1))

{

result--;

}

for (int k = j + 1; k < (L - 1) + j + 1; k++)

{

if (arr3[i][k] != 0)

{

result--;

k = (L - 1) + j + 1;

}

}

if(arr3[i][(L-1)+j+1]==0)

j = (L - 1) + j + 1 ;

else

j = (L - 1) + j + 1 - 1;

count = 0;

}

if (temp2 != result)

j = N - 1;

}

}

print_info();

return 0;

}