蓝桥杯 试题 算法训练 找数2

资源限制

内存限制：256.0MB C/C++时间限制：1.0s Java时间限制：3.0s Python时间限制：5.0s

---

找数2

【问题描述】在一个小到大的有序序列中（不存在重复数字），查找某个数所在的位置。如果该数不在该数列中，输出其应插入点的位置。

输入说明

　　第一行输入N表示有N个数据（N<=1000)

　　第二行N个小到大非重复的数字（<=10000)

　　第三行数字X，查找X

输出说明

　　若X在序列中则输出其相应的位置，若不在序列中输出其应插入的位置。

样例输入

10

12 34 56 78 88 99 101 134 145 233

88

样例输出

5

样例输入

10

23 34 56 78 99 123 143 155 167 178

128

样例输出

7

说明：保证输入数据是有序的。

思路：

有点坑，就是最开始我一直没注意n==1的时候和当给的数大于序列中最后一个最大的数的时候的输出特判

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e6+10;
int n,m,t,d;
int a[N],b[N];
string sa,sb;
bool check(int x){
	if(a[x]>m)return true;
	else return false;
}
signed main(){
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	cin>>m;
	if(n==1){     
		if(m<=a[0])cout<<1<<endl;
		else cout<<2<<endl;
	}
	else if(m>a[n-1])cout<<n+1<<endl;
	else{
		int l=0,r=n-1;
		while(l<r){
			int mid = (l+r)>>1;
			if(check(mid))r=mid;
			else l = mid+1;
		}
//		cout<<"l:"<<l<<endl;
		if(a[l-1]==m)cout<<l<<endl;
		else cout<<l+1<<endl;
	}
    return 0;
}