[코드트리] 격자로서의 2차원 배열: 동전이 있는 위치

 

 

썸네일 메이커

 

---

 

문제

 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

 

 

 

 

실행 결과

 

 

 

 

제출한 코드(Python)

n, m = map(int, input().split())  # n = 격자의 크기, m = 동전 개수
arr_2d = [[0 for _ in range(n)] for _ in range(n)]  # n*n의 격자 2차원 배열 생성

for _ in range(m):
    r, c = tuple(map(int, input().split()))  # 동전 위치 입력받기
    arr_2d[r - 1][c - 1] = 1  # 동전이 있는 위치를 1로 나타내기

for row in arr_2d:
    for elem in row:
        print(elem, end=" ")
    print()

 

 

 

코드설명

 

문제가 길어서 약간 당황했지만 생각보다 풀이는 간단하다.

동전이 있는 위치를 0으로 초기화한 후,

행렬을 위치 r, c로 받아 해당 인덱스에 위치해 있는 부분만 1로 바꾸면 된다.

 

 

 

---