题目内容:
定义一个 prime() 函数求整数 n 以内(不包括n)的所有素数(1不是素数),并返回一个按照升序排列的素数列表。使用递归来实现一个二分查找算法函数bi_search(),该函数实现检索任意一个整数在 prime() 函数生成的素数列表中位置(索引)的功能,并返回该位置的索引值,若该数不存在则返回 -1。
输入格式:
第一行为正整数 n
接下来若干行为待查找的数字,每行输入一个数字
输出格式:
每行输出相应的待查找数字的索引值
输入样例:
10
2
4
6
7
输出样例:
0
-1
-1
3
import math
import sys
def prime(n):
list=[]
for i in range(2,n):
flag=True
for j in range(2,int(math.sqrt(i))+1):
if i%j==0:
flag=False
break
if flag:
list.append(i)
return list
def search(list,x):
l=0
r=len(list)-1
while l<=r:
mid=(l+r)//2
if list[mid]==x:
return mid
elif list[mid]>x:
r=mid-1
else:
l=mid+1
if list[l]==x:
return l
else:
return -1
n=int(input())
result=prime(n)
for line in sys.stdin:
num=int(line)
if num<2 or num>=n: #防止越界
print(-1)
else:
print(search(result,num))