題目

給你一個數(shù)組??和一個值?，你需要?原地?移除所有數(shù)值等于??的元素，并返回移除后數(shù)組的新長度。

不要使用額外的數(shù)組空間，你必須僅使用??額外空間并?原地?修改輸入數(shù)組。

元素的順序可以改變。你不需要考慮數(shù)組中超出新長度后面的元素。

解題

def remove_element(nums, val):# 初始化新數(shù)組索引new_index = 0for current_index in range(len(nums)):if nums[current_index] != val:nums[new_index] = nums[current_index]new_index += 1return new_index# 測試例子
nums = [3, 2, 2, 3]
value = 3
new_length = remove_element(nums, value)
print("新數(shù)組的長度：", new_length)
print("移除后的數(shù)組：", nums[:new_length])

新數(shù)組的長度： 2
移除后的數(shù)組： [2, 2]