题目链接:https://leetcode.cn/problems/count-good-triplets/description/
题目描述
给你一个整数数组 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
Code
解法一:暴力遍历
因为这个题目的限制条件比较多,直接暴力法虽然时间复杂度不低,但是实际运行效率也还行
class Solution {
public int countGoodTriplets(int[] arr, int a, int b, int c) {
int resNum = 0;
for (int i = 0; i < arr.length - 2; i++) {
for (int j = i + 1; j < arr.length - 1; j++) {
if (Math.abs(arr[i] - arr[j]) <= a) {
for (int k = j + 1; k < arr.length; k++) {
if (Math.abs(arr[j] - arr[k]) <= b && Math.abs(arr[i] - arr[k]) <= c) {
resNum++;
}
}
}
}
}
return resNum;
}
}