개발자/알고리즘
[백준] 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.
음... 걍 웬만하면 round함수는 쓰지 말아야겠다.
그리고 제일 중요한 파이썬에서 이차원배열 만드는 방법!!
이건 그냥 어느정도 머릿속에 박아둔다고 생각하는 게 마음 편할 거 같다.
[[0 for j in range(COL)] for i in range(ROW)]