搜索插入位置(Java實現)

搜索插入位置(Java實現)

給定一個排序數組和一個目標值,在數組中找到目標值,並返回其索引。如果目標值不存在於數組中,返回它將會被按順序插入的位置。
你可以假設數組中無重複元素。

示例 1:
輸入: [1,3,5,6], 5
輸出: 2

示例 2:
輸入: [1,3,5,6], 2
輸出: 1

示例 3:
輸入: [1,3,5,6], 7
輸出: 4

示例 4:
輸入: [1,3,5,6], 0
輸出: 0

我的思路很簡單,就是先遍歷查找,藉助數據結構ArrayList,如果找到就返回下標;找不到就add進ArrayList中,然後對ArrayList排序,使用indexOf(target)來返回target的下標即可。
代碼實現之:

package Day49;

import java.util.ArrayList;
import java.util.Collections;

/**
 * @Author Zhongger
 * @Description 給定一個排序數組和一個目標值,在數組中找到目標值,並返回其索引。如果目標值不存在於數組中,返回它將會被按順序插入的位置。
 * 你可以假設數組中無重複元素。
 * @Date 2020.3.23
 */
public class SearchInsertSolution {
    public static void main(String[] args) {
        int[] nums={1,3,5,6};
        int target=7;
        System.out.println(new SearchInsertSolution().searchInsert(nums, target));
    }
    public int searchInsert(int[] nums, int target) {
        if (nums==null){
            return -1;
        }
        if (nums.length<=0){
            return -1;
        }
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            list.add(nums[i]);
            if (target==nums[i]){
                return i;
            }
        }
        list.add(target);
        Collections.sort(list);
        return list.indexOf(target);
    }
}

還有一種方法就是二分查找法,簡單得很:

 	public int searchInsert2(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        while(left <= right) {
            int mid = (left + right) / 2;
            if(nums[mid] == target) {
                return mid;
            } else if(nums[mid] < target) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
        return left;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章