剑指offer题目练习
题目描述:给定一个数组A[0,1,…,n-1],请构建一个数组B[0,1,…,n-1],其中B中的元素B[i]=A[0]A[1]…*A[i-1]A[i+1]…*A[n-1]。不能使用除法。(注意:规定B[0] = A[1] * A[2] * … * A[n-1],B[n-1] = A[0] * A[1] * … * A[n-2];)。
对于A长度为1的情况,B无意义,故而无法构建,因此该情况不会存在。
题目解析:数组B的开头和结尾已经规定好了,然后就是数组中间的值,观察上面的规律A[i-1]*A[i+1]可以看出,中间第i个数值的计算方法就是数组A中所有数的连乘积除了A[i]。
下面是我的代码,由于刚开始写编程题,也没考虑内存和时间。
import java.util.ArrayList;
public class Solution {
public int[] multiply(int[] A) {
int A_length=A.length;
int[] B=new int[A_length];
for(int i=0;i<A_length;i++){
if(i==0){
B[i]=1;
for(int j=0;j<A_length;j++){
B[i]=A[j]*B[i];
}
}
if(i==A_length-1){
B[i]=1;
for(int j=0;j<A_length-1;j++){
B[i]=A[j]*B[i];
}
}
if(i==A_length-2){
B[i]=1;
for(int j=0;j<A_length-2;j++){
B[i]=A[j]*B[i];
}
B[i]=B[i]*A[A_length-1];
}
if(i!=0&&i!=A_length-1&&i!=A_length-1){
B[i]=1;
for(int j=0;j<A_length;j++){
if(j==i){
j++;
}
B[i]=B[i]*A[j];
}}
}
return B;
}
}
第三个if和第四个if都是中间值计算,本来是写在一起的,但是发现B[N-2]跳过时会越界,所以分开来结算。
剑指offer的思路:
数组B中的每个元素可由矩阵每行的元素累积,因此可以先计算矩阵中的下三角,先用下三角的每行表示数组B中的元素,再计算上三角,将上三角的每行值与下三角对应乘积就得到完整的数值。
public class Solution {
public int[] multiply(int[] A) {
int length = A.length;
int[] B = new int[length];
if(length != 0 ){
B[0] = 1;
//计算下三角连乘
for(int i = 1; i < length; i++){
B[i] = B[i-1] * A[i-1];
}
int temp = 1;
//计算上三角
for(int j = length-2; j >= 0; j--){
temp *= A[j+1];
B[j] *= temp;
}
}
return B;
}
}