给定一个非空数组,返回此数组中第三大的数。如果不存在,则返回数组中最大的数。要求算法时间复杂度必须是O(n)。
程序员文章站
2022-03-13 12:50:30
...
1.给定一个非空数组,返回此数组中第三大的数。如果不存在,则返回数组中最大的数。要求算法时间复杂度必须是O(n)。
示例 1:
输入: [3, 2, 1]
输出: 1
解释: 第三大的数是 1.
示例 2:
输入: [1, 2]
输出: 2
解释: 第三大的数不存在, 所以返回最大的数 2 .
示例 3:
输入: [2, 2, 3, 1]
输出: 1
解释: 注意,要求返回第三大的数,是指第三大且唯一出现的数。
存在两个值为2的数,它们都排第二。
2.代码展示
int thirdMax(int* nums, int numsSize)
{ int max = nums[0];
int min = nums[0];
for (int i=0; i<numsSize; i++)
{
if (max<nums[i])
{
int tmp = nums[i];
nums[i] = max;
max = tmp;
}
if (min>nums[i])
{
int tmp = nums[i];
nums[i] = min;
min = tmp;
}
}
int mid = min;
for (int i=0; i<numsSize; i++)
{
if ( nums[i]<max && mid<nums[i])
{
int tmp = nums[i];
nums[i] = mid;
mid = tmp;
}
if (nums[i]<mid && min<nums[i] )
{
int tmp = nums[i];
nums[i] = min;
min = tmp;
}
}
if (mid == min)
{
return max;
}
return min;
}
3.解题思路
首先通过排序找到最大值和最小值,然后再通过最小值,找到第二小和第三小的值,如果第二小和第三小的值相同说明,只有两个数,那么返回最大的数,不相同说明,有第三小的值,那么返回第三小的值。
由于题目要求了时间复杂度,所以我们需要注意,如果还可以优化,欢迎各位留言交流。