51构建乘积数组

题目描述

给定一个数组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] = A[1] * A[2] * … * A[n-1],B[n-1] = A[0] * A[1] * … * A[n-2];)

思路分析

  • 思路一:暴力解,两次循环,复杂度O(N^2)。
  • 思路二:B[i]值能看作是矩阵对角线上的值,相当于计算上下三角,然后将两部分相乘。
  • 为求简单,也能用两个数组存储上下三角,最后遍历相乘。
    借用牛客上剑指书的图
    在这里插入图片描述

代码实现

    public int[] multiply(int[] A) {
        int[] B=new int[A.length];
        B[0]=1;
        //从上往下计算下三角
        for (int i = 1; i <A.length; i++) {
            B[i]=B[i-1]*A[i-1];
        }
        //从下往上计算上三角
        int tmp=1;
        for (int i = A.length-2; i >=0 ; i--) {
            tmp*=A[i+1];
            B[i]*=tmp;
        }

        return B;
    }
发布了118 篇原创文章 · 获赞 8 · 访问量 3717

猜你喜欢

转载自blog.csdn.net/qq_34761012/article/details/104473650