LeetCode 练习——剑指 Offer 66. 构建乘积数组

1.题目描述

剑指 Offer 66. 构建乘积数组

给定一个数组 A[0,1,…,n-1],请构建一个数组 B[0,1,…,n-1],其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积, 即 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]

2.解题思路与代码

2.1 解题思路

题目要求得到数组除当前位以外的数的乘积,首先想到的是求数组所有数的乘积,然后依次除以当前位,但是这样的方法如果遇到 0 就会报错了。这里就可以使用前缀和数组的思想,先从左往右遍历得到当前位左边的数的乘积,然后在从右往左遍历得到当前数右边所有位的乘积,最后将当前数左右两边的乘积相乘便能够得到最终结果。以 [1, 2, 3, 4, 5] 为例进行详解
首先我们准备一个数组存放数左边所有数的乘积。让第一位为 1,然后从左往右便利,依次得到当前数左边的乘积


然后再准备一个数组存放数右边所有数的乘积。设置数组最后一位为 1,然后从右往左进行遍历,依次得到当前数右边所有数的乘积


最后将两个数组依次按位相乘便得到最终的结果


2.2 代码

class Solution {
    public int[] constructArr(int[] a) {
        int n = a.length;
        int[] ans = new int[n];
        if (n == 0) {
            return ans;
        }
        int[] left = new int[n];
        int[] right = new int[n];
        left[0] = 1;
        for (int i = 1; i < n; i++) {
            left[i] = a[i - 1] * left[i - 1];
        }
        right[n - 1] = 1;
        for (int i = n - 2; i >= 0; i--) {
            right[i] = a[i + 1] * right[i + 1];
        }
        for (int i = 0; i < n; i++) {
            ans[i] = left[i] * right[i];
        }
        return ans;

    }
}

2.3 测试结果

通过测试


测试结果

3.总结

  • 使用前缀和数组思想获取左边所有数乘积和右边所有数乘积
  • 将左右两边乘积再依次相乘便得到最终结果

你可能感兴趣的:(LeetCode 练习——剑指 Offer 66. 构建乘积数组)