[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)

Pythonは初めて実現した.これからパイソンも一緒に勉強するつもりです.