611. Valid Triangle Number

最新推荐文章于 2024-06-26 15:32:29 发布

原创最新推荐文章于 2024-06-26 15:32:29 发布 · 147 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

数据结构&算法专栏收录该内容

94 篇文章

订阅专栏

本文介绍了一种高效计算数组中能构成三角形的三元组数量的方法。通过一次排序和两次循环配合，利用索引差值计算有效组合，实现了O(n^2)的时间复杂度。

Given an array consists of non-negative integers, your task is to count the number of triplets chosen from the array that can make triangles if we take them as side lengths of a triangle.

Example 1:

Input: [2,2,3,4]

Output: 3

Explanation:

Valid combinations are:

2,3,4 (using the first 2)

2,3,4 (using the second 2)

2,2,3

Note:

The length of the given array won't exceed 1000.
The integers in the given array are in the range of [0, 1000].

不是很难，但代码不是很好写

开始想到了binarysearch，但是感觉O(n^2lgn)的时间复杂度应该不行

O(n^2)的解法写了个大概但是有点复杂

2，5，6，7，9

i=0,j=1,k=2当k=3的时候，结束循环，但是k不需要改变位置，因为j是增加的，所以对于之前的k，总满足

nums[i] + nums[j] > nums[k]

想的是存一下上次的count,第二次使用，但是使用的时候不能直接在上次的基础上加上这次的count

因为当j++的时候，上次的count中k=j+1的情况不成立了，因为j占据了k的位置

比如上次i=0,j=1,k=2 这次不会出现这种情况了，因为j是从2开始

同理，i++的时候，也有问题，使得运算很麻烦

但实际上解决的办法很简单，并不需要存储上次的count，k-j-1这个索引的差值就是这次的count

public class Solution {
    public int triangleNumber(int[] nums) {
        int count = 0;
        Arrays.sort(nums);

        for (int i = 0; i < nums.length - 2; i++) {
            int k = i + 2;
            for (int j = i + 1; j < nums.length - 1 && nums[i] != 0; j++) {
                while (k < nums.length && nums[i] + nums[j] > nums[k])
                    k++;
                count += k - j - 1;
            }
        }
        return count;
    }
}