Leetcode 1775. 通过最少操作次数使数组的和相等
admin
2024-03-23 16:12:03
0次
给你两个长度可能不等的整数数组 nums1 和 nums2 。两个数组中的所有值都在 1 到 6 之间(包含 1 和 6)。每次操作中,你可以选择 任意 数组中的任意一个整数,将它变成 1 到 6 之间 任意 的值(包含 1 和 6)。请你返回使 nums1 中所有数的和与 nums2 中所有数的和相等的最少操作次数。如果无法使两个数组的和相等,请返回 -1 。示例 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 <= 1051 <= nums1[i], nums2[i] <= 6

解法一:贪心 + 计数

我们首先应该从正反两个方向去分析,对于不合法的情况:只要两个数组一个全部变成6,一个全部变成1后仍然达不到要求,即变成6的数组和小于变成1的数组和,那么这种情况直接返回-1。


对于合法情况,一定能够使得两个数组的和相等,那么如何让总的操作次数最少呢?


我们用sum1sum1sum1和sum2sum2sum2分别代表nums1nums1nums1和nums2nums2nums2的和,这里不妨假设sum1>sum2sum1 > sum2sum1>sum2。那么为了能使得两个数组的和相等,我们需要增加sum2的值,减少sum1的值,并且让操作次数尽可能少。为了使操作次数尽可能地少,很直观的一种想法就是尽可能每一次让增加或减少的数尽可能大。我们可以用targettargettarget表示sum1−sum2sum1 - sum2sum1−sum2,我们的目标是使targettargettarget变为0。


为了使每一次增加的数尽可能大,减少的数尽可能小,我们首先让nums1nums1nums1中所有的666先减少,然后再让nums2nums2nums2中所有的111增加,接下来一次是5和2,…2和5,直至targettargettarget变为0。


我们可以用哈希表来统计每次的增加量和减少量,对于nums1nums1nums1中的6每次减少量为5,num2num2num2中的1每次增加量为5,这些操作都会使得target减少,那么统计nums1nums1nums1中的6和nums2nums2nums2中的1保存再mp[5]中,代表每次target能够减少的次数。

  • 时间复杂度:O(n)O(n)O(n)
  • 空间复杂度:O(1)O(1)O(1)
class Solution {int[] mp = new int[6];public int minOperations(int[] nums1, int[] nums2) {int n1 = nums1.length, n2 = nums2.length, ans = 0;if (n1 * 6 < n2 || n2 * 6 < n1) return -1; //一定不可行int target = Arrays.stream(nums1).sum() - Arrays.stream(nums2).sum(); //可行情况if (target < 0) return minOperations(nums2, nums1);//默认nums1为sum大的数组 for (int x : nums1) mp[x - 1]++; //对于每个数可以减少的量for (int x : nums2) mp[6 - x]++; //对于每个数可以增加的量 for (int i = 5; i >= 1 && target > 0; i--) {int cnt = Math.min(mp[i], (target + i - 1) / i); //最少需要的个数 +(i- 1)为了向上取整。ans += cnt;target -= cnt * i; //减去能够减少的最大值,若小于等等于0,代表已经使得两个数组相等}return ans;}
}
class Solution {
public:  int minOperations(vector& nums1, vector& nums2) {    int n1 = nums1.size(), n2 = nums2.size(), ans = 0;if (n1 * 6 < n2 || n2 * 6 < n1) return -1; //一定不可行int target = accumulate(nums1.begin(), nums1.end(), 0) - accumulate(nums2.begin(), nums2.end(), 0); //可行情况if (target < 0) return minOperations(nums2, nums1);//默认nums1为sum大的数组   vector mp(6, 0);for (int x : nums1) mp[x - 1]++; //对于每个数可以减少的量for (int x : nums2) mp[6 - x]++; //对于每个数可以增加的量 for (int i = 5; i >= 1 && target > 0; i--) {int cnt = min(mp[i], (target + i - 1) / i); //最少需要的个数 +(i- 1)为了向上取整。ans += cnt;target -= cnt * i; //减去能够减少的最大值,若小于等等于0,代表已经使得两个数组相等}return ans;}
};

如果有问题,欢迎评论区交流, 如果有帮助到你,请给题解点个赞和收藏哈~~~

相关内容

热门资讯

AI做攻略、“双向奔赴”……国... 本文转自:人民网人民网记者 许维娜今年国庆假期,AI(人工智能)成了不少年轻人的“旅行搭子”。行前做...
合肥的这个“梗”, 翻篇了 社交平台的评论区,曾长期流传一句专属于合肥的黑色幽默。有人问合肥哪里好玩,本地网友半开玩笑地回复:合...
7万+旅游中高层都在的群,今天... 做旅游这行,资源卡位、渠道互通、项目落地——往往就差一个“对的机会”。我们的行业社群已聚集70,00...
一场帐篷音乐节,如何让井研从“... 帐篷音乐节现场封面新闻记者 周洪攀10月的研溪湿地,晚风里裹着柑橘的清甜与民谣的回响。夜幕降临,10...
白鹤翩翩落星愿,上海迪士尼度假... 10月9日,世界候鸟日前夕,上海迪士尼度假区星愿公园的湖风里多了一分白鹤的灵动。三座白鹤雕塑正式揭幕...
成都锦江边赏初秋枫叶飘飘 目前,成都锦江边国华街枫叶树红了。近年来,国华街附近的几条街都栽种了红枫树,是城区较集中看枫叶的点位...
GBTA:商务旅行正在复苏,但... 亚太正在成为全球商务旅行增长的重要引擎。在经历疫情后的快速反弹之后,全球商务旅行正在进入一个新的增长...
从景区焕新到节赛引流 邹城双节... 齐鲁网·闪电新闻10月9日讯 中秋、国庆“双节”假期,邹城市文化和旅游局提前谋划、周密部署,立足自身...
江山如此多娇 且看石楼今朝 金秋十月,黄河岸畔秋意盎然,丰收盛会喜迎国庆。本次石楼国庆文旅活动以“江山如此多娇,且看石楼今朝”为...
海南国庆假期接待游客475.4... 2026年国庆假期,海南接待游客475.40万人次,较2025年增长13.1%,较2024年增长15...