codibility - PermMissingElem

문제

• 1부터 N+1까지의 element로 구성된 길이 N의 array
• 즉 한 element가 빠져있는 array
• 이때의 element를 찾아 반환

내 코드

def solution(A):
    A_ = sorted(A)
    for i, a_ in enumerate(A_):
        if (i+1) != a_:
            return i+1

• 빈 리스트일 때 처리 안됨
• sorted(A)에서 O(NlogN) 만큼의 시간 복잡도

지피티 코드

def solution(A):
    N = len(A)
    total = (N + 1) * (N + 2) // 2  # 1부터 N+1까지의 합
    return total - sum(A)

• 수학적 논리로 접근
• 이건 꼼수 아니야 ..?