给你一个整数数组 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 的绝对值。
返回 好三元组的数量 。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/count-good-triplets
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解:只需用i,j,k遍历数组并取绝对值再按题目意思比较,期间使用continue和break即可。
class Solution(object):
def countGoodTriplets(self, arr, a, b, c):
"""
:type arr: List[int]
:type a: int
:type b: int
:type c: int
:rtype: int
"""
sum=0
for i in range(0,len(arr)):
for j in range(i+1,len(arr)):
if abs(arr[i]-arr[j])>a:
continue
for k in range(j+1,len(arr)):
if abs(arr[j]-arr[k])>b:
continue
if abs(arr[i]-arr[k])<=c:
sum+=1
return sum