欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

STL二分查找的使用

程序员文章站 2024-03-17 18:58:22
...

STL二分查找

使用对象:有序数组

头文件:algorithm

lower_bound()返回值是一个迭代器,返回指向大于等于key的第一个值的位置
函数参数中的数组边界a,a+8为左开右闭。查找失败返回后一个地址(越界地址)

    #include <algorithm>
    #include <iostream>
    using namespace std;
    int main()
    {
    	int a[]={1,2,3,4,5,7,8,9};
    	printf("%d",lower_bound(a,a+8,6)-a); 
    	
     return 0;	
    } 

输出:5

将key换成10,所有val都小于key,返回last的位置

#include <algorithm>
#include <iostream>
using namespace std;
int main()
{
	int a[]={1,2,3,4,5,7,8,9};
	printf("%d",lower_bound(a,a+8,10)-a); 
	
 return 0;	
} 

输出: 8

upper_bound()函数,它返回大于等于key的最后一个元素

例题:CF1077E
AC代码:

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e6 + 7;

int a[maxn];
map<int, int> mp;

int main()
{
	int n;
	scanf("%d",&n);
	int cnt=0,x;
	for(int i=1;i<=n;i++)
    {
		scanf ("%d",&x);
		if(!mp[x])
		{
			mp[x]=++cnt;
		}
		++a[mp[x]];
	}
	sort(a+1,a+cnt+1);
	int sum=0;
	for(int j=1;j<=n;++j)
    {
		int m=0;
		int l=1;
		for (int k=j;k<=n;k*=2)
        {
			int p=lower_bound(a+l,a+cnt+1,k)-a;
			if(p==cnt+1)
                break;
            m+=k;
			l=p+1;
		}
		sum=max(sum,m);
	}
	printf ("%d\n",sum);
	return 0;
}