这是我用Java解决的尝试。对于非标准格式,我们深表歉意,但是代码中有很多重复项,这是我能做到的最好的方法。
import java.util.Arrays;
public class Products {
static int[] products(int... nums) {
final int N = nums.length;
int[] prods = new int[N];
Arrays.fill(prods, 1);
for (int
i = 0, pi = 1 , j = N-1, pj = 1 ;
(i < N) && (j >= 0) ;
pi *= nums[i++] , pj *= nums[j--] )
{
prods[i] *= pi ; prods[j] *= pj ;
}
return prods;
}
public static void main(String[] args) {
System.out.println(
Arrays.toString(products(1, 2, 3, 4, 5))
); // prints "[120, 60, 40, 30, 24]"
}
}
循环不变式为pi = nums[0] * nums[1] *.. nums[i-1]
和pj = nums[N-1] * nums[N-2] *.. nums[j+1]
。i
左边的部分是“前缀”逻辑,j
右边的部分是“后缀”逻辑。
递归一线
Jasmeet提供了一个(漂亮的!)递归解决方案;我已经将它变成了这种(令人发指的!)Java单行代码。它在O(N)
堆栈中使用临时空间进行就地修改。
static int multiply(int[] nums, int p, int n) {
return (n == nums.length) ? 1
: nums[n] * (p = multiply(nums, nums[n] * (nums[n] = p), n + 1))
+ 0*(nums[n] *= p);
}
int[] arr = {1,2,3,4,5};
multiply(arr, 1, 0);
System.out.println(Arrays.toString(arr));
// prints "[120, 60, 40, 30, 24]"