[백준] 13300번: 방 배정 (python)

13300번: 방 배정

---

문제 )
정보 초등학교에서는 단체로 2박 3일 수학여행을 가기로 했다. 여러 학년이 같은 장소로 수학여행을 가려고 하는데 1학년부터 6학년까지 학생들이 묵을 방을 배정해야 한다. 남학생은 남학생끼리, 여학생은 여학생끼리 방을 배정해야 한다. 또한 한 방에는 같은 학년의 학생들을 배정해야 한다. 물론 한 방에 한 명만 배정하는 것도 가능하다.

한 방에 배정할 수 있는 최대 인원 수 K가 주어졌을 때, 조건에 맞게 모든 학생을 배정하기 위해 필요한 방의 최소 개수를 구하는 프로그램을 작성하시오.

예를 들어, 수학여행을 가는 학생이 다음과 같고 K = 2일 때 12개의 방이 필요하다. 왜냐하면 3학년 남학생을 배정하기 위해 방 두 개가 필요하고 4학년 여학생에는 방을 배정하지 않아도 되기 때문이다.

 

입력 :

표준 입력으로 다음 정보가 주어진다. 첫 번째 줄에는 수학여행에 참가하는 학생 수를 나타내는 정수 N(1 ≤ N ≤ 1,000)과 한 방에 배정할 수 있는 최대 인원 수 K(1 < K ≤ 1,000)가 공백으로 분리되어 주어진다. 다음 N 개의 각 줄에는 학생의 성별 S와 학년 Y(1 ≤ Y ≤ 6)가 공백으로 분리되어 주어진다. 성별 S는 0, 1중 하나로서 여학생인 경우에 0, 남학생인 경우에 1로 나타낸다. 

 

출력 :

표준 출력으로 학생들을 모두 배정하기 위해 필요한 최소한의 방의 수를 출력한다.

 

 

풀이)

import sys
 
n, k = map(int, sys.stdin.readline().strip().split())
 
dic = {}
for i in range(n):
    s, y = map(int, sys.stdin.readline().strip().split())
    if not y in dic: # 딕셔너리에 해당학년의 자료가 없으면 생성
        dic[y] = [0, 0]
 
    if s == 0: # 여자이면
        dic[y][0] += 1
    else: # 남자이면
        dic[y][1] += 1
# ex) {3: [1, 0], 5: [0, 1], 6: [1, 0]}
 
cnt = 0
for i in dic.values():
    for j in i:
        if j == 0: # 0명일 경우 방배정 x
            pass
        else:
            if j % k == 0: # 방의 최대인원이 맞을 경우
                cnt += (j // k)
            else: # 최대인원에 여유롭게 들어갈 경우
                cnt += (j // k) + 1
print(cnt)

 

---

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

13300번: 방 배정