[ BOJ / Python ] 13300번 방 배정

이번 문제는 단순한 구현으로 해결할 수 있는 문제였다

• n과 k를 입력받는다.
• 학생의 정보를 입력받는다.
• m,w 배열을 만들고, 배열의 인덱스를 학년으로 정한다. (0 2 => m[2])
• 전체 학생 수 만큼 반복문을 돌며 m,w 배열을 증가시킨다.
• 반복문을 range(1,7)만큼 돌며 해당 성별,학년에 해당하는 학생 수를 k로 나눠 방의 수를 구하고, m,w 배열에 저장한다.
• sum 내장함수를 이용하여 m,w배열의 값들을 더하고 이를 출력한다.

n, k=map(int, input().split())
w=[0]*1000
m=[0]*1000
for i in range(n):
    s, g=map(int, input().split())
    if s==0:
        m[g]+=1
    elif s==1:
        w[g]+=1
for i in range(1,7):
    if(m[i]%k==0):
        m[i]=m[i]//k
    else:
        m[i]=m[i]//k+1
    
    if(w[i]%k==0):
        w[i]=w[i]//k
    else:
        w[i]=w[i]//k+1

result=sum(m)+sum(w)
print(result)

파이썬은 처음 구현해본다. 앞으로 파이썬도 같이 공부 할 예정이다.