class Solution:
def countGoodTriplets(self, arr: List[int], a: int, b: int, c: int) -> int:
res = 0
for i in range(len(arr)):
for j in range(i+1, len(arr)):
for k in range(j+1, len(arr)):
if abs(arr[i] - arr[j]) <= a and abs(arr[j] - arr[k]) <= b and abs(arr[i] - arr[k]) <= c:
res += 1
return res
剪枝过的
class Solution:
def countGoodTriplets(self, arr: List[int], a: int, b: int, c: int) -> int:
count = 0
for i in range(len(arr)-2):
for j in range(i + 1, len(arr)-1):
if abs(arr[i] - arr[j]) > a:
continue
for k in range(j + 1, len(arr)):
if abs(arr[j] - arr[k]) <= b and abs(arr[i] - arr[k]) <= c:
count += 1
return count
作者:LiUrEs
链接:https://leetcode.cn/problems/count-good-triplets/solutions/358460/python3-mei-na-yao-bao-li-de-bao-li-jian-zhi-by-la/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。