611. Valid Triangle Number

来源:互联网 发布:淘宝客推广位怎么设置 编辑:程序博客网 时间:2024/06/07 14:37
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:

  1. The length of the given array won't exceed 1000.
  2. 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;    }}