剑指offer之构建乘积数组

题目描述

给定一个数组A[0,1,…,n-1],请构建一个数组B[0,1,…,n-1],其中B中的元素$B[i]=A[0]\ast A[1]\ast ...\ast A[i-1]\ast A[i+1]\ast ...\ast A[n-1]$。不能使用除法。

思路

暴力求解

AC代码

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        std::vector<int> B;
        if (A.empty()) {
            return B;
        }
        int n = A.size();
        for (int i = 0; i < n; ++i) {
            int res = 1;
            for (int j = 0; j < n; ++j) {
                if (i == j) {
                    continue;
                } else if (A[j] == 0) {
                    res = 0;
                    break;
                } else {
                    res *= A[j];
                }
            }
            B.push_back(res);
        }
        return B;
    }
};