[Swift] 백준 13300번 - 방 배정

백준 문제 모음

문제 링크

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

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로 나타낸다. 

출력

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

입출력 예시

알고리즘 분류

• 수학
• 구현

---

문제 풀이

• 1학년부터 6학년까지 6개의 학년과 남녀 2개의 방이 존재하기 때문에 6*2 2차원 배열 arr를 생성한다.
• 참가인원 n과 한 방에 배정할 수 있는 최대인원 k를 입력받는다.
• 성별과 학년을 입력받아 temp에 저장한 후 arr[학년-1][성별]에 1을 더한다
• result변수에 각 학년별로 배정된 방의 개수를 구해 더한다.
  • i 학년의 여자 학생의 수를 k로 나눴을 때 나머지가 0이라면 학생수에 k를 나눈 값을 result에 더하고, 아니라면 학생수에 k를 나눈 값에 1을 더한 후 result 변수에 더한다.
  • 남학생도 같은 방법으로 계산한다.

소스코드

var arr = Array(repeating: [0, 0], count: 6)
let nk = readLine()!.split(separator: " ").map{Int(String($0))!}
let n = nk[0], k = nk[1]
var result = 0

for _ in 0..<n {
    let temp = readLine()!.split(separator: " ").map{Int(String($0))!}
    arr[temp[1]-1][temp[0]] += 1
}

for i in 0..<6 {
    result += (arr[i][0]%k == 0 ? arr[i][0]/k : arr[i][0]/k+1)
    result += (arr[i][1]%k == 0 ? arr[i][1]/k : arr[i][1]/k+1)
}
print(result)

• 12ms