414. 第三大的数 AND 611. 有效三角形的个数

357 阅读2分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动

414. 第三大的数

给你一个非空数组,返回此数组中 第三大的数 。如果不存在,则返回数组中最大的数。

示例 1:

输入:[3, 2, 1]
输出:1
解释:第三大的数是 1 。
示例 2:

输入:[1, 2]
输出:2
解释:第三大的数不存在, 所以返回最大的数 2 。
示例 3:

输入:[2, 2, 3, 1]
输出:1
解释:注意,要求返回第三大的数,是指在所有不同数字中排第三大的数。
此例中存在两个值为 2 的数,它们都排第二。在所有不同数字中排第三大的数为 1

解题思路

维护三个变量代表第一,二,三大的数字

  1. 当num比第一大数字还要大的时候,将原来的第一,第二往后移
  2. 当num比第二大数字还要大的时候,将原来的第二往后移
  3. 当num比第三大数字还要大的时候,直接赋值为第三大的数字

代码

class Solution {
    public int thirdMax(int[] nums) {

        Integer first=null,sec=null,third=null;
        for (int num : nums) {
            if (first!=null&&num==first||sec!=null&&num==sec||third!=null&&num==third)
                continue;
            if(first==null||num>first)
            {
                third=sec;
                sec=first;
                first=num;
            }else if (sec==null||num>sec){
                third=sec;
                sec=num;
            }else if (third==null||num>third)
                third=num;
        }
        return third==null?first:third;

    }
}

611. 有效三角形的个数

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

示例 1:

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

注意:

  • 数组长度不超过1000。
  • 数组里整数的范围为 [0, 1000]。

解题思路

先将数组从小到大进行排序,固定一条边i,遍历另一条边j,我们发现当j在不断增大的时候,我们的第三边也可以继续在原来的基础上加大,因此我们只需要维护第三边的指针k

代码

class Solution {
    public int triangleNumber(int[] nums) {

        Arrays.sort(nums);
        int n=nums.length,res=0;
        for(int i=0;i<n;i++)
        {
            int k=i;
            for (int j=i+1;j<n;j++)
            {
                while (k+1<n&&nums[k+1]<nums[i]+nums[j])
                    k++;
                res+= Math.max(0,k-j);
            }
        }
        return res;
    }
}