문제
정보 초등학교에서는 단체로 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로 나타낸다.
출력
표준 출력으로 학생들을 모두 배정하기 위해 필요한 최소한의 방의 수를 출력한다.
예제 입력16 2
1 1 0 1 1 1 0 2 1 2 0 2 0 3 1 3 1 4 1 3 1 3 0 6 1 5 0 5 1 5 1 6 |
예제 출력12
|
풀이
배열을 위의 예제 설명과 같이 학년별, 그리고 남녀별로 나누어 입력을 받는다.
그 후 각 학년별 남녀별 수를 k로 나눈 올림만큼 더해 출력하면 된다.
import sys, math
input = sys.stdin.readline
n,k = map(int,input().split())
arr = [[[],[]] for _ in range(7)]
for _ in range(n):
s,y = map(int,input().split())
arr[y][s].append(1)
ans = 0
for w,m in arr:
ans += math.ceil(len(w)/k) + math.ceil(len(m)/k)
print(ans)
'알고리즘 연습 > 구현, 문자열' 카테고리의 다른 글
[🥈4 / 백준 2776 / 파이썬] 암기왕 (0) | 2022.04.17 |
---|---|
[🥈4 / 백준 2491 / 파이썬] 수열 (0) | 2022.04.16 |
[🥈4 / 백준 1213 / 파이썬] 팰린드롬 만들기 (0) | 2022.04.12 |
[🥉1 / 백준 1652 / 파이썬] 누울 자리를 찾아라 (0) | 2022.04.07 |
[🥉2 / 백준 1568 / 파이썬] 새 (0) | 2022.04.06 |