除自身以外的数组乘积
前言
学会认真观察思考问题本身的个性,结合积累的知识完成解题甚至低时空复杂度解题。
一、除自身以外的数组乘积
给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。
题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
二、题解
1)可以先用两个数组存入每个数的前缀积和后缀积,然后再一次前缀积后缀积对应相乘。
2)遍历数组的前一部分,将返回数组左边存入前缀积,右边存入后缀积;接下来遍历数组后一部分,将返回数组的右边乘上前缀积,左边乘上后缀积,这样就实现了前后缀积相乘。
package com.xhu.offer.tencent;
import com.xhu.alg.NumberOfArithmeticSlices446;
public class ProductExceptSelf {
public int[] productExceptSelf(int[] nums) {
//用额外的数组空间存前缀积和后缀积
int[] prefix = new int[nums.length];
int[] suffix = new int[nums.length];
prefix[0] = 1;
suffix[nums.length - 1] = 1;
for (int i = 0, j = nums.length - 1; j > 0; i++, j--) {
prefix[i + 1] = prefix[i] * nums[i];
suffix[j - 1] = suffix[j] * nums[j];
}
int[] res = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
res[i] = prefix[i] * suffix[i];
}
return res;
}
//进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
public int[] productExceptSelf2(int[] nums) {
//双指针,同时记录左缀积和右缀积。
int len = nums.length;
int[] res = new int[len];
for (int i = 0; i < len; i++) res[i] = 1;
int prefix = 1, suffix = 1;
//左边的先存入左缀积,右边的先存入右缀积。当过半时,右边存入的后缀积开始乘积累的prefix,左边存入的前缀积开始乘积累的suffix
for (int i = 0; i < len; i++) {
//先存入左缀积
res[i] *= prefix;
prefix *= nums[i];
//存入右缀积
res[len - 1 - i] *= suffix;
suffix *= nums[len - 1 - i];
}
return res;
}
}
总结
1)当解决不了问题时,试试开辟额外的空间来解决;当问题解决了,试试利用题目潜在有利条件来降低时空复杂度。
参考文献
[1]LeetCode 原题