Product of array except self

Product of array except self

vector productExceptSelf(vector& nums) {

    int product = 1;
    //乘积从1 开始
    int zero_product = 1;
    //除0元素外的乘积

    int tmp;
    //存放output临时的值

    vector<int> output;

    //对input的每个元素 求乘积
    for (int i = 0; i < nums.size(); i++) {

        product *= nums[i];

    }

    //对output的每个元素
    for (int i = 0; i < nums.size(); i++) {

        if (nums[i] != 0) {

            tmp = product / nums[i];

        }
        else {

            for (int j = 0; j < nums.size(); j++) {

                if (j != i) {

                    zero_product *= nums[j];

                }

            }

            tmp = zero_product;

        }

        //放入output数组
        output.push_back(tmp);

    }

    return output;

}