(코딜리티) Lesson 6. [Sorting] : Distinct - python

문제 : app.codility.com/programmers/lessons/6-sorting/

6. Sorting lesson - Learn to Code - Codility

문제해설

N개의 배열 A에서 구별되는 요소의 갯수를 return합니다.

 

예를 들어,

A = [2, 1, 1, 2, 3, 1]일 때, 구별되는 요소는 1, 2, 3으로 3을 return합니다.

제한 사항

• N의 범위는 0~100,000사이
• A의 각 요소의 범위는 -1,000,000~1,000,000사이

코드

def solution(A):
    
    A = sorted(A)
    if A:
        temp = A[0]
        count = 1
    else :
        count = 0
    
    for a in A:
        if temp != a:
            count += 1
            temp = a
            
            
    return count

해설

배열을 정렬하여, 이전에 들어오는 값과 다르면 count를 올려줍니다. 그리고 배열은 비어있는 경우도 있으니 if문으로 걸러줍니다.