인프런 커뮤니티 질문&답변

정답은 제대로 나오긴 했는데 시간 초과라던가 하는 문제가 없을까요?

4중 for문은 생각지도 못했네요..ㅠㅠ

sudoku = [list(map(int, input().split())) for _ in range(9)]


def solution(sudoku):
    length = len(sudoku)
    a, b, c, = list(), list(), list()
    # 행열 검사
    for i in range(length):
        row, col = list(), list()
        for j in range(length):
            row.append(sudoku[i][j])
            col.append(sudoku[j][i])
        if len(set(row)) != 9 or len(set(col)) != 9:
            return "NO"
        
        # 3x3 격자판 검사
        a.extend(row[0:3])
        b.extend(row[3:6])
        c.extend(row[6:])
        if i == 2 or i == 5 or i == 8:
            if len(set(a)) != 9 or len(set(b)) != 9 or len(set(c)) != 9:
                return "NO"
            a, b, c, = list(), list(), list()
    
    return "YES"


print(solution(sudoku))