分治算法——二分法 PTA 二分查找

例题7-1 找数

在这里插#include<iostream>
using namespace std;

int n,m,a[110000];

int main()
{
    
    
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
    
    
        cin>>a[i];
    }
    a[0]=-1;
    for(int i=1;i<=m;i++)
    {
    
    
        int x;
        int left=1,right=n,mid;
        cin>>x;
        while(left<=right)
        {
    
    
            mid=(left+right)/2;
            if(a[mid]<=x)left=mid+1;
            else right=mid-1;
        }
        cout<<a[right]<<endl;
    }
    
}

PTA 二分查找
利用二分查找找出所给出的数在数组中的下标
输入格式:

第一行输入n和m表示数组有n个数据,m表示要对m个数进行查找
输出格式:

所有输出在一行完成,行末没有多余空格和多余回车。
输入样例:

5 5
1 2 3 4 5
1 2 3 4 5```

输出样例:
out
0 1 2 3 4

#include<iostream>
using namespace std;
int bs(int a[],int n,int x)
{
    
    
    int l,r,mid;
    l=0;
    r=n;
    while(l<=r)
    {
    
    
        mid=(l+r)/2;
        if(x<a[mid])r=mid-1;
        else if(x>a[mid])l=mid+1;
        else return mid;
    }
    return 0;
}
int main()
{
    
    
    int n,m;
    int a[100000],b[100000];
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
    
    
        cin>>a[i];
    }
    for(int j=0;j<m;j++)
    {
    
    
        cin>>b[j];
    }
    for(int i=0;i<n;i++)
    {
    
    
        if(i==0)cout<<bs(a,n,b[0]);
        else cout<<" "<<bs(a,n,b[i]);
    }
}

789. 数的范围

给定一个按照升序排列的长度为n的整数数组,以及 q 个查询。

对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。

如果数组中不存在该元素,则返回“-1 -1”。
输入格式

第一行包含整数n和q,表示数组长度和询问个数。

第二行包含n个整数(均在1~10000范围内),表示完整数组。

接下来q行,每行包含一个整数k,表示一个询问元素。
输出格式

共q行,每行包含两个整数,表示所求元素的起始位置和终止位置。

如果数组中不存在该元素,则返回“-1 -1”。

#include<iostream>
using namespace std;
const int maxn=10000005;
int a[maxn];
int bsl(int a[],int n,int x)
{
    
    
    int l,r,mid;
    l=0;
    r=n-1;
    int flag=0;
    while(l<=r)
    {
    
    
        mid=(l+r)/2;
        if(x<a[mid]){
    
    
            r=mid-1;
        }
        if(x==a[mid]){
    
    
            r=mid-1;
            flag=1;
        }
        if(x>a[mid])l=mid+1;
    }
    if(flag==1)return l;
    else return -1;
}
int bsr(int a[],int n,int x)
{
    
    
    int l,r,mid;
    l=0;
    r=n-1;
    int flag=0;
    while(l<=r)
    {
    
    
        mid=(l+r)/2;
        if(x>a[mid])
        {
    
    
            l=mid+1;
        }
        else if(x==a[mid])
        {
    
    
            l=mid+1;
            flag=1;
        }
        else r=mid-1;
    }
    if(flag==1)return r;
    else return -1;
}



int main()
{
    
    
    int n,q;
    cin>>n>>q;
    for(int i=0;i<n;i++)
    {
    
    
        cin>>a[i];
    }
    while(q--)
    {
    
    
        int k;
        cin>>k;
        cout<<bsl(a,n,k)<<" "<<bsr(a,n,k)<<endl;
    }
}

猜你喜欢

转载自blog.csdn.net/m0_51344983/article/details/112913367