[LEETCODE] 1822. Sign of the Product of an Array

https://leetcode.com/

TITLE : 1822. Sign of the Product of an Array

Description : There is a function signFunc(x) that returns:

• 1 if x is positive.
• -1 if x is negative.
• 0 if x is equal to 0.

You are given an integer array nums. Let product be the product of all values in the array nums.

Return signFunc(product).

 

Example 1:

Input: nums = [-1,-2,-3,-4,3,2,1]
Output: 1
Explanation: The product of all values in the array is 144, and signFunc(144) = 1

Example 2:

Input: nums = [1,5,0,2,-3]
Output: 0
Explanation: The product of all values in the array is 0, and signFunc(0) = 0

Example 3:

Input: nums = [-1,1,-1,1,-1]
Output: -1
Explanation: The product of all values in the array is -1, and signFunc(-1) = -1

 

Constraints:

• 1 <= nums.length <= 1000
• -100 <= nums[i] <= 100

---

array 에 들어있는 원소들의 곱의 부호에 따라 -1, 0, 1 을 출력해주면 된다. 사실 처음에는 실제로 그냥 다 곱해서 부호를 뽑아내면 되겠다고 생각했는데, Constraints 를 보면 원소의 크기는 100, 개수는 1000 이 들어온다고 치면 long long 형으로도 다 표현하지 못하는 경우가 발생한다. 구해야하는 거는 부호이고, 부호는 음수가 몇개 있는지에 따라서 결정된다. 그러므로 음수의 개수를 세고, 음수가 짝수이면 1, 아니면 -1 을 반환한다. ( 반복문을 돌다가 array 의 원소가 0 이 나오면 바로 0 을 반환한다. )

 

시간 복잡도 : O(N)

 

[C++]

class Solution {
public:
    int arraySign(vector<int>& nums) {
        int positive_cnt = 0; int negative_cnt = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] == 0) return 0;
            else if (nums[i] > 1) positive_cnt++;
            else if (nums[i] < 0) negative_cnt++;
        }
        if (negative_cnt % 2 == 1) return -1;
        else return 1;
    }
};