程式碼隨想錄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;
}
}