1. 程式人生 > >LintCode 50. 數組剔除元素後的乘積

LintCode 50. 數組剔除元素後的乘積

clas and turn long -- long long () style vector

題目:

給定一個整數數組A。

定義B[i] = A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1], 計算B的時候請不要使用除法。

樣例

給出A=[1, 2, 3],返回 B為[6, 3, 2]

解:看樣例,B[0]=A[1]*A[2],B[1]=A[0]*A[2],B[2]=A[0]*A[1];

代碼中left保存左邊的值,

right保存右邊的值

class Solution {
public:
    /*
     * @param nums: Given an integers array A
     * @return: A long long array B and B[i]= A[0] * ... * A[i-1] * A[i+1] * ... * A[n-1]
     
*/ vector<long long> productExcludeItself(vector<int> &nums) { // write your code here int n = nums.size(),i; vector<long long> left(n, 1); vector<long long> right(n, 1); vector<long long> res(n, 1); for(i = 1; i < n; ++i) left[i]
= left[i-1] * nums[i-1]; for(i = n-2; i >= 0; --i) right[i] = right[i+1] * nums[i+1]; for(i = 0; i != nums.size(); ++i) res[i] = left[i] * right[i]; return res; } };

LintCode 50. 數組剔除元素後的乘積