样例过了,除了最后一个TLE,其他全部WA,求助!

P2249 【深基13.例1】查找

paulpao @ 2023-04-04 21:23:25

#include<iostream>
using namespace std;
int main(){
    int n,m,a[100000000+5],ask[100000000+5];
    bool s;
    cin>>n>>m;
    int l=1,r=n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=m;i++){
        cin>>ask[i];
    }
    for(int i=1;i<=m;i++){
        s=1;        
        l=1;
        r=n;
        while(l<r){

            int mid=(l+r)/2;
            if(a[mid]==ask[i]){
                if(a[mid-1]==ask[i]){
                    r--;
                }
                else{
                    cout<<mid<<" ";
                    s=0;
                    break;
                }
            } 
            else if(a[mid]>ask[i]){
                r=mid;
            }
            else{
                l=mid+1;
            }

        }
        if(s)cout<<-1;
    }
    return 0;
} 

by qinxiang @ 2023-04-05 10:53:50

用scanf输入 或者封装一个函数

#include <iostream>
using namespace std;
long long n,m;
long long a[1000005];
int find(int k)
{
    int l=1,r=n;
    while(l<r)
    {
        int mid=(l+r)/2;
        if(a[mid]>=k)
        {
            r=mid;
        } 
        else
        {
            l=mid+1;
        }
    }
    if(a[l]==k)
    {
        return l;
    }
    else
    {
        return -1;
    }   
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    for(int i=1;i<=m;i++)
    {
        int k;
        cin>>k;
        cout<<find(k)<<" ";
    }
    return 0;
}

|