[TIL/10.14] 백준 - 2차원 배열 골라풀기

💛 Python3 코드

 

백준 2738

➤ N*M크기의 두 행렬이 주어졌을 때, 두 행렬을 더하기 

A, B = [], []

N, M = map(int, input().split())

for row in range(N):
    row = list(map(int, input().split()))
    A.append(row)

for row in range(N):
    row = list(map(int, input().split()))
    B.append(row)
    
for row in range(N):
    for col in range(M):
        print(A[row][col] + B[row][col], end=' ')
    print()

• 한 행씩 입력을 받으므로, 행의 크기만큼 입력을 반복하여 받는다.
• 입력 받은 행을 행렬에 삽입한다.

백준 2566

➤ 행렬에서 최댓값 찾기

max_num=0
col=0
row=0
for i in range(9):
    line = list(map(int,input().split()))
    if max(line)>max_num:
        max_num=max(line)
        col=i
        row=line.index(max_num)
print(max_num)
print(col+1,row+1)

 

백준 2563

➤ 2차원 배열을 이용해 색종이에서 겹치는 부분의 넓이 구하기

arr = [[0 for _ in range(101)] for _ in range(101)] # 2차원 배열 선언
N = int(input())
result = 0

for _ in range(N):
	x, y = list(map(int, input().split()))
    
    for row in range(x, x+10):
    	for col in range(y, y+10):
        	arr[row][col] = 1
            
for i in arr:
	result += sum(i)
 
 print(result)

• 2차원 배열을 모두 0을 갖도록 초기화하여 생성한다.
• 가로, 세로의 크기는 10이므로 반복문에서 범위를 +10으로 지정한다.
• 색종이를 붙인 위치에 1을 넣어준다. 
• 2차원 배열에서 요소의 모든 합을 구한다. (색종이 붙여진 위치만 1이므로 겹치는 부분의 넓이는 이렇게 구할 수 있다!