数组和字符串 -- 搜索插入位置

1、搜索插入位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4

2、解法

from typing import List


class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:

        # for循环
        # 因为涉及到元素、下标，所以我们使用 enum 遍历序列
        for i,j in enumerate(nums):
            if j == target:
                return i
        nums.append(target)  # 如果数组中没有这个元素，则没有这个元素的下标，我们插入后排序，继续返回下标即可
        nums.sort()
        for i,j in enumerate(nums):
            if j == target:
                return i
        
        # 第一次 优化重复的代码
        try:
            nums.index(target)
        except:
            nums.append(target)  # 如果数组中没有这个元素，则没有这个元素的下标，我们插入后排序，继续返回下标即可
            nums.sort()
        finally:
            for i, j in enumerate(nums):
                if j == target:
                    print(i)
                    return i
        
        # 第二次优化重复的代码
        # 使用 try except finally 优化重复的代码，这里的index会抛出异常，如果元素不在列表中的话
        try:
            nums.index(target)
        except:
            nums.append(target)
            nums.sort()
        finally:
            return nums.index(target)




class TestSolution:

    def test_three(self):
        b = Solution()
        b.searchInsert(nums = [1,3,5,6], target = 5)

    def test_two(self):
        b = Solution()
        b.searchInsert(nums = [1,3,5,6], target = 2)

    def test_two1(self):
        b = Solution()
        b.searchInsert(nums = [1,3,5,6], target = 7)

3、思路

（1）for循环，因为涉及到元素、下标，所以我们使用 enum 遍历序列
（2）如果数组中没有这个元素，则没有这个元素的下标，我们插入后排序，继续返回下标即可
（3）使用 try except finally 优化重复的代码，这里的index会抛出异常，如果元素不在列表中的话，这里注意:index()方法，如果找不到就会抛出异常