[Programmers] LV 0. 머쓱이보다 키 큰 사람 (Python/파이썬)

▶ 문제 설명

머쓱이는 학교에서 키 순으로 줄을 설 때 몇 번째로 서야 하는지 궁금해졌습니다. 머쓱이네 반 친구들의 키가 담긴 정수 배열 array와 머쓱이의 키 height가 매개변수로 주어질 때, 머쓱이보다 키 큰 사람 수를 return 하도록 solution 함수를 완성해보세요.

---

▶ 제한 사항

• 1 ≤ array의 길이 ≤ 100
• 1 ≤ height ≤ 200
• 1 ≤ array의 원소 ≤ 200

---

▶ 입출력 예

array	height	result
[149, 180, 192, 170]	167	3
[180, 120, 140]	190	0

---

▶입출력 예 설명

입출력 예 #1

• 149, 180, 192, 170 중 머쓱이보다 키가 큰 사람은 180, 192, 170으로 세 명입니다.

입출력 예 #2

• 180, 120, 140 중 190보다 큰 수는 없으므로 0명입니다.

---

▶ Code

def solution(array, height):
    answer = 0
    for i in range(len(array)):
        if height < array[i]:
            answer += 1
    return answer

▶ Point

array의 list 첫 번째부터 마지막 원소까지를 대상으로 조건문을 작성하기 위해 'for i in range(len(array))' 라고 작성하고, height 보다 array의 i 번째 원소가 클때마다  answer에 1을 추가하여 출력한다.

 

다른 풀이

def solution(array, height):
    return sum(1 for a in array if a > height)

간단하게 다음과 같이 코드를 작성할 수도 있다.