从有序的数组中查找某个值
问题:
给定长度为N的单调不下降数列,求满足ai>=k的最小i。
输入:
5 3
2 3 3 5 6
输出:
1
#include<stdio.h> int main() { int n,k; int a[100]; scanf("%d%d",&n,&k); for(int i=0;i<n;i++) scanf("%d",&a[i]); int st=-1,en=n; while(en-st>1) { int mid=(en+st)/2; if(a[mid]>=k) en=mid; else st=mid; } printf("%d ",en); }
这样是不是很麻烦呢? 现在附上STL的lower_bound 函数的实现;
首先简单理解下lower_bound 的基本应用:
头文件:#include<algorithm>
#include <algorithm>//必须包含的头文件 #include <stdio.h> using namespace std; int main() { int n,a[100],m; int left,right,i; scanf("%d",&n);//设初始数组内元素有n个 for(i=0;i<n;i++) scanf("%d",&a[i]); scanf("%d",&m);//插入的数为m left = upper_bound(a,a+n,m)-a;//按从小到大,m最多能插入数组a的哪个位置 right = lower_bound(a,a+n,m)-a;//按从小到大,m最少能插入数组a的哪个位置 printf("m最多能插入数组a的%d ",left); for(i=0;i<left;i++) printf("%d ",a[i]); printf("%d ",m); for(i=left;i<n;i++) printf("%d ",a[i]); printf(" "); printf("m最少能插入数组a的%d ",right); for(i=0;i<right;i++) printf("%d ",a[i]); printf("%d ",m); for(i=right;i<n;i++) printf("%d ",a[i]); return 0; }