九度教程第20題 查找（二分法）

輸入數組長度 n 輸入數組 a[1…n] 輸入查找個數m 輸入查找數字b[1…m] 輸出 YES or NO 查找有則YES 否則NO 。

輸入描述:
輸入有多組數據。
每組輸入n，然後輸入n個整數，再輸入m，然後再輸入m個整數（1<=m,n<=100）。

輸出描述:
如果在n個數組中輸出YES否則輸出NO。
示例1
輸入
5
1 5 2 4 3
3
2 5 6
輸出
YES
YES
NO
 

我的代碼（排序二分查找），其實可以把二分查找的方式拉出來寫成函數調用:)

#include <stdio.h>
#include <algorithm>
using namespace std;

int main()
{
	int n, m;
	int num[101], w[101];
	while(scanf("%d", &n) != EOF)
	{
		for(int i = 1; i <= n; i++)
			scanf("%d", &num[i]);
		sort(num+1, num+n+1);
		scanf("%d", &m);
		for(int i = 1; i <= m; i++)
		{
			scanf("%d", &w[i]);
		}
		for(int i = 1; i <= m; i++)
		{
			bool isYes = false;
			int base = 1, top = n;
			while(top >= base)
			{
				int mid = (top + base) / 2;
				if(mid == w[i])
				{
					printf("YES\n");
					isYes = true;
					break;
				}	
				else if(mid > w[i])
					top = mid - 1;
				else
					base = mid + 1; 
			}
			if(!isYes)
				printf("NO\n");
		}
		
	}
	return 0;
}