LeetCode OJ 系列之35 Search Insert Position --Python

最新推荐文章于 2017-04-12 10:48:00 发布

阅读量461

点赞数

分类专栏： LeetCode Python 文章标签： LeetCode Python OJ

Python 同时被 2 个专栏收录

26 篇文章

订阅专栏

LeetCode

22 篇文章

订阅专栏

本文介绍了一个算法问题：在一个已排序的数组中找到目标值的插入位置。如果目标值存在于数组中，则返回其索引；若不存在，则返回按顺序插入该值后的索引位置。示例包括针对不同输入情况的解决方案。

Problem:

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You may assume no duplicates in the array.

Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

Answer:

class Solution(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        for index in range(len(nums)):
        	if nums[index] > target:
        		return index
        	if nums[index] == target:
        		return index
        return len(nums)