给你一个整数数组 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 的绝对值。
返回 好三元组的数量 。
示例 1:
输入:arr = [3,0,1,1,9,7], a = 7, b = 2, c = 3
输出:4
解释:一共有 4 个好三元组:[(3,0,1), (3,0,1), (3,1,1), (0,1,1)] 。
示例 2:
输入:arr = [1,1,2,2,3], a = 0, b = 0, c = 1
输出:0
解释:不存在满足所有条件的三元组。
提示:
3 <= arr.length <= 100
0 <= arr[i] <= 1000
0 <= a, b, c <= 1000
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-good-triplets
思路一:枚举法,循环取ijk的下标进行比对然后统计好三元组的数量
思路清晰,但是代码效率太低
代码如下:
public class TestHsyz { public static void main(String[] args) { int[] arr = new int[] {3,0,1,1,9,7}; int a = 7, b = 2, c = 3; System.out.println(countGoodTriplets(arr,a,b,c)); } public static int countGoodTriplets(int[] arr, int a, int b, int c) { int L = arr.length,count =0; for (int i = 0; i < L; i++) { for (int j = i+1; j < L; j++) { for (int k = j+1; k <L ; k++) { if (Math.abs(arr[i]-arr[j])<=a && Math.abs(arr[j]-arr[k])<=b && Math.abs(arr[i]-arr[k])<=c){ count++; } } } } return count; } }
思路二:由题我们可以拆开绝对值来判断
∣arr[i]−arr[j]∣ ≤ a ;∣arr[i]−arr[k]∣ ≤ c;
-a ≤ arr[i]−arr[j] ≤ a; -c≤arr[i]−arr[j] ≤ c;
arr[j]-a ≤ arr[i] ≤ arr[j]+a;
arr[k]-c ≤ arr[i] ≤ arr[k]+c;
得到。 max(arr[j]-a,arr[k]-c) ≤ arr[i] ≤ min(arr[j]+a,arr[k]+c);
由此我们可以 枚举 ∣arr[j]−arr[k]∣≤b 的情况,循环jk,来统计 满足max(arr[j]-a,arr[k]-c) ≤ arr[i] ≤ min(arr[j]+a,arr[k]+c) 的i的个数。
public class TestHsyz { public static void main(String[] args) { int[] arr = new int[] {3,0,1,1,9,7}; int a = 7, b = 2, c = 3; System.out.println(countGoodTriplets2(arr,a,b,c)); } public static int countGoodTriplets(int[] arr, int a, int b, int c) { int L = arr.length,count =0; for (int i = 0; i < L; i++) { for (int j = i+1; j < L; j++) { for (int k = j+1; k <L ; k++) { if (Math.abs(arr[i]-arr[j])<=a && Math.abs(arr[j]-arr[k])<=b && Math.abs(arr[i]-arr[k])<=c){ count++; } } } } return count; } //思路二 public static int countGoodTriplets2(int[] arr, int a, int b, int c) { int count = 0, n = arr.length; int[] sum = new int[1001]; for (int j = 0; j < n; ++j) { for (int k = j + 1 ; k < n; ++k) { if (Math.abs(arr[j] - arr[k]) <= b) { int lj = arr[j] - a, rj = arr[j] + a; int lk = arr[k] - c, rk = arr[k] + c; int l = Math.max(0, Math.max(lj, lk)), r = Math.min(1000, Math.min(rj, rk)); if (l <= r) { if (l == 0) { count += sum[r]; System.out.println(sum[r]); } else { count += sum[r] - sum[l - 1]; } } } } for (int k = arr[j]; k <= 1000; ++k) { ++sum[k]; } } return count; } }