LC-6241. 数组中不等三元组的数目【周赛320】

6241. 数组中不等三元组的数目

难度简单5

给你一个下标从 0 开始的正整数数组 nums 。请你找出并统计满足下述条件的三元组 (i, j, k) 的数目：

• 0 <= i < j < k < nums.length
• nums[i]、nums[j] 和 nums[k] 两两不同 。
  • 换句话说：nums[i] != nums[j]、nums[i] != nums[k] 且 nums[j] != nums[k] 。

返回满足上述条件三元组的数目。

示例 1：

输入：nums = [4,4,2,4,3]
输出：3
解释：下面列出的三元组均满足题目条件：
- (0, 2, 4) 因为 4 != 2 != 3
- (1, 2, 4) 因为 4 != 2 != 3
- (2, 3, 4) 因为 2 != 4 != 3
共计 3 个三元组，返回 3 。
注意 (2, 0, 4) 不是有效的三元组，因为 2 > 0 。

示例 2：

输入：nums = [1,1,1,1,1]
输出：0
解释：不存在满足条件的三元组，所以返回 0 。

提示：

• 3 <= nums.length <= 100
• 1 <= nums[i] <= 1000

贡献法

对于 x*x，设：

• 小于 x 的数有 a 个；
• 等于 x 的数有 b 个；
• 大于 x 的数有 c 个。

那么 x 对答案的贡献是 abc。

累加所有贡献，得到答案。

代码实现时，通过排序可以快速求出 abc。

class Solution {
    public int unequalTriplets(int[] nums) {
        int ans = 0;
        Arrays.sort(nums);
        for(int l = 0,r = l; r < nums.length;){
            while(r < nums.length && nums[r] == nums[l]) r++;
            ans += (r-l) * l * (nums.length-r); 
            l = r;
        }
        return ans;
    }
}