#50 Product of Array Exclude Itself

最新推荐文章于 2023-08-29 11:16:19 发布

易月

最新推荐文章于 2023-08-29 11:16:19 发布

阅读量365

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode 文章标签： lintcode array

本文链接：https://blog.youkuaiyun.com/haifischxia/article/details/52343593

lintcode 专栏收录该内容

217 篇文章

订阅专栏

解决一个整数数组中每个元素的乘积问题，不使用除法操作。通过预先计算左右累积乘积来高效求解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

Given an integers array A.

Define B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1], calculate BWITHOUT divide operation.

Have you met this question in a real interview?

Yes

Example

For A = [1, 2, 3], return [6, 3, 2].

题目思路：

这题用一个left数组去记录i往左所有数的乘积；right数组去记录i往右所有数的乘积。然后重新遍历一遍数组，每个i的答案就是left[i - 1] * right[i + 1]。

Mycode（AC = 12ms）：

class Solution {
public:
    /**
     * @param A: Given an integers array A
     * @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
     */
    vector<long long> productExcludeItself(vector<int> &nums) {
        // write your code here
        vector<long long> ans;
        if (nums.size() == 0) return ans;
        else if (nums.size() == 1) {
            ans.push_back(1);
            return ans;
        }
        
        ans.resize(nums.size(), 1);
        vector<long long> left(nums.size(), nums[0]), right(nums.size(), nums[nums.size() - 1]);
        
        // left[i] denote the product of all numbers
        // from 0...i
        for (int i = 1; i < nums.size(); i++) {
            left[i] = left[i - 1] * nums[i];
        }
        
        // right[i] denote the product of all numbers
        // from i...nums.size() - 1
        for (int i = nums.size() - 2; i >= 0; i--) {
            right[i] = right[i + 1] * nums[i];
        }
        
        for (int i = 0; i < nums.size(); i++) {
            if (i == 0) {
                ans[i] = right[i + 1];
            }
            else if (i == nums.size() - 1) {
                ans[i] = left[i - 1];
            }
            else {
                ans[i] = right[i + 1] * left[i - 1];
            }
        }
        
        return ans;
    }
};