leetcode 611 有效三角形个数

Problem describe:
给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 1:

输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:

数组长度不超过1000。
数组里整数的范围为 [0, 1000]。
解法:
如果我们将给定的数组由大到小排序时,问题就转变为了找一个三元组(a,b,c)其中只要满足a+b>c的条件就可以了,因为由于数组时有序的,所以a<=b<=c,因此a+c>b和b+c>a的条件是肯定可以满足的。
然后,对于每个二元组(i,j),我们只要在(j+1,nums.length-1)中找到满足条件nums[i]+nums[j]>nums[k]的个数,由于数组是有序的,所以我们只需找到第一个不满足条件nums[i]+nums[j]>nums[k]的下标k,这样(j+1,k-1)内的所有数都是满足组成三角型的解,于是,对于二元组(i,j)来讲,一共有(k-1-(j+1)+1)=k-j-1个解。
至于如何有效地找到k,我们可以采用二分搜索来找k下标。
此外,还要注意到当找到二元组(i,j)所对应的下标k,进一步求(i,j+1)的下标k时,我们只需要搜索(k,nums.length-1)这个范围。

class Solution {
     public int binarySearch(int[] nums,int l,int r,int x)
        {
            while(r>=l&&r<nums.length)
            {
                int mid=(l+r)/2;
                if(nums[mid]>=x)
                    r=mid-1;
                else
                    l=mid+1;
            }
            return l;
        }
    public int triangleNumber(int[] nums) {
       int count=0;
        Arrays.sort(nums);
        for(int i=0;i<nums.length-2;++i)
            for(int j=i+1;j<nums.length-1;++j)
            {
                int k=binarySearch(nums,j+1,nums.length-1,nums[i]+nums[j]);
                    count+=k-j-1;
            }
        return count;
        
    }
}

猜你喜欢

转载自blog.csdn.net/shine10076/article/details/83591658