#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()