0
点赞
收藏
分享

微信扫一扫

Leetcode 1005. K 次取反后最大化的数组和(终于解决)

Leetcode 1005. K 次取反后最大化的数组和(终于解决)_i++


给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:

  • 选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。

重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

以这种方式修改数组后,返回数组 可能的最大和 。

示例 1:

输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3] 。

示例 2:

输入:nums = [3,-1,0,2], k = 3
输出:6
解释:选择下标 (1, 2, 2) ,nums 变为 [3,1,0,2] 。

示例 3:

输入:nums = [2,-3,-1,5,-4], k = 2
输出:13
解释:选择下标 (1, 4) ,nums 变为 [2,3,-1,5,4] 。

提示:

  • 1 <= nums.length <= 10^4
  • -100 <= nums[i] <= 100
  • 1 <= k <= 10^4

主要思路:
1.先排序
2.判断负数个数
1)如果负数个数大于等于要反转的次数直接从最小值开始一次反转K次
2)如果负数小于要反转的次数则先将所有负数转变为正数,然后排一次序,之后将剩余次数对2取余即可

Code:

class Solution {
public:
int largestSumAfterKNegations(vector<int>& nums, int k) {

sort(nums.begin(),nums.end());
int index=-1;
int sum=0;
bool flag=false;
for(int i=0;i<nums.size();i++)
{

if(nums[i]>0)
{
index=i;
break;
}
}
if((index)>=k)//负数个数
{

for(int i=0;i<nums.size();i++)
{
if(i<k)
{
sum-=nums[i];
}
else
sum+=nums[i];
}
}
else
{

int cnt=0;
for(int i=0;i<nums.size();i++)
{
if(nums[i]>0)
{
break;
}
else
{
nums[i]=-nums[i];
cnt++;
}
}
sort(nums.begin(),nums.end());
k=(k-cnt)%2;
cout<<k<<endl;
for(int i=0;i<nums.size();i++)
{
if(i<k)
{
if(nums[i]==0)
{
k=0;
}
sum+=(-nums[i]);
}
else
sum+=nums[i];
}
}

cout<<sum<<endl;
return sum;

}
};

举报

相关推荐

0 条评论