#7. 找位置

找位置

题目

小朋友出操,按学号从小到大排成一列; 小红来迟了,请你给小红出个主意,让她尽快找到她应该排的位置。 算法复杂度要求不高于nLog(n);学号为整数类型,队列规模 ≤ 10000;

输入描述

第一行:输入已排成队列的小朋友的学号(正整数),以","隔开;例如:
93,95,97,100,102,123,155
第二行:小红学号,如:
110

输出描述

输出一个数字,代表队列位置(从1开始)。例如:
6

解题思路

将已排好序的队列转换为整数数组,并对数组进行排序。二分查找在排序后的队列中找到小红的学号应该插入的位置。如果小红的学号恰好在队列中已有学号的位置,则返回该位置;否则,返回小红学号应该插入的位置(即该位置之前的学号都小于小红的学号,该位置之后的学号都大于小红的学号)。

代码

# -*- coding: utf-8 -*-

def find_position(queue, target):
    queue.sort()
    left, right = 0, len(queue) - 1
    while left <= right:
        mid = left + (right - left) // 2
        if queue[mid] == target:
            return mid + 1
        elif queue[mid] < target:
            left = mid + 1
        else:
            right = mid - 1
    return left + 1

def main():
    queue = list(map(int, input().split(',')))
    target = int(input())

    position = find_position(queue, target)
    print(position)

if __name__ == "__main__":
    main()