枚举——统计好三元组

枚举——统计好三元组 一、题目描述

给你一个整数数组 arr ，以及 a、b 、c 三个整数。请你统计其中好三元组的数量。

如果三元组 (arr[i], arr[j], arr[k]) 满足下列全部条件，则认为它是一个 好三元组 。

0 <= i < j < k < arr.length
|arr[i] - arr[j]| <= a
|arr[j] - arr[k]| <= b
|arr[i] - arr[k]| <= c
其中 |x| 表示 x 的绝对值。

二、代码

int countGoodTriplets(int* arr, int arrSize, int a, int b, int c){
    int i, j, k, gap1, gap2, gap3;
    int count = 0;
    for(i = 0; i < arrSize - 2; i++){
        for(j = i + 1; j < arrSize - 1; j++){
            for(k = j + 1; k < arrSize; k++){
                gap1 = (arr[i] > arr[j] ? arr[i] - arr[j] : arr[j] - arr[i]);
                gap2 = (arr[j] > arr[k] ? arr[j] - arr[k] : arr[k] - arr[j]);
                gap3 = (arr[i] > arr[k] ? arr[i] - arr[k] : arr[k] - arr[i]);
                if(gap1 <= a && gap2 <= b && gap3 <= c)
                    count++; 
            }
        }
    }
    return count;
}