数组乘积

数组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]。即除过自身位置外的其他元素乘积。
输入: [1,2,3,4,5]
输出: [120,60,40,30,24]

解题思路:
构建两个数组left,right:
其中left从左到右计算出前半部分元素的成绩,right从右往左计算出后半部分元素的成绩,则所求位置的B[i]为该元素位置的前半部分与后半部分的乘积。时间复杂度o(n),代码如下:

class Solution{
public:
    vector<int> constructVec(vector<int> A)
    {
        vector<int> ans;
        int len=A.size();
        if(len==0) return ans;
        vector<int> left(len,1);//保存左半部分乘积
        vector<int> right(len,1);//保存右半部分乘积
        for(int i=1;i!=len;i++)
        {
          left[i]=left[i-1]*A[i-1];
        }
        for(int j=len-2;j!=-1;j--)
        {
          right[j]=right[j+1]*A[j+1];
        }
        for(int k=0;k!=len;k++)
        {
          ans.push_back(left[k]*right[k]);
        }
        return ans;
    }
}
发布了22 篇原创文章 · 获赞 1 · 访问量 349

猜你喜欢

转载自blog.csdn.net/weixin_43086349/article/details/104634926