개발자/알고리즘

[백준] 13300번 (feat. 2차원 배열)

june__kim 2022. 1. 15. 15:59

[백준] 13300번 (feat. 2차원 배열)

 

 

https://www.acmicpc.net/problem/13300

 

13300번: 방 배정

표준 입력으로 다음 정보가 주어진다. 첫 번째 줄에는 수학여행에 참가하는 학생 수를 나타내는 정수 N(1 ≤ N ≤ 1,000)과 한 방에 배정할 수 있는 최대 인원 수 K(1 < K ≤ 1,000)가 공백으로 분리되어

www.acmicpc.net

 

import sys

n, p = map(int, input().split())

lst = [[0 for j in range(6)] for i in range(2)]

for i in range(n):
    a, b = map(int, sys.stdin.readline().split())
    lst[a][b - 1] += 1

room = 0

for i in range(2):
    for j in range(6):
        if lst[i][j] == 0:
            continue
        if lst[i][j] % p == 0:
            room += int(lst[i][j] / p)
        else:
            room += round((lst[i][j] / p) + 0.5)

print(room)

 

여기서 내가 볼 점은, 먼저 반올림 함수 round.

 

https://wikidocs.net/21113

음... 걍 웬만하면 round함수는 쓰지 말아야겠다.


그리고 제일 중요한 파이썬에서 이차원배열 만드는 방법!!

이건 그냥 어느정도 머릿속에 박아둔다고 생각하는 게 마음 편할 거 같다.

 

[[0 for j in range(COL)] for i in range(ROW)]