나만 보는 개발 블로그

리스트 컴프리헨션 문법의 중요성을 느껴버린 문제 문제 자체는 단순하다. 2차원 배열을 초기화 후 각 인덱스끼리 더해주는 간단한 문제 # 초기 풀이 A = input() N = int(A.split()[0]) M = int(A.split()[1]) arr1 = [[0] * M for i in range(N)] arr2 = [[0] * M for i in range(N)] for i in range(N): B = input() for j in range(M): arr1[i][j] = int(B.split()[j]) for i in range(N): B = input() for j in range(M): arr2[i][j] = int(B.split()[j]) result = [[arr1[i][j] + arr2..

최대값 최소값 구하는 방식 그대로 인덱스 까지 변수로 담아두면 되는 간단한 문제 arr = [list(map(int, input().split())) for i in range(9)] max_val = -1 row_index = 0 col_index = 0 for i in range(9): for j in range(9): if max_val < arr[i][j]: max_val = arr[i][j] row_index = i+1#출력 조건때문에 1씩 더해줌 col_index = j+1 print(max_val) print(row_index, col_index)

행 마다 배열들의 길이가 다른게 포인트 조건문만 걸어주면 되는 쉬운 문제 arr = [input() for i in range(5)] # 배열 초기화 for i in range(15): for j in range(5): if i < len(arr[j]): # 조건문을 통해 i가 j행의 길이를 넘으면 패스 print(arr[j][i], end="")

문제를 보고 처음에 수학적인 방법으로 접근할려다보니 막막했다. 정답은 픽셀단위로 접근하는것 arr = [[0]*100 for i in range(100)] # 100이라는 제한조건을 통해 0으로 구성된 100 * 100 배열 초기화 # 입력 받기 for i in range(int(input())): x, y = map(int, input().split())# x축, y축 입력받아 for i in range(x, x + 10):# 색종이 크기인 10만큼 반복 for j in range(y, y + 10): arr[i][j] = 1# 해당하는 범위만 1로 업데이트 # 값이 1인 것의 수 세기 count = sum(sum(i) for i in arr) # 결과 출력 print(count)