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;}
};

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

相关内容

热门资讯

linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
C++ 机房预约系统(六):学... 8、 学生模块 8.1 学生子菜单、登录和注销 实现步骤: 在Student.cpp的...
A.机器学习入门算法(三):基... 机器学习算法(三):K近邻(k-nearest neigh...
数字温湿度传感器DHT11模块... 模块实例https://blog.csdn.net/qq_38393591/article/deta...
有限元三角形单元的等效节点力 文章目录前言一、重新复习一下有限元三角形单元的理论1、三角形单元的形函数(Nÿ...
Redis 所有支持的数据结构... Redis 是一种开源的基于键值对存储的 NoSQL 数据库,支持多种数据结构。以下是...
win下pytorch安装—c... 安装目录一、cuda安装1.1、cuda版本选择1.2、下载安装二、cudnn安装三、pytorch...
MySQL基础-多表查询 文章目录MySQL基础-多表查询一、案例及引入1、基础概念2、笛卡尔积的理解二、多表查询的分类1、等...
keil调试专题篇 调试的前提是需要连接调试器比如STLINK。 然后点击菜单或者快捷图标均可进入调试模式。 如果前面...
MATLAB | 全网最详细网... 一篇超超超长,超超超全面网络图绘制教程,本篇基本能讲清楚所有绘制要点&#...
IHome主页 - 让你的浏览... 随着互联网的发展,人们越来越离不开浏览器了。每天上班、学习、娱乐,浏览器...
TCP 协议 一、TCP 协议概念 TCP即传输控制协议(Transmission Control ...
营业执照的经营范围有哪些 营业执照的经营范围有哪些 经营范围是指企业可以从事的生产经营与服务项目,是进行公司注册...
C++ 可变体(variant... 一、可变体(variant) 基础用法 Union的问题: 无法知道当前使用的类型是什...
血压计语音芯片,电子医疗设备声... 语音电子血压计是带有语音提示功能的电子血压计,测量前至测量结果全程语音播报࿰...
MySQL OCP888题解0... 文章目录1、原题1.1、英文原题1.2、答案2、题目解析2.1、题干解析2.2、选项解析3、知识点3...
【2023-Pytorch-检... (肆十二想说的一些话)Yolo这个系列我们已经更新了大概一年的时间,现在基本的流程也走走通了,包含数...
实战项目:保险行业用户分类 这里写目录标题1、项目介绍1.1 行业背景1.2 数据介绍2、代码实现导入数据探索数据处理列标签名异...
记录--我在前端干工地(thr... 这里给大家分享我在网上总结出来的一些知识,希望对大家有所帮助 前段时间接触了Th...
43 openEuler搭建A... 文章目录43 openEuler搭建Apache服务器-配置文件说明和管理模块43.1 配置文件说明...