Commit41dfdb7

authored

Merge pull requestneetcode-gh#3198 from doublewhy/patch-2

Update 0027-remove-element.py

2 parents14daefb +02c2b65 commit41dfdb7Copy full SHA for 41dfdb7

File tree

+17

-0

lines changed

+17

-0

lines changed

Lines changed: 17 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -6,3 +6,20 @@ def removeElement(self, nums: List[int], val: int) -> int:`
`6`	`6`	`nums[k]=nums[i]`
`7`	`7`	`k+=1`
`8`	`8`	`returnk`
	`9`	`+`
	`10`	`+# Optimized solution with the same time and space complexity`
	`11`	`+classSolution:`
	`12`	`+defremoveElement(self,nums:List[int],val:int)->int:`
	`13`	`+# Avoid unessary copy operations in a previous solution, when k == i and nums[i] != val`
	`14`	`+# by swapping nums[i] and the last element of the array (nums[n])`
	`15`	`+n=len(nums)`
	`16`	`+i=0`
	`17`	`+`
	`18`	`+whilei<n:`
	`19`	`+ifnums[i]==val:`
	`20`	`+nums[i],nums[n-1]=nums[n-1],nums[i]`
	`21`	`+n-=1# decrement the length of the array by discarding the last element`
	`22`	`+else:`
	`23`	`+i+=1`
	`24`	`+`
	`25`	`+returnn`

Comments

(0)