[SWEA] - 4836. 색칠하기

초기코드

from pprint import pprint
import sys
sys.stdin = open('sample_input (1).txt', 'r')

for test_case in range(1, int(input())+1):
    N = int(input())
    arr = [list(map(int, input().split())) for _ in range(N)]
    white = [[0] * 10 for _ in range(10)]

    for i in range(len(arr)) :
        r1 = arr[i][0]
        c1 = arr[i][1]
        r2 = arr[i][2]
        c2 = arr[i][3]
        color = arr[i][4]
        for j in range(r1, r2 + 1) :
            for h in range(c1, c2 + 1):
                white[j][h] += 1
    purple = 0
    for n in range(len(white)):
        for m in range(len(white[n])):
            if white[n][m] == 2:
                purple += 1

    print('#{} {}'.format(test_case, purple))

수정코드

for test_case in range(1, int(input())+1):
    N = int(input())
    white = [[0] * 10 for _ in range(10)]

    for i in range(N) :
        r1, c1, r2, c2, color = map(int, input().split())
        for j in range(r1, r2 + 1) :
            for h in range(c1, c2 + 1):
                white[j][h] += 1
    purple = 0
    for n in range(len(white)):
        for m in range(len(white[n])):
            if white[n][m] == 2:
                purple += 1

    print('#{} {}'.format(test_case, purple))