생각하는 아져씨

[코드트리 챌린지] 격자안에서 완전탐색 본문

Study/Algorithm

[코드트리 챌린지] 격자안에서 완전탐색

azeomi 2023. 9. 8. 01:57

문제

https://www.codetree.ai/missions/2/problems/best-place-of-33?&utm_source=clipboard&utm_medium=text 

 

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

국가대표가 만든 코딩 공부의 가이드북 코딩 왕초보부터 꿈의 직장 코테 합격까지, 국가대표가 엄선한 커리큘럼으로 준비해보세요.

www.codetree.ai

 

 

 

 

코드

N = int(input())    # 격자 크기
maps = [list(map(int, input().split())) for _ in range(N)]

def count_coins(x, y):
    count = 0
    for i in range(x, x+3):
        for j in range(y, y+3):
            if maps[i][j] == 1:
                count += 1
    return count

answer = 0
for i in range(N):
    for j in range(N):
        if (i+2) >= N or (j+2) >= N:    # 격자 범위를 벗어난다면,
            continue
        coins = count_coins(i, j)
        
        answer = max(answer, coins)

print(answer)