学习耗时:40min
在一个给定的数组nums中,总是存在一个最大元素 。
查找数组中的最大元素是否至少是数组中每个其他数字的两倍。
如果是,则返回最大元素的索引,否则返回-1。
示例 1:
输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.
解答:
(1)排序解法
class Solution{
public:
int dominantIndex(vector<int>& nums){
int ans=-1;//若不存在符合条件的元素,返回-1
if(nums.size()==1) return 0;
vector<int>copy(nums.begin(),nums.end());
sort(nums.begin(),nums.end(),std::greater<int>());//把nums从大到小排序
if(nums[0]>=2*nums[1])
{
for(int i=0;i<copy.size();i++)
{
if(copy[i]==nums[0])
return ans=i;
}
}
return ans;
}
}
STL排序语句:
sort(nums.begin(),nums.end(),std::less()); // 从小到大
sort(nums.begin(),nums.end(),std::greater()); //从大到小
is_sorted(nums.begin(),nums.end(),std::less()); // 从小到大
is_sorted(nums.begin(),nums.end(),std::greater()); // 从大到小
(2)贪心算法(O(N)的极速解法)
class Solution {
public:
int dominantIndex(vector<int>& nums) {
if(nums.size==0) return -1;
if(nums.size==1) return 0;
int maxindex=0;
bool flag=true;
for(int i=1;i<nums.size();i++)
{
if(nums[maxindex]>nums[i])
{
if(flag)
{
flag=nums[maxindex]>=2*nums[i];
}
}
else
{
flag=nums[i]>=2*nums[maxindex];
maxindex=i;
}
if(!flag) return -1;
else return maxindex;
}
}
}