Oto moja próba rozwiązania tego w Javie. Przepraszamy za niestandardowe formatowanie, ale kod ma wiele powielania i jest to najlepsze, co mogę zrobić, aby był czytelny.
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]"
}
}
Niezmiennikami pętli są pi = nums[0] * nums[1] *.. nums[i-1]
i pj = nums[N-1] * nums[N-2] *.. nums[j+1]
. i
Część po lewej stronie jest „prefix” logika, a j
część po prawej stronie jest „sufiks” logika.
Jednowarstwowa rekurencyjna
Jasmeet dał (piękne!) Rekurencyjne rozwiązanie; Zamieniłem go w tę (ohydną!) Jednowierszową Javę. Dokonuje modyfikacji na miejscu , z O(N)
tymczasowym miejscem na stosie.
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]"