关注

611.有效三角形的个数

解题思路

最直观的方法,使用暴力循环。三重循环,满足条件则计数加1.
这里注意要先排序,因为要使得满足组成三角形,则满足任意两条边之和大于第三边。
则可以通过排序使得三条边:a,b,c,中的a + c >b, b+c > a,则只需要判断a+b > c即可。

class Solution {
public int triangleNumber(int[] nums) {
// 排序
Arrays.sort(nums);
int n = nums.length;
int ans = 0;
for (int i = 0; i < n; i++){
for (int j = i+1; j < n; j++){
for (int k = j+1; k < n; k++){
if (nums[i] + nums[j] > nums[k]){
ans++;
}
}
}
}
return ans;
}
}
上述的时间复杂度是O(n^3),
如何进一步优化呢?对于a,b,c,我们可以枚举前两个数a,b的下标i,j,然后对于第三个数c的下标k进行二分查找,找出满足条件的k的范围区间,然后累加即可。

class Solution {
    public int triangleNumber(int[] nums) {
        int n = nums.length;
        Arrays.sort(nums);
        int ans = 0;
        for (int i = 0; i < n; ++i){
            for (int j = i+1; j < n; j++){
                // 二分查找
                int left = j+1, right = n - 1, k = j;
                while (left <= right){
                    int mid = (left + right) / 2;
                    if (nums[mid] < nums[i] + nums[j]){
                        k = mid;
                        left = mid + 1;
                    }
                    else{
                        right = mid - 1;
                    }
                }
                ans += k - j; // 区间[j+1, k]
            }
        }
        return ans;
    }
}

上述时间复杂度是O(n^2logn)

如何进一步优化呢?

假设只固定i,则随着j的增加,不等式右侧的nums[i] + nums[j]也是递增的,满足条件的k也是递增的,
这样的话,可以将j和k看成两个同向移动的指针,

使用一重循环枚举i,当i固定,使用双指针同时维护j和k,初始值为i。
将j右移一个位置,并不断尝试向右移动k,使得k是最大的满足nums[k] < nums[i] + nums[j]的下标。
将max(k-j, 0)加入答案。


class Solution {
    public int triangleNumber(int[] nums) {
        int n = nums.length;
        Arrays.sort(nums);
        int ans = 0;
        for (int i = 0; i < n; i++){
            // 枚举j,k
            int k = i;
            for (int j = i + 1; j < n; j++){
                while(k+1 < n && nums[k+1] < nums[i] + nums[j]){
                    ++k;
                }
                ans += Math.max(k - j, 0);
            }
        }
        return ans;
    }
}

转载自CSDN-专业IT技术社区

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

原文链接:https://blog.csdn.net/Misnearch/article/details/152126712

评论

赞0

评论列表

微信小程序
QQ小程序

关于作者

点赞数:0
关注数:0
粉丝:0
文章:0
关注标签:0
加入于:--