R7-1 二分查找 python解法

R7-1 二分查找
分数 10
作者 usx程序设计类课程组
单位 绍兴文理学院
对于输入的n个整数,先进行升序排序,然后进行二分查找。

输入格式:
测试数据有多组,处理到文件尾。每组测试数据第一行输入一个整数n(1≤n≤100),第二行输入n个各不相同的待排序的整数,第三行是查询次数m(1≤m≤100),第四行输入m个待查找的整数。

输出格式:
对于每组测试,分2行输出,第一行是排序后的升序的结果,每两个数据之间留一个空格;第二行是查找的结果,若找到则输出排序后元素的位置(从1开始,每两个数据之间留一个空格),否则输出0。

输入样例:
9
4 7 2 1 8 5 9 3 6
5
10 9 8 7 -1
输出样例:
1 2 3 4 5 6 7 8 9
0 9 8 7 0
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

try :
    while True:
        n = int(input())
        ls = list(map(int, input().split()))
        ls.sort()
        print(' '.join(map(str, ls)))
        s = int(input())
        ls2 = list(map(int, input().split()))
        for i in ls2:
            if i not in ls:
                if i == ls2[-1]:
                    print(0)
                else:
                    print(0, end=' ')
                continue
            l, r = 0, n-1
            while True:
                mid = (l+r)//2
                if i < ls[mid]:
                    r = mid - 1
                elif i >ls[mid]:
                    l = mid + 1
                else:
        #             i == ls[mid]
                    if i == ls2[-1]:
                        print(mid+1)
                    else:
                        print(mid+1, end=' ')
                    break
except:
    pass
    
        
    

你可能感兴趣的:(pta,算法)