[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;
}