2022.8.12今天你刷题了吗?
题目:
给你一个整数数组 nums ,其中总是存在 唯一的 一个最大整数 。
请你找出数组中的最大元素并检查它是否 至少是数组中每个其他数字的两倍 。如果是,则返回 最大元素的下标 ,否则返回 -1 。
分析:
给定一个数组,判断最大数是不是第二大数的两倍,如果满足,则返回最大数下标,不满足则返回-1,对于数组只有一个元素的情况返回下标1。
思路:先找出最大数和它的下标,这个通过一个for循环可以实现,然后对数组排序,得到第二大数,再最后进行判断满不满足两倍条件。
解析:
1.暴力法
class Solution {
public:
int dominantIndex(vector<int>& nums) {
int temp = -1;
int index = -1;
if (nums.size() == 1)
{
return 0;
}
for (int i = 0; i < nums.size(); i++)
{
//temp = max(temp, nums[i]);
if (nums[i] > temp)
{
temp = nums[i];
index = i;
}
}
sort(nums.begin(), nums.end());
if (temp >= 2 * nums[nums.size() - 2])
{
return index;
}
return -1;
}
};
2.优化
我们在找最大数时,可以同时把第二大数找出来,这样子就不用再进行排序了。而找第二大数的方法,就是判断当前数大于最大或者第二大,满足条件,更新对应的值即可。
class Solution {
public:
int dominantIndex(vector<int>& nums) {
int max1 = -1;
int max2 = -1;
int index = -1;
if (nums.size() == 1)
{
return 0;
}
for (int i = 0; i < nums.size(); i++)
{
if (nums[i] > max1)
{
max2 = max1;
max1 = nums[i];
index = i;
}
else if (nums[i] > max2)
{
max2 = nums[i];
}
}
if (max1 >= 2 * max2)
{
return index;
}
return -1;
}
};