数据结构上机实验之二分查找

Problem Description
 在一个递增的序列里，查找元素是否存在，若存在输出YES,不存在输出NO. 
Input
 本题多组数据，首先输入一个数字n(n>=100000)，然后输入n个数，数据保证数列递增，然后再输入一个查找数字。 
Output
 若存在输出YES,不存在输出NO. 
Example Input
4
1 3 5 8
3Example Output
YESHint

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
int search(int a[],int l,int r,int key){
	int i=l,j=r;
	while(i<j){
		int mid=(i+j)/2;
		if(key==a[mid])
		return 1;
		else if(key<mid)
		j=mid-1;
		else if(key>mid)
		i=mid+1;
	}
	return 0;
}
int main(){
	int n;
	while(scanf("%d",&n)!=EOF){
		int a[n+1];
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
		}
		int m,k;
		scanf("%d",&m);
		k=search(a,0,n-1,m);
		if(k==1)
		printf("YES\n");
		else
		printf("NO\n");
	}
	return 0;
}