python-leetcode-2475. 数组中不等三元组的数目

2475. 数组中不等三元组的数目 - 力扣(LeetCode)

我们要找的是满足以下条件的三元组 (i, j, k):

  • 0 \le i < j < k < \text{nums.length}

  • nums[i] \ne nums[j] \ne nums[k] \ne nums[i](三者两两不同)

解法思路:

我们可以按以下思路来解决:

  1. 计数每个数字的出现次数

  2. 枚举所有不同的三元组组合数,通过数学组合方式。

  3. 避免暴力三重循环(时间复杂度太高)。

步骤:

  1. 统计每个数字的出现次数。

  2. 将不同的数放入一个列表中,对这个列表进行组合。

  3. 每个组合 (a, b, c) 对应的有效三元组数是:
    count[a] \times count[b] \times count[c]


示例代码(Python):

from collections import Counter
from itertools import combinations

def count_unique_triplets(nums):
    counter = Counter(nums)
    unique_vals = list(counter.keys())
    res = 0

    for a, b, c in combinations(unique_vals, 3):
        res += counter[a] * counter[b] * counter[c]

    return res

示例:

nums = [1, 2, 3, 4]
# 输出:4,因为组合是 (1,2,3), (1,2,4), (1,3,4), (2,3,4)

你也可以直接用暴力三重循环方法,如果数据量不大。但如果数据很多,上面的方案更高效。

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值