Leetcode练习<十三> 判断数据的插入位置

'''
@greg 20170609
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
给定一个排序好的列表和整数，如果整数在列表中存在，返回与其相等的整数的索引
如果不相等，返回这个数组插入这个列表中的索引（即把这个整数插入这个列表中，索引值是什么？）
'''

'''
我的思路很简单：先判断是否有相等的，相等就返回这个数的索引，如果没有相等的，那就接着去比较大小，因为原数组是排序之后的
如果有一个数字a比他大，那a之后的数字就都比他大，返回a的索引即可
如果都不如target大，那就返回列表的长度，也就是target插入列表的索引值
'''
class Solution(object):
    def searchInsert(self, nums, target):  # %8
        for count, num in enumerate(nums):
            if target == num:
                return count

        for count, num in enumerate(nums):
            if target < num:
                return count

        return count + 1
# 大神１的思路：只要大于等于target, 就返回num的索引，否则就返回列表的长度，是我那个思路的缩减版，简单明了

    def searchInsert1(self, nums, target):  # %62
        for num in nums:
            if num >= target:
                return nums.index(num)
        return len(nums)
# 大神２的思路：相当于我新建一个列表把所有小于target的num放到新的列表中，如果不小于了，就不放了，最后返回列表的长度
    def searchInsert2(self, nums, target):  # %24
        return len([x for x in nums if x < target])

if __name__ == '__main__':
    nums = [1, 3, 4, 5]
    target = 7
    s = Solution()
    print(s.searchInsert2(nums, target))