leetcode.1775 通过最少操作次数使数组的和相等 - 贪心 + 哈希表

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

思路：

数字和较小的数组尽可能将数变成6，数字和较大的数组尽可能将数变成1，这样能最快将差距d缩小

• 计算数组一的sum值s1，数组二的sum值s2 如果s1=s2 则返回操作数为0
• 否则我们设 s2 > s1
• 让和大的数组变为1，让和小的数组变为6，记录变化差值d
• 因为题目变化值为0~5，所以开一个cnt[6]数组记录变化量的个数
• 我们从最大变化量 i=5开始，每次贪心地将 d 减去最大变化量 i，直到 d≤0，返回此时的操作次数即可

class Solution {
    public int minOperations(int[] nums1, int[] nums2) {
        int s1=Arrays.stream(nums1).sum();
        int s2=Arrays.stream(nums2).sum();
        if(s1==s2) return 0;
        if(s1>s2) return minOperations(nums2,nums1);
        int d=s2-s1;
        int[] cnt=new int[6];
        int res=0;

        for(int x:nums1) cnt[6-x]++;
        for(int x:nums2) cnt[x-1]++;
        for(int i=5;i>=0;i--)
            while(cnt[i]>0&&d>0)
            {
                d-=i;
                res++;
                cnt[i]--;
            }
        return d<=0? res:-1;
}
}

class Solution {
public:
    int minOperations(vector<int>& nums1, vector<int>& nums2) {
        int s1 = accumulate(nums1.begin(), nums1.end(), 0);
        int s2 = accumulate(nums2.begin(), nums2.end(), 0);
        if (s1 == s2) return 0;
        if (s1 > s2) return minOperations(nums2, nums1);
        int d = s2 - s1;
        int cnt[6] = {0};
        for (int& v : nums1) ++cnt[6 - v];
        for (int& v : nums2) ++cnt[v - 1];
        int ans=0;
        for (int i=5; i; --i)
            while(cnt[i]&&d>0) 
            {
                d -= i;
                --cnt[i];
                ++ans;
            }
        return d <= 0 ? ans : -1;
    }
};