12.6 每日一题——K 次取反后最大化的数组和
输入一个整数数组和一个整数k,按以下方式修改该数组
1.选择某个下标 i 并将 nums[i] 替换为 -nums[i]
2.重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后,返回数组可能的最大和 。
(这次还是交博客链接)
输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3] 。
输入:nums = [3,-1,0,2], k = 3
输出:6
输入:nums = [2,-3,-1,5,-4], k = 2
输出:13
输入:nums =
[-2,5,0,2,-2],
k = 3
输出:11
其中:
1 <= nums.length <= 10000
-100 <= nums[i] <= 100
1 <= k <= 10000
题解:
方法一:快速排序+贪心
我们在对数组进行排序后,如果想要尽可能的让数组在取反k次后和最大,当然优先去将数组里面的负数变成正数,这时候就会涉及到k和数组里面负数的个数之间的关系。
- 如果k大于负数个数,那么我们先将所有的负数变成正数后,对于剩下的k,我们判断是否为偶数。如果是偶数的话,很显然根据取反两次为本身的性质,我们不用再去进行操作,直接累加数组和即可。如果是奇数的话,我们只要找到排序过的数组的正数最小与负数最大进行绝对值比较即可,谁的绝对值小,那么根据贪心的思想,我们就将他变为负数即可。
- 如果k小于负数的个数,那么我们能变多少负数为正就变多少负数为正即可。
public class Test {
public static void main(String[] args) {
int[] a = {3,-1,0,2};
int[] b = {2,-3,-1,5,-4};
int k1 = 3;
int k2 = 4;
System.out.println("和为:"+solution(a, k1));
}
public static int solution(int[] res,int k){
int sum=0;
int flag= 0;
Arrays.sort(res);
for(int i=0;i<res.length;i++){
if(res[i]<0 && k>0){
res[i] = -res[i];
k--;
}
if (i-1>=0&&res[i-1] < 0 && res[i]>=0) {
flag=i;
}
sum+=res[i];
}
if (k%2==0){
return sum;
}else{
if(flag!=0) {
return sum-2*Math.min(-res[flag-1],res[flag]);
}else {
return sum-2*res[0];
}
}
}
}
方法二:计数排序+贪心
注意 nums[i] 的取值范围,我们可以看到其范围是较小的,因此我们这时候可以使用计数排序的思想求解。
我们可以通过一篇文章来简单了解一下:计数排序。
- 了解了计数排序后,我们可以创建一个hash数组作为我们的映射数组,将nums里面的数都映射进这个数组里面,因此这个hash数组既起到排序的作用又起到存储的作用,接着我们遍历这个hash数组,按照方法一的贪心思想即可,这里只是用计数排序代替了快速排序而已。
class Test {
public static void main(String[] args) {
int[] a = {3,-1,0,2};
int[] b = {2,-3,-1,5,-4};
int k1 = 3;
System.out.println("和为:"+largestSumAfterKNegations(a, k1));
}
public static int largestSumAfterKNegations(int[] nums, int k) {
int[] hash = new int[201];
for(int i=0;i<nums.length;i++){
hash[nums[i]+100]++;
}
int sum = 0;
int i = 0;
while(k>0){
while(hash[i]==0){ //此位置没有存储元素
i++;
}
if(i==100){ //该元素为0
k--;
continue;
}
else if(i<100){
hash[i]--; //因为取反了,所以此位置的元素没了
hash[200-i]++; //给取反后的位置加1元素
k--;
}
else{
if(k%2==0){
break;
}
hash[i]--;
hash[200-i]++;
i=200-i;
k--;
}
}
for(i=0;i<hash.length;i++){
sum+=hash[i]*(i-100);
}
return sum;
}
}
Leetcode:
class Solution {
public int largestSumAfterKNegations(int[] nums, int k) {
int[] hash = new int[201];
for(int i=0;i<nums.length;i++){
hash[nums[i]+100]++;
}
int sum = 0;
int i = 0;
while(k>0){
while(hash[i]==0){
i++;
}
if(i==100){
k--;
continue;
}
else if(i<100){
hash[i]--;
hash[200-i]++;
k--;
}
else{
if(k%2==0){
break;
}
hash[i]--;
hash[200-i]++;
k--;
}
}
for(i=0;i<hash.length;i++){
sum+=hash[i]*(i-100);
}
return sum;
}
}