【深基13.例1】查詢

題目描述

 

輸入 n(n≤10⁶) 個不超過 10⁹ 的單調不減的（就是後面的數字不小於前面的數字）非負整數 a1,a2,…,an​，然後進行m(m≤10⁵)次詢問。對於每次詢問，給出一個整數q(q≤10⁹)，要求輸出這個數字在序列中第一次出現的編號，如果沒有找到的話輸出 -1 。

 

輸入格式

 

第一行 2 個整數 n 和 m，表示數字個數和詢問次數。

第二行 n 個整數，表示這些待查詢的數字。

第三行 m 個整數，表示詢問這些數字的編號，從 1 開始編號。

 

輸出格式

 

m 個整數表示答案。

 

輸入輸出樣例

 

輸入

11 3

1 3 3 3 5 7 9 11 13 15 15

1 3 6

輸出

1 2 -1 

AC程式碼：

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    long long a[n+1]={0},b[m+1]={0};
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=m;i++){
        cin>>b[i];
    }
    for(int i=1;i<=m;i++){
        int l=1,r=n;
        while(l<r){
            int mid=(l+r)/2;
            if(a[mid]<b[i]){
                l=mid+1;
            }else{
                r=mid;
            }
        }
        if(a[r]!=b[i]){
            cout<<"-1 ";
        }else{
            cout<<l<<" ";
        }
    }
} 

WA程式碼(十分相近)：

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    long long a[n+1]={0},b[m+1]={0};
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=m;i++){
        cin>>b[i];
    }
    for(int i=1;i<=m;i++){
        int l=1,r=n;
        while(l<r){
            int mid=(l+r)/2;
            if(a[mid]<b[i]){
                l=mid+1;
            }else{
                r=mid-1;
            }
        }
        if(a[r]!=b[i]){
            cout<<"-1 ";
        }else{
            cout<<l<<" ";
        }
    }
} 

 

 

 

讓我們看看那出了錯誤！

操！——r=mid-1;

它為何錯！——

（red=r,yellow=l,green=mid）

第一次：

執行——r=mid-1;

第二次：

 

執行——r=mid-1;

第三次：

 

 

執行——r=mid-1;

r=0;

a[0]=0;

所以輸出的都是-1；