Commit 4f97d39

committed

Optimize memory complexity

1 parent f3bd95f commit 4f97d39Copy full SHA for 4f97d39

File tree

-5

lines changed

-5

lines changed

Lines changed: 8 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -7,16 +7,19 @@ def product_except_self(nums: list[int]) -> list[int]:`
`7`	`7`	`"""`
`8`	`8`
`9`	`9`	`length = len(nums)`
`10`		`- left_prefix = [1] * length`
`11`		`- right_prefix = [1] * length`
	`10`	`+ left = 1`
	`11`	`+ right = 1`
	`12`	`+ output = [1] * length`
`12`	`13`
`13`	`14`	`for index in range(length - 1):`
`14`		`- left_prefix[index + 1] = nums[index] left_prefix[index]`
	`15`	`+ left *= nums[index]`
	`16`	`+ output[index + 1] *= left`
`15`	`17`
`16`	`18`	`for index in range(length - 1, 0, -1):`
`17`		`- right_prefix[index - 1] = nums[index] right_prefix[index]`
	`19`	`+ right *= nums[index]`
	`20`	`+ output[index - 1] *= right`
`18`	`21`
`19`		`- return [l*rforl, rinzip(left_prefix, right_prefix)]`
	`22`	`+ return output`
`20`	`23`
`21`	`24`
`22`	`25`	`assert product_except_self([1, 2, 3, 4]) == [24, 12, 8, 6], 'Test 1 failed'`

Comments

(0)