백준(Python) 13300번 방 배정 풀이

Python으로 구현한 13300번 방 배정 문제 풀이입니다.

 

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

13300번: 방 배정

---

import math

n, k = map(int, input().split())
info = [[0] * 7 for _ in range(3)]

for _ in range(n) :
  s, y = map(int, input().split())
  info[s][y] += 1

room = 0

for i in info :
  for j in i :
    room += math.ceil(j / k)

print(room)

 

학년과 성별을 담을 수 있는 2차원 리스트를 생성해주고 반복문을 통해 입력받은 n만큼 반복하여 성별과 학년을

 

입력받는다. 입력받은 성별에 해당하는 학년 인덱스 값을 1 증가시킨다. (info[s][y] += 1)

 

이중 for문을 이용하여 j / k 를 올림한 값을 room에 누적해나간 후 최종적으로 room을 출력한다.