Python LeetCode(35.挿入位置検索)

4504 ワード

Python LeetCode(35.挿入位置検索)
ソート配列とターゲット値を指定し、配列内にターゲット値を見つけ、インデックスを返します.ターゲット値が配列に存在しない場合は、順番に挿入される位置を返します.
配列に重複要素がないと仮定できます.
例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
Solution:(targetがarrayにある場合は対応するindexを直接返します;いない場合はarrayに追加し、ソートしてからindexを返します)
class Solution(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        if target in nums:
            return nums.index(target)
        
        nums.append(target)
        for i in range(len(nums)-1):
            min_index = i
            for j in range(i+1, len(nums)):
                if nums[j] < nums[min_index]:
                    min_index = j
            nums[i], nums[min_index] = nums[min_index], nums[i]
        return nums.index(target)
solution = Solution()
print(solution.searchInsert([1, 3, 5, 6], 2))
1