p2_二分_数的范围O(logn)
题目:
给定一个按照升序排列的长度为 n 的整数数组,以及 q 个查询。
对于每个查询,返回一个元素 k 的起始位置和终止位置(位置从 0 开始计数)。
如果数组中不存在该元素,则返回 -1 -1。
输入格式
第一行包含整数 n 和 q,表示数组长度和询问个数。
第二行包含 n 个整数(均在 1∼10000 范围内),表示完整数组。
接下来 q 行,每行包含一个整数 k,表示一个询问元素。
输出格式
共 q 行,每行包含两个整数,表示所求元素的起始位置和终止位置。
如果数组中不存在该元素,则返回 -1 -1。
数据范围
1≤n≤100000
1≤q≤10000
1≤k≤10000
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
#二分包括:整数二分+浮点二分
#二分的思想本质就是猜数字,比如有1-n个数蔡,每次猜一半,再知道是大是小 ,所以猜logn次
#所以复杂度为O(logn)
#有单调性一定二分,没有单调性也可能可以二分【二分查找的前提是必须单调
#整数二分:对一列整数,可以从某个位置分开,前面满足性质,后面不满足性质 且没有交点(因为是整数离散)对于这种可以通过二分找到临界的两个点
#核心:取最右边的数,如一列数据找数字2,如果有多个2,则要得到最右/左边的2的位置,所以整体有往右/左的趋势
#对于序列0 1 1 2 2 2 3 3 4 4 5
#情况1【真值往右的趋势】:找最右边的数字2,则为保证所有2都在一个区间里面则想找到如下的竖线:0 1 1 2 2 2 | 3 3 4 4 5,即以最后一个2的左边为True,右边为false
#!!!!!若为True则【包括其值】向右找,若为False则【不包括其值】向左找True
#取mid=(l+r+1)/2 为啥分子+1:因为整除是向下取整的,若l=r-1,则(l+r)/2=下取整=l,则若True l=mid=l,没变,会造成死循环;若mid分子+1,则Ture为l=mid=r,即[r,r],循环停止
#if(check(mid))=True 所以【包括其值】向右找 即新区间为[mid,r],即:l=mid
#if(check(mid))=False 所以【不包括其值】向左找True 即新区间为[l,mid-1],即:r=mid-1
#情况2【真值往左的趋势】:找最左边的数字2,则为保证所有2都在一个区间里面则想找到如下的竖线:0 1 1 | 2 2 2 3 3 4 4 5,即以第一个2的右边为True,左边为false
#!!!!若为True则【包括其值】向左找,若为False则【不包括其值】向右找True
#取mid=(l+r)/2
#if(check(mid))=True 所以【包括其值】向左找 即新区间为[l,mid],即:r=mid
#if(check(mid))=False 所以【不包括其值】向右找True 即新区间为[mid+1, r],即:l=mid+1
#上面举例是数字,本质上其实就是找临界点!【二分循环中每一次l/r根据mid相应变化就相当于区间变化了,结束后l=r(二分算法一定有解,题目可能误解) 所以若没有这个值,循环结束后l=r==序列从左往右看第一个为True的数;所以二分一定可以得到一个数,这个数是不是你要找的所有要判断是否==这个数】
#总结:左真,true则左=mid;False则右=mid-1 【mid还要分子还要补一个+1】记左加右减
# 右真,true则右=mid;False则左=mid+1 【mid不用补
#题目:数的范围
def right_sort(x,nums,l,r):
while l<r:
mid = (l + r) // 2
if nums[mid]>=x: r=mid#先找起始位置,右(>=x)为True,真则r=mid,假则l=mid+1
else: l=mid+1
if nums[l]==x: return l
else: return -1
def left_sort(x,nums,l,r):
while l<r:
mid = (l + r + 1) // 2
if nums[mid]<=x: l=mid#先找末位置,左(<=x)为True,真则l=mid,假则r=mid-1
else: r=mid-1
if nums[l]==x: return l
else: return -1
if __name__=="__main__":
n,q=map(int,input("n&q\n").split())
nums=list(map(int,input("nums\n").split()))
num=[]
# chaxun=0 #用for in range循环更简单
# while chaxun<q:
# num.append(int(input(num)))
# chaxun+=1
for qq in range(q):#q是一个数,如果要用for in 循环则需要range,range(30=0,1,2不包括3
num.append(int(input()))
for x in num:
a=right_sort(x,nums,0,n-1)#找首位
b=left_sort(x, nums, 0, n - 1)#找末位
print(a,b)