[HackerRank] Picking Numbers

[문제 링크]

[입력]

int a[n]: 정수형 배열 , 크기는 n 이다.

[출력]

int: 기준을 충족하는 subarray 중 최대 길이

[제한사항]

1. 배열의 길이는 2이상 100이하이다.
2. 배열 a 의 각 원소는 모두 0초과 100 미만이다.
3. 항상 답은 2보다 크다.
4. 출력에서 말하는 기준은 "SubArray에 속하는 모든 두 원소의 차는 절댓값이 1보다 작아야 한다."이다.

[코드]

def pickingNumbers(a):
    # Write your code here
    set_a = set(a)
    result = 0
    for i in set_a:
        count = list()
        for j in a:
            if(j==i or j==i+1):
                count.append(j)
        count.sort()
        if(len(count)>=result):
            result = len(count)
    return result