题目地址:
https://leetcode.com/problems/minimum-equal-sum-of-two-arrays-after-replacing-zeros/description/
给定两个非负整数数组 a a a和 b b b,要求必须将所有的 0 0 0变为任意正整数,使得两个数组的和相等。问这个最小的相等的和是多少。如果做不到则返回 − 1 -1 −1。
先求数组各自的和和各自有多少个 0 0 0。由于必须变成正整数,所以最终答案一定不小于 max { ∑ a + c a , ∑ b + c b } \max\{\sum a+c_a, \sum b+c_b\} max{∑a+ca,∑b+cb}, c a c_a ca表示 a a a里 0 0 0的个数, c b c_b cb一样。设 s a = ∑ a + c a , s b = ∑ b + c b s_a=\sum a+c_a, s_b=\sum b+c_b sa=∑a+ca,sb=∑b+cb,如果 s a = s b s_a=s_b sa=sb,那最终答案就是 s a s_a sa;如果 s a > s b s_a>s_b sa>sb,此时要看 c b c_b cb,如果 c b = 0 c_b=0 cb=0,那 b b b的数组和无法再变大了,此时无解;否则可以让 b b b中为 0 0 0的数变得再大一些即可,返回 s a s_a sa;反之亦然。代码如下:
class Solution {
public:
using ll = long long;
ll minSum(vector<int>& a, vector<int>& b) {
auto f = [&](vector<int>& a) -> pair<ll, int> {
ll sum = 0;
int cnt = 0;
for (int i = 0; i < a.size(); i++)
if (!a[i]) cnt++;
else sum += a[i];
return {sum, cnt};
};
auto p1 = f(a), p2 = f(b);
ll sum1 = p1.first, sum2 = p2.first;
int cnt1 = p1.second, cnt2 = p2.second;
sum1 += cnt1;
sum2 += cnt2;
if (sum1 > sum2) {
if (!cnt2) return -1;
else return sum1;
} else if (sum1 < sum2) {
if (!cnt1) return -1;
else return sum2;
} else return sum1;
}
};
时间复杂度 O ( l a + l b ) O(l_a+l_b) O(la+lb),空间 O ( 1 ) O(1) O(1)。