파이썬으로 풀어보는 백준 14890번: 경사로 (삼성 A형 기출 문제)

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

14890번: 경사로

 

내 풀이:

N, L = map(int,input().split())
arr = [list(map(int,input().split())) for _ in range(N)]
ans = 0
for i in range(N):
    pre = arr[i][0]
    cnt = 1
    for j in range(1,N):
        if arr[i][j] == pre:
            cnt += 1
            pre = arr[i][j]
        elif arr[i][j] == pre+1 and cnt >= 0:
            if cnt >= L:
                cnt = 1
                pre = arr[i][j]
            else:
                break
        elif arr[i][j] == pre-1 and cnt >= 0:
            cnt = - L + 1
            pre = arr[i][j]
        else:
            break
    else:
        if cnt >= 0:
            ans += 1
 
for j in range(N):
    pre = arr[0][j]
    cnt = 1
    for i in range(1,N):
        if arr[i][j] == pre:
            cnt += 1
            pre = arr[i][j]
        elif arr[i][j] == pre+1 and cnt >= 0:
            if cnt >= L:
                cnt = 1
                pre = arr[i][j]
            else:
                break
        elif arr[i][j] == pre-1 and cnt >= 0:
            cnt = - L + 1
            pre = arr[i][j]
        else:
            break
    else:
        if cnt >= 0:
            ans += 1
 
print(ans)

Python3, 60ms

문제에서 준 조건만 만족하면 별다른 히든케이스 없이 해결되는 간단한 문제이다. 사실 두 개의 for문이 거의 똑같기 때문에 함수로 만들어서 구현할까 하다가 그냥 작성했다.