前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1775. 通过最少操作次数使数组的和相等(贪心+双指针)

LeetCode 1775. 通过最少操作次数使数组的和相等(贪心+双指针)

作者头像
Michael阿明
发布2021-09-06 10:04:30
4400
发布2021-09-06 10:04:30
举报
文章被收录于专栏:Michael阿明学习之路

文章目录

1. 题目

给你两个长度可能不等的整数数组 nums1 和 nums2 。 两个数组中的所有值都在 1 到 6 之间(包含 1 和 6)。

每次操作中,你可以选择 任意 数组中的任意一个整数,将它变成 1 到 6 之间 任意 的值(包含 1 和 6)。

请你返回使 nums1 中所有数的和与 nums2 中所有数的和相等最少操作次数。 如果无法使两个数组的和相等,请返回 -1 。

代码语言:javascript
复制
示例 1:
输入:nums1 = [1,2,3,4,5,6], nums2 = [1,1,2,2,2,2]
输出:3
解释:你可以通过 3 次操作使 nums1 中所有数的和与 nums2 中所有数的和相等。
以下数组下标都从 0 开始。
- 将 nums2[0] 变为 6 。 nums1 = [1,2,3,4,5,6], nums2 = [6,1,2,2,2,2] 。
- 将 nums1[5] 变为 1 。 nums1 = [1,2,3,4,5,1], nums2 = [6,1,2,2,2,2] 。
- 将 nums1[2] 变为 2 。 nums1 = [1,2,2,4,5,1], nums2 = [6,1,2,2,2,2] 。

示例 2:
输入:nums1 = [1,1,1,1,1,1,1], nums2 = [6]
输出:-1
解释:没有办法减少 nums1 的和或者增加 nums2 的和使二者相等。

示例 3:
输入:nums1 = [6,6], nums2 = [1]
输出:3
解释:你可以通过 3 次操作使 nums1 中所有数的和与 nums2 中所有数的和相等。
以下数组下标都从 0 开始。
- 将 nums1[0] 变为 2 。 nums1 = [2,6], nums2 = [1] 。
- 将 nums1[1] 变为 2 。 nums1 = [2,2], nums2 = [1] 。
- 将 nums2[0] 变为 4 。 nums1 = [2,2], nums2 = [4] 。
 
提示:
1 <= nums1.length, nums2.length <= 10^5
1 <= nums1[i], nums2[i] <= 6

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/equal-sum-arrays-with-minimum-number-of-operations 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 排序,优先使用 sum 大的数组 能降低的最多的,或者 sum 小的数组能升高最多的
代码语言:javascript
复制
class Solution {
public:
    int minOperations(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size(), len2 = nums2.size();
        if(len1 > 6*len2 || len2 > 6*len1) return -1;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int s1 = accumulate(nums1.begin(), nums1.end(),0);
        int s2 = accumulate(nums2.begin(), nums2.end(),0);
        if(s1 > s2)
        {
            swap(nums1, nums2);
            swap(s1, s2);
            swap(len1, len2);
        }
        // 令 s1 <= s2
        int ans = 0, i = 0, j = len2-1;
        int delta1, delta2;
        while(s1 < s2 && (i < len1 || j >= 0))
        {
            delta1 = i < len1 ? 6-nums1[i] : 0;
            delta2 = j >= 0 ? nums2[j]-1 : 0;
            if(i < len1 && delta1 >= delta2)
            {
                s1 += delta1;
                i++;
                ans++;
            }
            else if(j >= 0 && delta2 >= delta1)
            {
                s2 -= delta2;
                j--;
                ans++;
            }
        }
        return ans;
    }
};

144 ms 111 MB C++


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/02/28 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档