代码随想录day34 | 1005.K次取反后最大化的数组和 134.加油站 135. 分发糖果

1005.K次取反后最大化的数组和

题目|文章
image

思路

如何让翻转后的数组和最大,就是尽可能的反转绝对值大的负数。当反转次数多余时,不断反转绝对值最小的数。

  1. 首先将整个数组按照绝对值的大小从大到小排列
  2. 当有反转次数并且为负数时,反转负数
  3. 如果反转次数有剩余,那么反转绝对值最小的数,即排序后的数组尾数。

实现

点击查看代码
class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int result = 0;
        sort(nums.begin(), nums.end(), cmp);
        for(int i = 0 ; i < nums.size()-1; i++) {
           if(nums[i] < 0 && k > 0) {
               result += -nums[i];
               k--;
           }
           else {
               result += nums[i];
           }
           cout << result << endl;
        }    
        if( k >=1 && k % 2 == 1) {
            result += -nums[nums.size() - 1];
        }
        else {
            result += nums[nums.size() - 1];
        }
        
        return result;
        
    }
    bool static cmp(int a, int b) {
        return abs(a) > abs(b);
    }
};

复杂度分析

  • 时间复杂度:O(nlogn)
  • 空间复杂度:O(n)

134.加油站

题目|文章
image

思路

这个题的思路比较难想,记住同类题目比较好。
如果跑完一圈后,计算每个加油站的剩余油量。如果总剩余油量小于零,那么不可能跑完。
从0开始,如果[i,j]之间加油站油量总和为负数,那么就不能从这i-j开始出发,至少要从j+1开始出发。更新当前总油量为0,从j+1开始计算。

实现

点击查看代码
class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int curSum = 0;
        int totalSum = 0;
        int start = 0;
        for(int i = 0; i < gas.size(); i++) {
            curSum += gas[i] - cost[i];
            totalSum += gas[i] - cost[i];
            if(curSum < 0) {
                curSum = 0;
                start = i + 1;
            }
        }
        if(totalSum < 0) return -1;
        return start;
    }
};

复杂度分析

  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

135. 分发糖果

题目|文章
image

思路

这道题开始做的时候,我想只遍历一次就计算出结果,但总是顾此失彼,没办法同时考虑两边。想建立数组统计连续增加和减小的次数,但是没办法考虑左右两边。
这道题的思路是对整个数组进行两次贪心计算。

  1. 考虑右边孩子比左边孩子优秀的情况‘
  2. 考虑左边孩子比右边孩子优秀的情况

实现

点击查看代码
class Solution {
public:
    int candy(vector<int>& ratings) {
        vector<int> candyCount(ratings.size(), 1);
        for(int i = 0; i < ratings.size() - 1; i++) {
            if(ratings[i+1] > ratings[i]) candyCount[i+1] = candyCount[i] + 1;
            //cout << candyCount[i] <<endl;
        }
        for(int i = ratings.size() - 2; i >= 0; i--) {
            if(ratings[i] > ratings[i+1]) candyCount[i] = max(candyCount[i], candyCount[i+1] + 1);
            //cout<<candyCount[i]<<endl;
        }
        int result = 0;
        for(auto a : candyCount)  result += a;
        return result;
    }
};

复杂度分析

  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

原文地址:http://www.cnblogs.com/suodi/p/16852424.html

发表评论

您的电子邮箱地址不会被公开。