剑指 Offer 66. 构建乘积数组

 题目如上。思路如下：

我们假设应该返回的答案为数组ans。显而易见ans的数组长度应等于入参a数组的长度。

按照题意，对应于ans数组中的每一个元素ans[i],其大小应为：

\[ans[i] = a[0] \times a[1] \times  \cdots  \times a[i - 1] \times a[i + 1] \times  \cdots  \times a[length - 1]\]

对于上式，我们可以把\[ans[i]\]分为两部分相乘，即

\[\left\{ \begin{array}{l}
ans[i] = C[i] \times D[i]\\
C[i] = a[0] \times a[1] \times  \cdots  \times a[i - 1]\\
D[i] = a[i + 1] \times  \cdots  \times a[length - 1]
\end{array} \right.\]

其中C和D两个数组我们可以不用另外开辟空间，刚好有入参a数组和我们新建的返回数组ans能满足运算要求。

易知\[\left\{ \begin{array}{l}
C[i] = C[i - 1] \times a[i - 1]\\
D[i] = D[i + 1] \times a[i + 1]
\end{array} \right.\]

值得注意得是需要给C[0]赋初始值为1，即ans[0]=1

我们可以首先利用ans的空间将C数组建立起来，根据上述的公式可得如下代码：

for (int i = 1; i < a.length; i++) {
        ans[i] = ans[i - 1] * a[i - 1];
     }

同理可以得到D数组的值，但是为了节省空间，我们用一个tmp去逆序存储D[i]，逆序的目的是为了方便与C[i] 相乘，得到ans[i]。一样也需要给tmp赋予初始值1.得到D[i]并求解ans[i]代码如下：

for (int i = a.length - 2; i >=0; i--) {
            tmp *= a[i + 1];
            ans[i] *= tmp;
        }

完整代码如下：

class Solution {
    public int[] constructArr(int[] a) {
        if (a == null || a.length < 1) return new int[0];
        int[] ans = new int[a.length];
        ans[0] = 1;
        for (int i = 1; i < a.length; i++) {
            ans[i] = ans[i - 1] * a[i - 1];
        }       
        int tmp = 1;
        for (int i = a.length - 2; i >=0; i--) {
            tmp *= a[i + 1];
            ans[i] *= tmp;
        }
        return ans;
    }
}