[COS Pro 2급] 키가 K보다 큰 사람 / 파이썬

문제 설명

---

학생들의 키가 들어있는 목록에서 키가 k보다 큰 사람은 몇 명인지 구하려합니다.

예를 들어 다음과 같은 목록에서

키가 175보다 큰 사람은 2명입니다.

학생들의 키가 들어있는 배열 height와 k 값이 매개변수로 주어졌을 때, k보다 키가 큰 학생의 수를 세서 return 하도록 solution 함수를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다. 주어진 코드에서 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.

매개변수 설명

---

학생들의 키가 들어있는 배열 height와 k 값이 solution 함수의 매개변수로 주어집니다.

• height의 길이는 1 이상 100 이하입니다.
• 학생들의 키는 100 이상 300 이하의 자연수입니다.
• k는 100 이상 300 이하의 자연수입니다.

return 값 설명

---

solution 함수는 k보다 키가 큰 학생의 수를 return 합니다.

 

예시 설명

---

예시

height	k	result
[165, 170, 175, 180, 184]	175	2

설명

예시 #1
175보다 큰 키는 [180, 184]로 두 명입니다.

작성 코드

---

def solution(height, k):
    answer = 0
    n = len(height)
    for h in height:
        if h > k:
            answer += 1
    return answer