AtCoder ABC 142 B – Roller Coaster Python解説

スポンサーリンク

Roller Coaster

高橋君の仲間たちは N 人で遊園地に遊びにいきました。
遊園地の一番人気のジェットコースターに乗るためには、身長が K cm以上必要です。
高橋君の i 番目の仲間の身長は hi cm です。
高橋君の仲間たちのうち、一番人気のジェットコースターに乗ることができる人の数を求めてください。

AtCoder Beginner Contest 「Roller Coaster」

高橋君の仲間たちN人の身長を受け取り、ジェットコースターに乗ることができるKcm以上かをチェックして、
条件を満たしていればカウントアップしていきます。

n, k = map(int, input().split())
h = list(map(int, input().split()))
cnt = 0

for i in h:
    if k <= i:
        cnt += 1
        
print(cnt)

if文でジェットコースターに乗るために必要な身長Kと高橋君の仲間たちの身長を比べて、基準以上ならcnt+1にします。最後にcntを出力して完了です。

上記のコードを短く書くなら次のようにも書けます。

n, k = map(int, input().split())
print(sum(1 if k <= i else 0 for i in map(int, input().split())))

AtCoderB問題

Posted by cheese