[코드트리] 행복한 수열의 개수 - 파이썬

https://www.codetree.ai/missions/2/problems/number-of-happy-sequence?&utm_source=clipboard&utm_medium=text

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

풀이

• 행 검사와 열 검사를 따로 진행
• count가 m보다 크거나 같다면 break 후 result에 +1

n, m = map(int, input().split())
graph = []

for i in range(n):
    a = list(map(int, input().split()))
    graph.append(a)

result = 0

# 행 검사
for i in range(n):
    count = 1
    for j in range(1, n):
        if graph[i][j] == graph[i][j - 1]:
            count += 1
            if count >= m:
                result += 1
                break
        else:
            count = 1
    else:
        if count >= m:
            result += 1

# 열 검사
for j in range(n):
    count = 1
    for i in range(1, n):
        if graph[i][j] == graph[i - 1][j]:
            count += 1
            if count >= m:
                result += 1
                break
        else:
            count = 1
    else:
        if count >= m:
            result += 1

print(result)