函数lower_bound()在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置。如果所有元素都小于val,则返回last的位置,upper_bound()同理,只是返回大于val的第一个元素位置。
所以,要记住:函数lower_bound()在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置。如果所有元素都小于val,则返回last的位置,且last的位置是越界的!
low_bount()返回查找元素的第一个可安插位置,也就是“元素值>=查找值”的第一个元素的位置.
普通数组(a[1],a[2]...,a[n])的写法是:j=upper_bound(a+1,a+1+n,b)-a;j是a[i]数组中第一个大于b的地址符。
另一种情况是如果定义结构体:
struct node{
int id,num;
}a[100006],q[100006];
如果这时用到upper_bound,如对于数字b,返回结构体a[i].num中第一个大于b的地址符,那么不能写成j=upper_bound(a+1,a+1+n,b)-a;
这里要把输入的数字存在结构体里,如上面写的q[100006],然后重载运算符‘<',比较的时候用j=upper_bound(a+1,a+1+n,p[i])-a;
整体写法如下:
#include<iostream> #include<stdio.h> #include<string.h> #include<math.h> #include<vector> #include<map> #include<queue> #include<stack> #include<string> #include<algorithm> using namespace std; struct node{ int id,num; }a[100006],q[100006]; /*bool operator <(const node &q,const node &w) { return q.num<w.num; }两种写法都行,但上面这种快*/ bool operator<(node a,node b){ return a.num<b.num; } int main() { int n,m,i,j,k; while(scanf("%d%d",&n,&m)!=EOF) { map<int,int>hash; hash.clear(); for(i=1;i<=n;i++){ scanf("%d",&a[i].num); a[i].id=i; hash[a[i].num]++; } sort(a+1,a+1+n); for(i=1;i<=m;i++){ scanf("%d",&q[i].num); if(hash[q[i].num]==0){ printf("-1\n");continue; } else{ hash[q[i].num]--; j=upper_bound(a+1,a+1+n,q[i])-a; printf("%d\n",a[j-1].id); for(k=j-1;k<=n-1;k++){ a[k].id=a[k+1].id; a[k].num=a[k+1].num; } n--; } } } return 0; }