刷题笔记LeetCode.1775

最新推荐文章于 2024-09-23 11:27:55 发布

T-Block

最新推荐文章于 2024-09-23 11:27:55 发布

阅读量91

点赞数

文章标签： leetcode

本文链接：https://blog.csdn.net/m0_53715593/article/details/128235188

版权

记录一下个人刷题，方便到时复习

LeetCode.1775

给你两个长度可能不等的整数数组 nums1 和 nums2 。两个数组中的所有值都在 1 到 6 之间（包含 1 和 6）。
每次操作中，你可以选择任意数组中的任意一个整数，将它变成 1 到 6 之间任意的值（包含 1 和 6）。
请你返回使 nums1 中所有数的和与 nums2 中所有数的和相等的最少操作次数。如果无法使两个数组的和相等，请返回 -1 。

想了10几分钟，不会，下面是参照灵茶山艾府的思路和代码，自己写的总结。

思路

贪心，让和较大的数组的每个元素变为1，和较小的数组的每个元素变为6，能使得差值迅速下降。
然后记录变化量分别为1，2，3，4，5的个数。
从大到小遍历这些变化量，比较能否使得差值变为0。

实现

int minOperations(vector<int>& num1, vector<int>& num2) {
        int d = (accumulate(num2.begin(),num2.end(), 0) - accumulate(num1.begin(),num1.end(), 0));
        if (d < 0)
        {
            d = -d;
            swap(num1, num2);
        }
        int cnt[6]{};
        for(auto& i : num1) ++cnt[6 - i];
        for(auto& i : num2) ++cnt[i - 1];
        for(int i = 5, ans = 0;i >= 0;--i)
        {
            if (d <= i * cnt[i])
            {
                return ans + (d + i - 1)/i;//向上取整
            }
            ans += cnt[i];//本轮的都得要
            d -= i*cnt[i];//去掉本轮变化量的贡献
        }
        return -1;
    }