C++程式:計算使和與積均不為零所需的步數


假設我們有一個包含n個元素的陣列A。在一個操作中,我們可以將陣列A中任何一個現有元素加1。如果陣列中所有元素的和或積等於零,我們可以再進行一次此操作。我們需要計算使陣列中所有元素的和與積都不為零所需的最小步數?

因此,如果輸入類似於A = [-1, 0, 0, 1],則輸出為2,因為積和和都為0。如果我們將1加到第二個和第三個元素,則陣列將變為[−1,1,1,1],和將等於2,積將等於−1。

步驟

為了解決這個問題,我們將遵循以下步驟:

sum := 0
cnt := 0
n := size of A
for initialize i := 0, when i < n, update (increase i by 1), do:
   x := A[i]
   sum := sum + x
   cnt := cnt + (if x is same as 0, then 1, otherwise 0)
return (if sum + cnt is same as 0, then cnt + 1, otherwise cnt)

示例

讓我們來看下面的實現以更好地理解:

#include <bits/stdc++.h>
using namespace std;

int solve(vector<int> A) {
   int sum = 0, cnt = 0;
   int n = A.size();
   for (int i = 0; i < n; i++) {
      int x = A[i];
      sum += x;
      cnt += x == 0 ? 1 : 0;
   }
   return sum + cnt == 0 ? cnt + 1 : cnt;
}
int main() {
   vector<int> A = { -1, 0, 0, 1 };
   cout << solve(A) << endl;
}

輸入

{ -1, 0, 0, 1 }

輸出

2

更新於:2022年3月3日

瀏覽量:237

開啟您的職業生涯

完成課程獲得認證

開始學習
廣告
© . All rights reserved.