构建乘积数组

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-2 20:53   920   0

题目:

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

思考:

既然不能用除法,那么就不能先算出A[0] - A[n - 1]的乘积,再除以 i 对应的A[i]的值。

那么,我们可以分两步来求出数组B中的B[i]对应的值。

1、先算出A[0] - A[i - 1]的乘积,放入数组B中。

即先算出B[i] = A[0] * A[1] * A[2] * ... A[i - 1]

在这一部分,存在B[i] = B[i - 1] * A[i - 1](n > 0); 其中,令B[0] = 1;

2、在上一步的基础上,再算B[i] *= B[i + 1] * B[i + 2] * ... B[n - 1];

在这一步中

B[n - 1] *= 1;

B[n - 2] *= A[n -1];

B[n -3] *= A[n - 2] * A[n - 1];

B[n - 4] *= A[n - 3] * A[n - 2] * A[n - 1];

......

B[0] *= A[1] * A[2] * ... A[n - 3] * A[n - 2] * A[n - 1];

不难发现,从n - 1到 0, 所乘上的值 t 满足, t(i) = t(i + 1) * A[i + 1];

class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        int len = A.size();
        vector<int> B(len);
        if(len == 0) return B;
        B[0] = 1;
        for(int i = 1; i < len; i++) {
            B[i] = B[i - 1] * A[i - 1];
        }
        int t = 1;
        for(int i = len - 2; i >= 0; i--) {
            t *= A[i + 1];
            B[i] *= t;
        }
        return B;
    }
};

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP