程式碼隨想錄DAY1 | 二分,雙指標移除元素

Wylie_Leslie發表於2024-04-04

程式碼隨想錄DAY1 | 二分,雙指標移除元素


題目描述

給定一個 n 個元素有序的(升序)整型陣列 nums 和一個目標值 target ,寫一個函式搜尋 nums 中的 target,如果目標值存在返回下標,否則返回 -1

示例 1:

輸入: nums = [-1,0,3,5,9,12], target = 9
輸出: 4
解釋: 9 出現在 nums 中並且下標為 4

示例 2:

輸入: nums = [-1,0,3,5,9,12], target = 2
輸出: -1
解釋: 2 不存在 nums 中因此返回 -1

之前有寫過c語言版本,這次寫java版本

注意:由於個人習慣,一般我寫的都是兩邊都為開區間版本

程式碼
class Solution {
    public int search(int[] nums, int target) {
        if(target < nums[0] || target > nums[nums.length - 1]){
            return -1;
        }
        int left = 0,right = nums.length -1;
        while(left <= right){
            int mid = left +(right -  left) / 2;//防止溢位
            if (nums [mid] ==  target)
                return mid;
            else if (nums [mid] < target)
                left = mid + 1;
            else if (nums[mid] > target)
                right = mid - 1;
        }
        return -1;
    }
}
/*
	– 程式碼注意mid的運算的溢位檢測,然後還可以最佳化的話,可以把
	    int mid = left + (right -  left) / 2;//防止溢位
	  最佳化成 int mid = left + (right - left)>>2;
	  
	  擊敗率會從11.04%到39.04%

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

不要使用額外的陣列空間,你必須僅使用 O(1) 額外空間並 原地 修改輸入陣列

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

說明:

為什麼返回數值是整數,但輸出的答案是陣列呢?

請注意,輸入陣列是以「引用」方式傳遞的,這意味著在函式里修改輸入陣列對於呼叫者是可見的。

你可以想象內部操作如下:

// nums 是以“引用”方式傳遞的。也就是說,不對實參作任何複製
int len = removeElement(nums, val);

// 在函式里修改輸入陣列對於呼叫者是可見的。
// 根據你的函式返回的長度, 它會列印出陣列中 該長度範圍內 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

示例 1:

輸入:nums = [3,2,2,3], val = 3
輸出:2, nums = [2,2]
解釋:函式應該返回新的長度 2, 並且 nums 中的前兩個元素均為 2。你不需要考慮陣列中超出新長度後面的元素。例如,函式返回的新長度為 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也會被視作正確答案。

示例 2:

輸入:nums = [0,1,2,2,3,0,4,2], val = 2
輸出:5, nums = [0,1,3,0,4]
解釋:函式應該返回新的長度 5, 並且 nums 中的前五個元素為 0, 1, 3, 0, 4。注意這五個元素可為任意順序。你不需要考慮陣列中超出新長度後面的元素。

這題在沒看題解之前第一反應就是暴力查詢,,,好久沒刷題了,忘記了雙指標法,看了題解之後才想起來雙指標法

寫雙指標的時候莫名想起來KMP演算法

程式碼
暴力寫法
---
class Solution {
	public int removeElement(int[] nums, int val){
		int size = nums.length;
		for(int i = 0;i < size ; i++){
			if(nums[i] == val ){
				for(int j= i+1;j < size ;j++){
					nums[j-1] = nums[j];
				}
				i--;
				size--;
			}
		}
		return size;
	}
}

雙指標法
---
/*
	快指標:尋找新陣列的元素,也就是不含目標元素的陣列
	慢陣列:指向更新,新陣列的下標位置
	
	class Solution {
    public int removeElement(int[] nums, int val) {
        // 快慢指標
        int slowIndex = 0;
        for (int fastIndex = 0; fastIndex < nums.length; fastIndex++) {
            if (nums[fastIndex] != val) {
                nums[slowIndex] = nums[fastIndex];
                slowIndex++;
            }
        }
        return slowIndex;
    }
}

相關文章