码迷,mamicode.com
首页 > 编程语言 > 详细

48 构建乘积数组

时间:2018-01-13 00:27:01      阅读:133      评论:0      收藏:0      [点我收藏+]

标签:优化   height   for循环   除法   gpo   题目   otto   ++   multi   

题目描述

给定一个数组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]。不能使用除法。
收获:先把公式想清楚了再编程序就会思路清晰,写程序写的快。
b[0] 1 A[1] A[2] A[n - 1]
b[1] A[0] 1 A[2] A[n - 1]
....... A[0] A[1] A[2] A[n - 1]
b[n - 2] A[0] A[1] 1 A[n - 1]
b[n - 1] A[0] A[1] A[2] 1
可以将B[i]分为两部分,左边乘以右边,B[i] = L[i] * R[i];
L[0] = 1,L[i] = L[i - 1] * A[i - 1];
R[n - 1] = 1,R[i] = R[i + 1] * A[i + 1];
这样一写,for循环进行计算的时候就会不写错边界和初始化条件,注意因为不能使用除法,所以L要从0开始往下计算,R要从下往上进行计算。
接下来考虑优化,只用一个vector就可以了,先将左边的遍历一遍,然后从右下角开始直接相乘计算得到结果。
class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        if(A.empty()){
            return {};
        }
        vector<int> res(A.size(),1);        
        int tmp = 1;        
        for(int i = 1;i < A.size();++i){ 
            tmp *= A[i - 1];
            res[i] = tmp;
            
        }
        tmp = 1;
        for(int i = A.size() - 2;i >= 0;--i){ 
            tmp *= A[i + 1];
            res[i] *= tmp;            
        }        
        return res;
    }
};

 

 
 
 
 
 
class Solution {
public:
    vector<int> multiply(const vector<int>& A) {
        if(A.empty()){
            return {};
        }
        vector<int> res(A.size());
        vector<int> l(A.size(),1),r(A.size(),1);
        int ls = 1,rs = 1;
        //l.push_back(1);
        for(int i = 1;i < A.size();++i){ 
            ls *= A[i - 1];
            l[i] = ls;
            
        }
        for(int i = A.size() - 2;i >= 0;--i){ 
            rs *= A[i + 1];
            r[i] = rs;            
        }
        for(int i = 0;i < A.size();++i){ 
           res[i] = l[i] * r[i];           
        }
        return res;
    }
};

 

48 构建乘积数组

标签:优化   height   for循环   除法   gpo   题目   otto   ++   multi   

原文地址:https://www.cnblogs.com/dingxiaoqiang/p/8277935.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!