【Leetcode】2918. Minimum Equal Sum of Two Arrays After Replacing Zeros

题目地址：

https://leetcode.com/problems/minimum-equal-sum-of-two-arrays-after-replacing-zeros/description/

给定两个非负整数数组$a$和$b$，要求必须将所有的$0$变为任意正整数，使得两个数组的和相等。问这个最小的相等的和是多少。如果做不到则返回$-1$。

先求数组各自的和和各自有多少个$0$。由于必须变成正整数，所以最终答案一定不小于 max ⁡ { ∑ a + c a , ∑ b + c b } \max\{\sum a+c_a, \sum b+c_b\} max{∑a+ca​,∑b+cb​}，$c_a$表示$a$里$0$的个数，$c_b$一样。设$s_a=\sum a+c_a,s_b=\sum b+c_b$，如果$s_a=s_b$，那最终答案就是$s_a$；如果$s_a>s_b$，此时要看$c_b$，如果$c_b=0$，那$b$的数组和无法再变大了，此时无解；否则可以让$b$中为$0$的数变得再大一些即可，返回$s_a$；反之亦然。代码如下：

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(1)$。