编写题解 2061: [STL训练]周瑜的反间计-二分

解题思路:

为了防止超时，使用二分
排序二分即可
注意事项:

注意每组数据后要输出空行

YES和NO要大写

首先要判断数据是否大于a[n]或小于a[1]

l < r-1

否则会死循环

要判断a[l]和a[r]

参考代码:

#include <bits/stdc++.h>
using namespace std;
int a[1001];
int main()
{
    int n,m,key;
    int l,r;
    while(1)
    {
        memset(a,0,sizeof(a));
        cin >> n;
        if(n == -1) break;
        for(int i = 1; i <= n; i++)
            cin >> a[i];
        sort(a+1,a+n+1);
        cin >> m;
        for(int i = 1; i <= m; i++)
        {
            cin >> key;
            l = 1; r = n;
            if(a[1] > key || a[n] < key)
            {
                cout << "NO\n";
                continue;
            }
            while(l < r-1)
            {
                if(a[(l+r)/2] < key) l = (l+r)/2;
                else r = (l+r)/2;
            }
            if(a[l] == key) cout << "YES\n";
            else if(a[r] == key) cout << "YES\n";
            else cout << "NO\n";
        }
        cout << endl;
    }
        return 0;
}