1534.Count Good Triplets

牧天️白衣

于 2024-06-24 18:00:14 发布

阅读量140

点赞数 2

分类专栏：力扣文章标签：算法

本文链接：https://blog.csdn.net/2401_84524453/article/details/139933489

版权

力扣专栏收录该内容

14 篇文章

订阅专栏

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）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。