1
class Solution {
2
 public:
3
  vector<int> productExceptSelf(vector<int> &nums) {
4
    int zero = 0, product = 1;
5
    for (auto i : nums) {
6
      if (i == 0)
7
        zero++;
8
      else
9
        product *= i;
10
    }
11
    for (int i = 0; i < nums.size(); i++) {
12
      if (nums[i] == 0 && zero > 1) {
13
        nums[i] = 0;
14
      } else if (nums[i] == 0) {
15
        nums[i] = product;
16
      } else if (zero > 0) {
17
        nums[i] = 0;
18
      } else
19
        nums[i] = product / nums[i];
20
    }
21
    return nums;
22
  }
23
};