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

最新推荐文章于 2022-12-07 18:05:56 发布

江南小张

最新推荐文章于 2022-12-07 18:05:56 发布

阅读量289

点赞数 1

分类专栏： leetcode 文章标签： java leetcode 算法

本文链接：https://blog.csdn.net/nihaoa_1898/article/details/114601168

版权

leetcode 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

leetcode_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

代码：

class Solution {
    public int minOperations(int[] nums1, int[] nums2) {
        if(nums1.length*6<nums2.length||nums2.length*6<nums1.length)    return -1;
        //记录两边目前的和以及差值，每个数出现的次数
        int[] arr1=new int[7];
        int[] arr2=new int[7];

        int sum1=0;
        int sum2=0;
        int count=0;
        //这种写法可以借鉴一下，多个数组求和其中的，可以合并在一起写，减少时间复杂度，让代码也更美观
        for(int i=0,j=0;i<nums1.length||j<nums2.length;i++,j++){
            if(i<nums1.length){
                arr1[nums1[i]]++;
                sum1+=nums1[i];
            }
            if(j<nums2.length){
                arr2[nums2[j]]++;
                sum2+=nums2[j];
            }
        }
        //看一下需要处理多少差值
        int diff=Math.abs(sum1-sum2);
        for(int c=1;diff>0&&c<7;c++){
             /*贪心算法的思想，尽可能每一步都是最优的解，
             和比较小的就让目前最小的数字变成6，和大的就让当前最大的数字变成1*/
            int i=sum1>sum2?7-c:c;
            int j=sum1>sum2?c:7-c;
            while(arr1[i]>0||arr2[j]>0){
                count++;
                //这里为什么是differ-（6-c）呢？因为count是先自加1，就代表当前已经处理过了
                if(diff-(6-c)>0){
                    if(arr1[i]>0)   arr1[i]--;
                    else if(arr2[j]>0) arr2[j]--;
                    diff-=6-c;
                }else
                    return count;
            }
        }
        return count;
    }
}