[백준] 10810번 파이썬(Python)

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.

공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀져 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)

도현이는 입력으로 주어진 순서대로 공을 넣는다.

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.

---

문제가 길면 지레 겁부터 먹는 나를 발견할 수 있다.

제발 겁먹지 말고 차분히 생각해보자! 제발..

 

'문제 이해'

알고리즘을 짜기 전에 문제 이해도 되지 않는 상황이라, 예제에 나온대로 하나하나 적어가며 이해하려 노력했다.

바구니를 그려보니 리스트로 접근하면 됨을 깨달았다.

ㄴ ⚠️ 여기서 잠깐!

왜 리스트를 사용하냐 묻는다면 리스트와 유사한 튜플 자료형은 한 번 선언된 값을 변경할 수 없기 때문에, 바구니에 공이 들어있으면 빼서 다시 넣어야하는 위 문제는 튜플보다 리스트가 더 적합하다고 생각했다. 하지만 튜플 자료형이 리스트 자료형보다 더 적은 양의 메모리를 사용하기 때문에 문제에 따라 튜플 자료형도 사용할 수 있다면 튜플을 사용하자.

 

'코드 짜면서 고민했던 부분'

- i번 바구니와 j번 바구니 사이에 공을 어떻게 넣어야 할까

-> 1) j - i == 1이라면 두 바구니는 연속되었기 때문에 그냥 basket[i], basket[j] 모두 k공을 넣어줌

-> 2) j - i == 0이라면 한 바구니만 의미하기 때문에 basket[i]에 k공을 넣어줌

-> 3) j - i > 1이라면 두 바구니 사이의 모든 바구니에 공을 넣어야하기 때문에 basket[i]부터 basket[i + (j - i값)]까지 k공을 넣어줌

 

'첫 제출, 그리고 컴파일 에러'

- 반복문의 범위를 설정할 때는 대괄호 안에 ','를 쓴다. 글쓴이는 ':'를 썼다.. 바보

- 바구니의 번호는 1번부터 시작한다. 그치만 리스트의 인덱스는 0부터 시작함을 잊지말자. 바보22

 

'정답 코드'

# N: 바구니 개수, M: 앞으로 넣을 공의 횟수
N, M = map(int, input().split())

# 모든 원소가 0인 N 크기의 리스트 생성
basket = list([0] * N)

for _ in range(M) :
    i, j, k = map(int, input().split())
    
    if j - i == 1 :
        basket[i - 1] = k
        basket[j - 1] = k
    elif j - i > 1 :
        num = j - i
        for x in range(i, i+num+1) :
            basket[x - 1] = k
    elif j - i == 0 :
        basket[i - 1] = k
        
print(*basket)

 

'공부한 것들'

1) 리스트 초기화하기 

- 파이썬 리스트 생성 방법 : 'List = []', 'list = list()'

- 0으로 이루어진 리스트 생성하기 : 'List = [0 for i in range(n)]', 'List = [0] * n'

~> 글쓴이는 List = list([0] * n) 형식으로 생성했는데 이것도 된다.. 뭐지?

 

2) 반복문 범위

- range()의 작성 형식 : range(초기치, 최종치, 증가치)

*최종치는 원하는 값에 +1을 해줘야 함

*증가치가 생략되었다면 기본적으로 반복하며 증가하는 수치는 +1

 

3) 리스트 괄호 없이 요소들만 출력하기

- 리스트 요소 괄호 없이 한번에 출력하는 법 : print(*리스트)

- for문을 통해 출력하기

~> for i in 리스트 : print(i, end=", ") ⚠️마지막 요소 뒤에도 ','출력됨

~> for i in 리스트 : print(', '.join(리스트))

- 리스트를 string 형식으로 바꿔 출력하기 : print(str(리스트))