백준 2차원 배열 python 문제풀이 (2738, 2566, 10798, 2563)

2738

문제
N*M크기의 두 행렬 A와 B가 주어졌을 때, 두 행렬을 더하는 프로그램을 작성하시오.

입력
첫째 줄에 행렬의 크기 N 과 M이 주어진다. 둘째 줄부터 N개의 줄에 행렬 A의 원소 M개가 차례대로 주어진다. 이어서 N개의 줄에 행렬 B의 원소 M개가 차례대로 주어진다. N과 M은 100보다 작거나 같고, 행렬의 원소는 절댓값이 100보다 작거나 같은 정수이다.

cols, rows = map(int, input().split())

A,B = [],[]


for i in range(cols):
    a = list(map(int,input().split()))
    A.append(a)

for j in range(cols):
    b = list(map(int,input().split()))
    B.append(b)

for i in range(cols):
    for j in range(rows):
        result = A[i][j] + B[i][j]
        print(result, end=' ')
    print()

 

2566

<그림 1>과 같이 9×9 격자판에 쓰여진 81개의 자연수 또는 0이 주어질 때, 이들 중 최댓값을 찾고 그 최댓값이 몇 행 몇 열에 위치한 수인지 구하는 프로그램을 작성하시오.

예를 들어, 다음과 같이 81개의 수가 주어지면

3	23	85	34	17	74	25	52	65
10	7	39	42	88	52	14	72	63
87	42	18	78	53	45	18	84	53
34	28	64	85	12	16	75	36	55
21	77	45	35	28	75	90	76	1
25	87	65	15	28	11	37	28	74
65	27	75	41	7	89	78	64	39
47	47	70	45	23	65	3	41	44
87	13	82	38	31	12	29	29	80

이들 중 최댓값은 90이고, 이 값은 5행 7열에 위치한다.

A = []
MAX = -1  # 0은 포함될 수 있으므로 초기값을 -1로 설정
row = 0
col = 0

# 입력 받기
for i in range(9):
    A.append(list(map(int, input().split())))  # 한 줄에 9개의 값을 입력받아 리스트로 추가

# 최댓값 찾기
for i in range(9):
    for j in range(9):
        if MAX < A[i][j]:
            MAX = A[i][j]
            row = i
            col = j

print(MAX)
print(row + 1, col + 1)  # 문제에서 1-indexing을 요구하므로 row+1, col+1 출력

 

10798

아직 글을 모르는 영석이가 벽에 걸린 칠판에 자석이 붙어있는 글자들을 붙이는 장난감을 가지고 놀고 있다.

이 장난감에 있는 글자들은 영어 대문자 ‘A’부터 ‘Z’, 영어 소문자 ‘a’부터 ‘z’, 숫자 ‘0’부터 ‘9’이다. 영석이는 칠판에 글자들을 수평으로 일렬로 붙여서 단어를 만든다. 다시 그 아래쪽에 글자들을 붙여서 또 다른 단어를 만든다. 이런 식으로 다섯 개의 단어를 만든다. 아래 그림 1은 영석이가 칠판에 붙여 만든 단어들의 예이다.

A A B C D D
a f z z
0 9 1 2 1
a 8 E W g 6
P 5 h 3 k x
<그림 1>

한 줄의 단어는 글자들을 빈칸 없이 연속으로 나열해서 최대 15개의 글자들로 이루어진다. 또한 만들어진 다섯 개의 단어들의 글자 개수는 서로 다를 수 있다.

심심해진 영석이는 칠판에 만들어진 다섯 개의 단어를 세로로 읽으려 한다. 세로로 읽을 때, 각 단어의 첫 번째 글자들을 위에서 아래로 세로로 읽는다. 다음에 두 번째 글자들을 세로로 읽는다. 이런 식으로 왼쪽에서 오른쪽으로 한 자리씩 이동 하면서 동일한 자리의 글자들을 세로로 읽어 나간다. 위의 그림 1의 다섯 번째 자리를 보면 두 번째 줄의 다섯 번째 자리의 글자는 없다. 이런 경우처럼 세로로 읽을 때 해당 자리의 글자가 없으면, 읽지 않고 그 다음 글자를 계속 읽는다. 그림 1의 다섯 번째 자리를 세로로 읽으면 D1gk로 읽는다.

그림 1에서 영석이가 세로로 읽은 순서대로 글자들을 공백 없이 출력하면 다음과 같다:

Aa0aPAf985Bz1EhCz2W3D1gkD6x

칠판에 붙여진 단어들이 주어질 때, 영석이가 세로로 읽은 순서대로 글자들을 출력하는 프로그램을 작성하시오.

A = []
max_length = 0

# 입력 받기: 각 문자열을 리스트의 요소로 추가
for i in range(5):
    A.append(input())  # 각 문자열을 리스트에 추가
    max_length = max(max_length, len(A[i]))  # 가장 긴 문자열의 길이를 계산

# 세로로 읽기: 문자열의 각 인덱스를 순서대로 읽기
for j in range(max_length):  # 가장 긴 문자열의 길이만큼 반복
    for i in range(5):  # 5개의 문자열을 세로로 읽기
        if j < len(A[i]):  # 인덱스가 문자열 길이보다 작은 경우만 출력
            print(A[i][j], end='')  # 세로로 이어서 출력

 

2563

가로, 세로의 크기가 각각 100인 정사각형 모양의 흰색 도화지가 있다. 이 도화지 위에 가로, 세로의 크기가 각각 10인 정사각형 모양의 검은색 색종이를 색종이의 변과 도화지의 변이 평행하도록 붙인다. 이러한 방식으로 색종이를 한 장 또는 여러 장 붙인 후 색종이가 붙은 검은 영역의 넓이를 구하는 프로그램을 작성하시오.

예를 들어 흰색 도화지 위에 세 장의 검은색 색종이를 그림과 같은 모양으로 붙였다면 검은색 영역의 넓이는 260이 된다.

입력

첫째 줄에 색종이의 수가 주어진다. 이어 둘째 줄부터 한 줄에 하나씩 색종이를 붙인 위치가 주어진다. 색종이를 붙인 위치는 두 개의 자연수로 주어지는데 첫 번째 자연수는 색종이의 왼쪽 변과 도화지의 왼쪽 변 사이의 거리이고, 두 번째 자연수는 색종이의 아래쪽 변과 도화지의 아래쪽 변 사이의 거리이다. 색종이의 수는 100 이하이며, 색종이가 도화지 밖으로 나가는 경우는 없다

 

 

다른 사람들 풀이 보니.. 굳이 2차원함수로 안 만들고 1차원으로 픽셀 찍듯이 0 -> 1로 바꿔주는 형식으로 만들면 훨씬 쉽게 풀린다...

# 도화지의 크기 (100x100)
paper = [[0] * 100 for _ in range(100)]
n = int(input())  # 색종이의 개수
covered_area = 0  # 색종이가 붙은 면적

# 색종이 정보 입력 받기
for _ in range(n):
    x, y = map(int, input().split())  # 색종이의 왼쪽 아래 좌표

    # 색종이는 10x10 크기
    for i in range(x, x + 10):
        for j in range(y, y + 10):
            # 만약 이미 색종이가 붙어있지 않다면, 면적 증가
            if paper[i][j] == 0:
                paper[i][j] = 1
                covered_area += 1

# 최종적으로 색종이가 덮은 면적 출력
print(covered_area)