LeetCode (238):Product of Array Except Self

Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i].

Solve it without division and in O(n).

For example, given [1,2,3,4], return [24,12,8,6].

Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)

[思路]:刚看到题时我想到的是用两重循环直接求解,这样实现很简单。

java代码:

public class Solution {
     public static int[] productExceptSelf(int[] nums) {
         int [] product=new int[nums.length];    
         for(int i=0;i<nums.length;i++){
             product[i]=1;
             for(int j=0;j<nums.length;j++){
                 if(j!=i)
                 product[i]*=nums[j];
             }
         }
            return product;
        }
}

测试结果正确,提交上去毫无意外地超时了。

继续思考,对于一个给定的数组nums[1,2,3,4],我们用一个res[4]的数组来存放结果,res中存放的数据应是

res[0]=2*3*4

res[1]=1*3*4

res[2]=1*2*4

res[3]=1*2*3

这样我们可以发现一些规律:

1.数组的两端直接就是正确的结果

2.两个数组的中间部分相乘就是正确结果

java代码:

public class Solution {
    public int[] productExceptSelf(int[] nums) {
      int[] res=new int[nums.length];
        res[res.length-1]=1;
        for(int i=nums.length-2;i>=0;i--){
            res[i]=res[i+1]*nums[i+1];
        }
        int[] right=new int[nums.length];
        right[0]=1;
        for(int i=1;i<nums.length;i++){
            right[i]=right[i-1]*nums[i-1];
            res[i]*=right[i];
        }        
         return res;
    }
}

 

posted @ 2015-10-18 13:34  六层楼  阅读(711)  评论(0编辑  收藏  举报