1775 通过最少操作次数使数组的和相等

题目描述:
给你两个长度可能不等的整数数组 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 <= 105
1 <= nums1[i], nums2[i] <= 6

方法1:
主要思路:解题链接汇总
(1)如代码;

class Solution {
    
    
public:
	//将diff逐渐消除到0
    int get_res(vector<int>&mp1, vector<int>&mp2,int diff) {
    
    
		int res = 0;
		//消除的方式是,和更大的数组尽量的向小变化,和更小的数组尽量向大变化
		//统计为一组数,统一为一个变换方向,既向更大的方向变化
		for (int i = 1; i < mp1.size(); ++i) {
    
    
			mp1[i] += mp2[7-i];
		}
		//消除差值
		while (diff != 0) {
    
    
			//尽量每次变化大些,减少变化次数
			for (int i = 1; i < 7; ++i) {
    
    
				if (mp1[i] != 0) {
    
    //当前有可以变换的数字
					//找出要变换的数字
					int t = 6 - i;
					while (diff < t) {
    
    
						--t;
					}
					//找出要变换的个数
					int c = diff / t;
					if (mp1[i] >= c) {
    
    
						mp1[i] -= c;
						mp1[i + t] += c;
						res += c;
						diff -= c * t;
					}
					else {
    
    
						mp1[i + t] += mp1[i];
						res += mp1[i];
						diff -= mp1[i] * t;
						mp1[i] = 0;
					}
					break;
				}
			}
		}
		return res;
		return res;
	}
	int minOperations(vector<int>& nums1, vector<int>& nums2) {
    
    
		//一定无解的情形
		if ((nums1.size() > nums2.size()&&nums1.size()>nums2.size()*6)||
			(nums1.size() < nums2.size() && nums1.size()*6 < nums2.size()) ) {
    
    
			return -1;
		}
		//统计各个数字的个数
		vector<int> mp1(7, 0);
		vector<int> mp2(7, 0);
		//计算两组元素的和
        int sum1 = 0, sum2 = 0;
		for (int&i : nums1) {
    
    
			mp1[i]++;
            sum1+=i;
		}
		for (int&i : nums2) {
    
    
			mp2[i]++;
            sum2+=i;
		}
		//若和相等,则直接返回true
		if (sum1 == sum2) {
    
    
			return 0;
		}
		//否则,获得和的差值
		int diff = abs(sum1 - sum2);
		//调用计算变换次数的数字
		if (sum1 < sum2) {
    
    
			return get_res(mp1, mp2,diff);
		}
		return get_res(mp2, mp1,diff);
	}
};

猜你喜欢

转载自blog.csdn.net/weixin_44171872/article/details/114266973