[HackerRank] Beautiful Triplets

[문제 링크]

[입력]

int d: 맞춰야 되는 숫자.
int arr[n]: 오름차순으로 정렬된 정수형 배열

[출력]

1<=i<j<k<=(n-1) 이고 arr[j]-arr[i] = arr[k]-arr[j] = k 인 (i,j,k) 순서쌍 개수

[코드]

def beautifulTriplets(d, arr):
    # Write your code here
    result = 0
    for idx in range(1 , len(arr)-1):
        upper = arr[idx+1:].count(arr[idx]+d)
        lower = arr[0:idx].count(arr[idx]-d)
        
        result += (upper*lower)
    return result