백준 14890번 경사로 ( C++ )

문제

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

 

해설

경우의 수만 잘 나누면 어렵지 않았던 문제

가로 세로 로직을 따로 구현하는 것이 아닌 map1[i][j] = map2[j][i]를 이용하여 하나의 로직으로 가로 세로 모두 확인

코드

#include <iostream>
#include <algorithm>
 
using namespace std;
int n, l, mp[104][104], mp2[104][104], ret;
 
void go(int num, int a[104][104])
{
  int now_num = a[num][0];
  int flag = 0;
  int cnt = 1;
 
  for (int i = 1; i < n; i++)
  {
    if (now_num == a[num][i])
      cnt++;
    else if (abs(now_num - a[num][i]) >= 2)
    {
      return;
    }
    else if (now_num < a[num][i])
    {
      if (cnt >= l && flag == 0)
      {
        cnt = 1;
        now_num = a[num][i];
      }
      else
      {
        return;
      }
    }
    else if (now_num > a[num][i])
    {
      if (flag == 0)
      {
        flag = 1;
        now_num = a[num][i];
        cnt = 1;
      }
      else
      {
        return;
      }
    }
    if (flag == 1 && cnt == l)
    {
      flag = 0;
      cnt = 0;
    }
  }
  if (flag == 0)
    ret++;
}
 
int main()
{
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  cout.tie(NULL);
 
  cin >> n >> l;
  for (int i = 0; i < n; i++)
  {
    for (int j = 0; j < n; j++)
    {
      cin >> mp[i][j];
      mp2[j][i] = mp[i][j];
    }
  }
 
  for (int i = 0; i < n; i++)
  {
    go(i, mp);
    go(i, mp2);
  }
 
  cout << ret;
}