【代码随想录】刷题Day33

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

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

最简单的思路就是每次取反前排序,将最小的数取反,这样子得到的数组和一定最大,所以思路就是循环k次,每次开始排序,将第一个数取反。最后累加即可

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int ret = 0;
        for(int i=0;i<k;i++)
        {
            sort(nums.begin(),nums.end());
            nums[0]=-nums[0];
        }
        for(auto&e:nums)
            ret+=e;
        return ret;
    }
};

其实与上面思路一致,就是为了找最小的值来取反,那么我们其实可以改造以一下,不需要多次的排序。首先我们对数组进行绝对值排序,绝对值大的排在前面。那么此时我们要遍历数组,将数组是负数的数进行取反,k--;最后如果没有负数了,那么取最小非负数取反。

class Solution {
public:
    static bool abs_cmp(int a,int b){
        return abs(a)>abs(b);
    }

    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int ret = 0;
        sort(nums.begin(),nums.end(),abs_cmp);
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]<0&&k>0)
            {
                nums[i]=-nums[i];
                k--;
            }
        }
        while(k--)
            nums[nums.size()-1]=-nums[nums.size()-1];
        for(auto&e:nums)
            ret+=e;
        return ret;
    }
};

2.加油站

134. 加油站

realSum:指所有的差之和,如果小于零,说明没有一个能绕一圈回来的

curSum:为当前位置开始的差求和,如果差求和是负数,说明该位置一定不是开始的位置,因为差之和为负数代表的就是有不够。那么我们就需要更新ret为下一次的i。

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int ret = 0;
        int realSum = 0;
        int curSum = 0;
        for(int i=0;i<gas.size();i++)
        {
            realSum += gas[i]-cost[i];
            curSum += gas[i]-cost[i];
            if(curSum<0)
            {
                ret = i+1;
                curSum = 0;
            }
        }
        if(realSum<0)
            return -1;
        return ret;
    }
};

3.分发糖果

135. 分发糖果

思路:构造target数组存放每一个小朋友的糖果数值,两边分别遍历。

1.第一次遍历走左边,比较后一个数与前一个数的大小,如果后一个大,后一个的糖果数量为前一个的糖果数量加一

2.第一次遍历走右边,比较前一个数和后一个数的大小,如果前一个大,则前一个糖果的数量为 当前糖果数与后一个糖果数加一的较大值

class Solution {
public:
    int candy(vector<int>& ratings) {
        vector<int> target(ratings.size(),1);
        int ret = 0;
        for(int i=0;i<ratings.size()-1;i++)
        {
            if(ratings[i]<ratings[i+1])
                target[i+1]=target[i]+1;
        }
        for(int i=ratings.size()-1;i>=1;i--)
        {
            if(ratings[i]<ratings[i-1])
                target[i-1]=max(target[i]+1,target[i-1]);
        }
        for(auto e:target)
            ret+=e;
        return ret;
    }
};

猜你喜欢

转载自blog.csdn.net/m0_63488627/article/details/130801841
今日推荐