質問リンク

問題の説明

に答える

に感銘を与える

コード#コード#

def possible(line):
    used = [False] * n
    i = 0
    while i+1 < len(line):
        # 높이 같음
        if line[i] == line[i+1]:
            i += 1
        # 높이 1 감소
        elif line[i] == line[i+1] + 1:
            # 범위를 벗어남
            if i+1+l > len(line):
                return False
            # 높이가 다름
            if len(set(line[i+1: i+1+l])) != 1:
                return False
            for j in range(i+1, i+1+l):
                # 경사로 중복
                if used[j]:
                    return False
                used[j] = True
            i += l
        # 높이 1 증가
        elif line[i] + 1 == line[i+1]:
            # 범위를 벗어남
            if i+1-l < 0:
                return False
            # 높이가 다름
            if len(set(line[i+1-l: i+1])) != 1:
                return False
            for j in range(i+1-l ,i+1):
                # 경사로 중복
                if used[j]:
                    return False
                used[j] = True
            i += 1
        else:
            return False
    return True


# init
import sys
read = sys.stdin.readline
n, l = map(int, read().split())
board = [list(map(int, read().split())) for _ in range(n)]
count = 0

# start
for i in range(n):
    row = [board[i][j] for j in range(n)]
    if possible(row):
        count += 1
    col = [board[j][i] for j in range(n)]
    if possible(col):
        count += 1
print(count)