[HackerRank] Angry Professor

[문제 링크]

[입력]

int k: 최소 학생 수
int a[n]: n명의 학생들의 도착 시간

[출력]

n명중 제시간에 도착하거나 일찍 도착한 사람의 수가 k명 보다 적으면 'YES'
그렇지 않으면 'NO'을 반환한다.

[코드]

def angryProfessor(k, a):
    # Write your code here
    count = 0
    for i in range(len(a)):
        if a[i]<=0:
            count+=1
    if(count < k):
        return 'YES'
    else:
        return 'NO'