洛谷 P2249 【深基13.例1】查找(手写二分查找+STL lower_bound()+map)

最新推荐文章于 2024-08-06 20:14:06 发布

小黑哎呀

最新推荐文章于 2024-08-06 20:14:06 发布

阅读量449

点赞数

分类专栏： STL map 文章标签：数据结构二分法算法

本文链接：https://blog.csdn.net/Thy__father/article/details/116381623

版权

STL 同时被 2 个专栏收录

9 篇文章 1 订阅

订阅专栏

map

5 篇文章 0 订阅

订阅专栏

菜鸟生成记(52)

P2249 【深基13.例1】查找

二分查找的模板题，也是map的模板练习题

map吸氧AC代码(不开O2优化,T两个点;讲真的,map的红黑树查找效率不低啊!居然会T)

#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
const int N=1e6+10;
int main()
{
	int n,k,x;
	map<int,vector<int> >m;
	cin>>n>>k;
	for(int i=0;i<n;i++)
	{
		cin>>x;
		m[x].push_back(i+1);
	}
	for(int i=0;i<k;i++)
	{
		cin>>x;
		if(m.find(x)==m.end())
		cout<<-1;
		else
		cout<<m[x][0];
		cout<<" ";
	}
	return 0;
}

手动二分查找(AC)

#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
const int N=1e6+10;
int a[N];
int find1(int l,int r,int x)
{
	int m=(l+r)/2;
	if(l==r)
	{
		if(a[m]!=x)//查找失败 
		return -1;
		if(x==a[m])//查找成功返回下标 
		return m+1;
	}
	else if(x>a[m])
	find1(m+1,r,x);//查找右边 
	else if(x<=a[m])
	find1(l,m,x);//查找左边 
}
int main()
{
	int n,k,x;
	cin>>n>>k;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<k;i++)
	{
		cin>>x;
	int s=find1(0,n-1,x);
	cout<<s<<" ";
	}
	return 0;
}

STL lower_bound()

#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
using namespace std;
const int N=1e6+10;
int a[N];
int main()
{
	int n,k,x;
	cin>>n>>k;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	for(int i=0;i<k;i++)
	{
		cin>>x;
		int s=lower_bound(a,a+n,x)-a;
		if(a[s]==x)
		cout<<s+1<<" ";
		else
		cout<<-1<<" ";
	}
	return 0;
}