【Leetcode 每日一题】2918. 数组的最小相等和

最新推荐文章于 2025-05-12 22:51:52 发布

冠位观测者

最新推荐文章于 2025-05-12 22:51:52 发布

阅读量527

点赞数 18

分类专栏： Leetcode Daily 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/2401_88859777/article/details/147846723

版权

Leetcode Daily 专栏收录该内容

187 篇文章

订阅专栏

问题背景

给你两个由正整数和 $0$ 组成的数组 $nums_1$ 和 $nums_2$ 。
你必须将两个数组中的所有 $0$ 替换为严格正整数，并且满足两个数组中所有元素的和相等。
返回最小相等和，如果无法使两数组相等，则返回 $- 1$ 。

数据约束

$\le nums_1.length, nums_2.length \le 10 ^ 5$
$\le nums_1[i], nums_2[i] \le 10 ^ 6$

解题过程

本题中替换的元素只有下限没有上限，意味着如果不要求最小只要求相等，最终在数学上两个数组的和可以无限大。
因此，能否实现相等取决于两个数组中有没有不包含 $0$ 且和很小的。能实现的情况下，最小和则是原本数组中元素之和与 $0$ 的数量的和。

具体实现

class Solution {
    public long minSum(int[] nums1, int[] nums2) {
        long sum1 = 0, sum2 = 0;
        int count1 = 0, count2 = 0;
        for (int num : nums1) {
            sum1 += num;
            count1 += num == 0 ? 1 : 0;
        }
        for (int num : nums2) {
            sum2 += num;
            count2 += num == 0 ? 1 : 0;
        }
        if ((sum1 + count1 > sum2 && count2 == 0) || (sum2 + count2 > sum1 && count1 == 0)) {
            return -1;
        }
        return Math.max(sum1 + count1, sum2 + count2);
    }
}