[Python] 백준13300번 : 방 배정

💡문제

백준 13300번 문제 링크

🍀풀이

코드

import math

n, k = map(int, input().split())
arr = [[0] * 7 for _ in range(2)]   # 성별과 학년을 담는 2차원 배열 생성

for _ in range(n):
    s, g = map(int, input().split())
    arr[s][g] += 1      # 성별에 해당하는 학년 인덱스 값을 +1
    
result = 0

for i in range(2):
    for j in range(7):
        result += math.ceil(arr[i][j] / k)
        
print(result)

성별과 학년을 담는 2차원 배열을 생성한다 성별은 [0], 학년은 [1]로 해서 [0]*7을 했다.
arr[s][g] += 1 이렇게 해서 먼저 그냥 방에 넣어버리고, for문을 통해서 배정 가능한 학생수 만큼 나눠서 올림해서 저장하는 식으로 했다.
비슷한 계열의 문제!
나는 배열을 선언하는게 어려운거 같다...