LeetCode - 238. 除自身以外数组的乘积

2019-03-03  本文已影响0人  安小辰

题目

给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。

进阶:

你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)

解答

思路:

1.用两个数组来保存数组,一个从左向右把前面的所有元素相乘,另一个从右向左把后面的所有元素相乘;
2.再把两个数组中对应角标的元素相乘,即得到除自身以外数组中其它元素的乘积。

代码:

  /**
 * 除自身以外数组的乘积
 * @param nums : 输入的数组。
 * @return : 返回新的数组,每一个元素是输入数组中除了当前序号的其它所有数的乘积。
 */
public int[] productExceptSelf(int[] nums) {
    int len = nums.length;
    //用两个数组来保存结果,一个记录从左到右的所有元素相乘的结果,另一个记录从右向左的。
    int[] left = new int[len];
    int[] right = new int[len];
    int tempLeft = 1;
    int tempRight = 1;
    for (int i = 0; i < len; i++) {
        left[i] = tempLeft;
        tempLeft *= nums[i];
        right[len - 1 - i] = tempRight;
        tempRight *= nums[len - 1 - i];
    }
    int[] result = new int[len];
    //把从和右的结果再相乘就得到最终的结果。
    for (int i = 0; i < len; i++) {
        result[i] = left[i] * right[len - 1 - i];
    }
    return result;
}
上一篇下一篇

猜你喜欢

热点阅读