LeetCode 611.有效三角形个数

LeetCode 611.有效三角形个数

题目：给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。

方法一： 排序+暴力枚举

思路：

排序，暴力枚举所有三元组,统计符合条件的三元组个数，注意不要重复枚举。固定两边枚举第三边。

排序时间复杂度为 O(nlogn)；三层遍历找所有三元组的复杂度为 O(n^3 )，整体复杂度O(n^3);数据量较大会TLE。

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int ans=0;
        for(int i=0;i<nums.size();i++){
            for(int j=i+1;j<nums.size();j++){
                for(int z=j+1;z<nums.size();z++){
                    if(nums[i]+nums[j]>nums[z] ) ans++;
                }
            }
        }
        return ans;
    }
};

方法二： 排序+二分

思路：
固定两边，查找第三边范围。枚举固定a[i],a[j]的值，之后二分查找能与a[i],a[j]构成三角形的最大的a[z]的位置，依据：a[i]+a[j]>a[z]。(j,z]之间的数都能满足a[i]+a[j]>a[z].

时间复杂度：排序时间复杂度为O(nlogn)；两层遍历加二分所有符合条件的三元组的复杂度为 O(n^2*logn)。整体复杂度为 O(n^2 * logn)

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int ans=0;
        for (int i = 0; i < nums.size(); i++) {
            for (int j = i+1; j <nums.size(); j++) {
                int l = j, r = nums.size()-1 ;
                while (l < r) {
                    int mid = (l + r + 1)  >> 1;
                    if (nums[i] + nums[j] > nums[mid])  l = mid ;
                    else r = mid-1 ;
                }
                ans+= l-j;
            }
        }
        return ans;

    }
};

方法三： 排序 + 双指针

思路：
固定两边，查找第三边范围，找到能与a[i],a[j]构成三角形的a[z]的范围（j，z），此时的（j，z）也必定能与a[i],a[j+1]构成三角形，直接扩大a[z]范围，找到此时a[z]的范围。

注意(0,0,0)

排序时间复杂度为 O(nlogn)，双指针找所有符合条件的三元组的复杂度为 O(n^2)。整体复杂度为 O(n^2)

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int ans = 0;
        for (int i = 0; i < nums.size(); i++) {
            if(nums[i] == 0) continue;
            int z = i + 1 ;
            for (int j = i + 1; j < nums.size(); j++) {
                while (z + 1 < nums.size() && nums[i] + nums[j] > nums[z+1]) z++;
                ans += z - j;
            }
        }
        return ans;

    }
};

方法四：

思路：

先求出各种长度的边的个数，存在st数组，用前缀和求出边长范围为（1，n）的边的数量sm[n]

三角形有三种形式：1. 两边相等；2. 三边相等；3.三边都不相等

从0枚举第一条边的长度 i ：

1. 数量st[i]>=2，即存在两边相等时，第三边长度为 (0,i) , (i,i+i) ;求出两边组合数 * 第三边可取个数 ，即ans += c[ st[i] ][2] * ( sm[i+i-1] - st[i] )
2. 数量st[i]>=3，即存在三边相等时 ，求出该情况组合数ans += c[ st[i] ][3]
3. 三边都不等，枚举两边，i和j，第三边范围 (j,j+i) ，即ans += ( sm[i+j–1] - sm[j] ) * st[i] * st[j]

复杂度取决于边长的范围和边的数量。

class Solution {
public:
    int sm[2010];
    int st[1010];

    int c[1010][4];

    void func()
    {
        for(int i=0;i<1010;i++){
            for(int j=0;j<=i&&j<=3;j++){
                if(!j) c[i][j]=1;
                else c[i][j]=(c[i-1][j]+c[i-1][j-1]);
            }
        }
    }
    int triangleNumber(vector<int>& nums) {
        func();
        int ans=0;
     
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++) st[nums[i]]++;
        st[0]=0;
        for(int i=1;i<1005;i++) sm[i]=sm[i-1]+st[i];
        for(int i=1005;i<2005;i++) sm[i]=sm[i-1];
       
        for(int i=1;i<=1000;i++){
            if(st[i] == 0) continue;
            if(st[i]>=2){
                ans+=c[st[i]][2]*(sm[i+i-1]-st[i]);
            }
            if(st[i]>=3){
                ans+=c[st[i]][3];
            }
            for(int j=i+1;j<=1000;j++){
                if(st[j] == 0) continue;
                int k=sm[i+j-1]-sm[j];
                
                ans+=st[i]*st[j]*k; 
            }
        }
        return ans;
    }
};