Leetcode之javascript解題(No33-34)

遊幕發表於2018-04-13
LeetCodeJavaScript

附上我的github倉庫，會不斷更新leetcode解題答案，提供一個思路，大家共勉

在我的主頁和github上可以看到更多的關於leetcode的解題報告！（因為不知道為什麼掘金沒有將其釋出出來，目前已經聯絡掘金客服）

希望可以給star，鼓勵繼續更新解題思路

author: thomas

Leetcode之javascript解題(No33-34)

No34：Search for Range(Medium)

題目

Given an array of integers sorted in ascending order, find the starting and ending position of a given target value.

Your algorithm's runtime complexity must be in the order of O(log n).

If the target is not found in the array, return [-1, -1].

For example

Given [5, 7, 7, 8, 8, 10] and target value 8,
return [3, 4]. // 下標3，4是陣列8
複製程式碼

這道題讓我們在一個有序整數陣列中尋找相同目標值的起始和結束位置,沒如果沒有找到就返回[-1,-1]

思路

這道題讓我們在一個有序整數陣列中尋找相同目標值的起始和結束位置，而且限定了時間複雜度為O(logn)，這是典型的二分查詢法的時間複雜度，所以這道題我們也需要用此方法。

  • 方法一：

    • 我們的思路是對原陣列使用兩次二分查詢法，分別找出一個起始和結束位置
  • 方法二：利用二分法找到起始位置，然後向右遍歷，找到邊界

程式碼

  • 方法一
let arr1 = [1,1,2,2,3,4,4,7,8];
let arr = [5,7,7,8,8,10],
  target = 8;
let searchRange = function(arr, target) {
  let len = arr.length,
    res = [-1, -1];
  for (let i = 0, j = len-1; i <= j;) {
    let mid = Math.floor((i + j) / 2);
    if (arr[mid] < target) { // 先判斷小於target的情況
      i = mid + 1;
    }else {
      j = mid - 1; // 應對剛好i = mid + 1後就指向了target值
      if (arr[mid] === target) {
        res[0] = mid; // 得到起始index
      }
    }
  }

  for (let i = 0, j = len-1; i <= j;) {
    let mid = Math.floor((i + j) / 2);
    if (arr[mid] > target) {// 先判斷大於target的情況
      j = mid - 1;
    }else {
      i = mid + 1; // 應對剛好i = mid + 1後就指向了target值
      if (arr[mid] === target) {
        res[1] = mid; // 得到結束index
      }
    }
  }

  return res;
};
console.log(searchRange(arr,target)); // [3, 4]
複製程式碼
  • 方法二
/**
 * 方法2
 *
 * 找到res[0]之後，就向右遍歷，直到不是該值，就可以得到右邊界了
 * 時間複雜度沒上面的方法好
 */

let searchRange1 = function(arr, target) {
  let len = arr.length,
    res = [-1, -1];
  for (let i = 0, j = len-1; i <= j;) {
    let mid = Math.floor((i + j) / 2);
    if (arr[mid] < target) {
      i = mid + 1;
    }else {
      j = mid - 1; // 應對剛好i = mid + 1後就指向了target值
      if (arr[mid] === target) {
        res[0] = mid; // 得到最左邊的值
      }
    }
  }
  let k;
  res[1] = res[0];
  for (k = res[0] + 1; k < len; k++) { // 找到右邊界
    if (arr[k] === target) {
      res[1] += 1;
    }
  }

  return res;
};
console.log(searchRange1([1],1)); // [0, 0]
console.log(searchRange1([2,2],2)); // [0, 1]
console.log(searchRange1([5,7,7,8,8,10],8)); // [3, 4]
console.log(searchRange1([1,3],1)); // [0, 0]
console.log(searchRange1([3,3,3],3)); // [0, 0]
複製程式碼

注：二分法：其假設我們找到目標值（但是有多個目標值連在一起）

  • 1、如果我們先判斷arr[mid] < target(先判斷小於target的情況),再判斷剩下的,最後得到的結果就是要找的多個目標值target的最左邊的值
  • 2、如果我們先判斷arr[mid] > target(也就是先判斷大於target的情況),再判斷剩下的,最後得到的結果就是要找的多個目標值target的最右邊的值

No35：Search Insert Position(Easy)

題目

從給定排好順序的陣列，找出給定目標數字下標，存在則返回下標，不存在則返回目標數應該插入的位置下標。 Example 1:

Input: [1,3,5,6], 5
Output: 2
複製程式碼

Example 2:

Input: [1,3,5,6], 2
Output: 1
複製程式碼

Example 3:

Input: [1,3,5,6], 7
Output: 4
複製程式碼

Example 4:

Input: [1,3,5,6], 0
Output: 0
複製程式碼

思路

思路就是每次取中間，如果等於目標即返回，否則根據大小關係切去一半。因此演算法複雜度是O(logn)，空間複雜度O(1)

程式碼

let arr = [1,3,5,6],
	target = 5;

let searchInset = function(arr, target) {
  let len = arr.length,
			res = 0;
  for (let i = 0, j = len -1; i <= j;) {
    let mid = Math.floor((i+j)/2);
  if (arr[mid] === target) {
			return mid;
  }
  if (arr[mid] < target) {
      i = mid + 1;
      res = mid+1; // 更新res
		}else {
    	j = mid - 1;
		}
	}
	return res; //
}
console.log(searchInset(arr,target)); // 2
console.log(searchInset([1,3,5,6],2)); // 2
複製程式碼

注意:二分法有一個好處:就是當迴圈結束時:

(1)如果找到目標元素，那麼就返回當前位置

(2)如果沒有找到目標元素，那麼i一定停在恰好比目標大的index上，j一定停在恰好比目標小的index上，所以個人比較推薦這種實現方式。(初始i在左，j在右)

相關文章