剑指offer之构建乘积数组

剑指offer之构建乘积数组

问题描述

问题分析

1. 题目意思很好理解，如果能够使用除法就很简单了，先算出A中各个元素的乘积s，然后B中对应元素就等于s除以对应的元素就行了
2. 现在不能使用除法，一开始我的思路是：当遇到当前元素的时候让这个元素置为1，然后循环让A中每个元素相乘。这样很明显对每一个元素都要求一次A中每个元素的乘积，O(n2)的复杂度。这时候我就在想能否使用记忆化记录已经算出的乘积然后推导出B中的元素，哈哈，没想到怎么做。
3. DP做法：首先要清楚，B中当前元素就是A中当前元素的左边所有元素的乘积和右边所有元素的乘积的乘积。想通这点，其实就是一个数组存着A中当前元素左边所有的元素的乘积，一个数组存A中当前元素右边所有的元素的乘积。然后二者相乘即可。

代码表述

暴力解法，超时。竟然过了43个用例，就一个例子没过。

class Solution {
public:
    vector<int> constructArr(vector<int>& a) {
        vector<int>res;
        for(int i=0;i<a.size();i++){
            int tmp=a[i];
            a[i]=1;
            int sum=1;
            for(int j=0;j<a.size();j++){
                sum*=a[j];
            }
            res.push_back(sum);
            a[i]=tmp;
        }
        return res;

    }
};

DP做法。
前缀积和后缀积？？

class Solution {
public:
    vector<int> constructArr(vector<int>& a) {
        //当前元素的左边所有元素的乘积*当前元素的右边所有元素的乘积就是B数组位置对应的元素
        //dp解法
        int n=a.size();
        vector<int>res(n,1);
        vector<int>left(n,1);//保存当前元素的左边所有元素的乘积
        vector<int>right(n,1);//保存当前元素的右边所有元素的乘积
        for(int i=1;i<n;i++){
            left[i]=left[i-1]*a[i-1];//乘积按照顺序的，所以由上一个元素得到的乘积乘上当前元素就好
        }
        for(int i=n-2;i>=0;i--){
            right[i]=right[i+1]*a[i+1];//同理
        }
        for(int i=0;i<n;i++){
            res[i]=left[i]*right[i];
        }
        return res;
    }
};

时间复杂度：O（N）。三次for循环都要将数组遍历一遍
空间复杂度：O(N).额外的两个左右数组存储乘积。
可以直接在原数组上进行操作。

class Solution {
public:
    vector<int> constructArr(vector<int>& a) {
        //当前元素的左边所有元素的乘积*当前元素的右边所有元素的乘积就是B数组位置对应的元素
        //dp解法
        int n=a.size();
        vector<int>res(n,1);
        for(int i=1;i<n;i++){
            res[i]=res[i-1]*a[i-1];
        }
        for(int i=n-2;i>=0;i--){
            a[i]*=a[i+1];//直接更改原数组的元素值
            res[i]=a[i+1]*res[i];//再次更改所求数组的元素值
        }
        return res;
    }
};

时间复杂度：O（N）。for循环都要将数组遍历一遍
空间复杂度：O(1).没使用额外的数组。

复杂度分析

见上。