1. 程式人生 > >移除元素(leetcode簡單篇二十七題)

移除元素(leetcode簡單篇二十七題)

給定一個數組 nums 和一個值 val,你需要原地移除所有數值等於 val 的元素,返回移除後陣列的新長度。

不要使用額外的陣列空間,你必須在原地修改輸入陣列並在使用 O(1) 額外空間的條件下完成。

元素的順序可以改變。你不需要考慮陣列中超出新長度後面的元素。

示例 1:

給定 nums = [3,2,2,3], val = 3,

函式應該返回新的長度 2, 並且 nums 中的前兩個元素均為 2。

你不需要考慮陣列中超出新長度後面的元素。

示例 2:

給定 nums = [0,1,2,2,3,0,4,2], val = 2,

函式應該返回新的長度 5, 並且 nums 中的前五個元素為 0, 1, 3, 0, 4。

注意這五個元素可為任意順序。

你不需要考慮陣列中超出新長度後面的元素

筆者拿到這道題時開始的思路是把遇到的第一個與val相等的元素直接丟到陣列的最後(題目裡說元素順序可以改變),讓陣列長度減一下(其實就是雖然元素還在數組裡,但不列印他),如果交換到前面的這個元素也與val相等就讓迴圈變數也–一次。筆者程式碼附上

int removeElement(int* nums, int numsSize, int val)//刪除陣列的重複元素
{
	int i = 0;
	int tmp = 0;
	for
(i = 0; i < numsSize; i++) { if(nums[i] != val)//如果不相等就跳出這次迴圈 { continue; } if(nums[i] == val)//相等與陣列最後一個元素交換 { tmp = nums[i]; nums[i] = nums[numsSize-1]; nums[numsSize-1] = tmp; if(nums[numsSize-1] == val)//如果最後一個數字也等於val讓迴圈i-- { i--; } numsSize--; } } return numsSize;
}

方法二(參考了官網的方法,還是應用了快指標與慢指標)

int removeElement(int* nums, int numsSize, int val)//哈哈,好像真的賊簡潔不過與筆者的執行速度沒差異
{
	int i = 0;
	int j = 0;
	for(j = 0; j < numsSize; j++)
	{
		if(nums[j] != val)
		{
			nums[i] = nums[j];
			i++;
		}
	}
	return i;
}