[Hackerrank] Week3 - Permuting Two Arrays

문제

코드

리스트 A와 리스트 B의 순서를 뒤섞어 각 리스트의 같은 인덱스에 있는 모든 원소의 합이 K보다 크거나 같다면 Yes를, 아니면 No를 반환해주면 되는 문제.

풀어서 생각해보면 A를 오름차순으로 정렬하고, B는 내림차순으로 정렬하여 각 원소의 합을 구해보면 조건을 만족하는지 쉽게 구해볼 수 있다.

def twoArrays(k, A, B):
    A.sort()
    B.sort(reverse=True)
    for i in range(len(A)):
        if A[i] + B[i] < k:
            return 'NO'
            break
        else:
            continue
    return 'YES'