はなたの日記

プログラミング(C言語)の勉強をしています。超初心者です。

ITP2_6_D Equal Range

問題
Equal Range | Aizu Online Judge

itp2の6は二分探索に関する問題です。今回の問題は、与えられた値のlower boundとupper boundを求める問題です。
以前6-cのlower boundを求める問題を解いたのですが、ぐっちゃぐちゃなコードだったので、この問題を解くために考え直しました。

アルゴリズム
1.a[mid]==keyかどうか調べる
2.(1)a[mid]==keyならmid付近を探索してreturn
(2)a[mid] > keyならa[left...mid-1]で1に戻る
(3)a[mid] < keyならa[mid+1...right]で1に戻る
3.二分探索終了後、最後のmid付近を探索

このような感じです。

#include<stdio.h>
#define N 100000

int lower_bound(int a[],int x,int n){
	int left=0;
	int right=n-1;
	int mid;
	
	while(left<=right){
	
		mid=(left+right)/2;
		
		if(a[mid]==x){
			for(int i=mid;i>=0;i--)
				if(a[i]!=x)return i+1;
			return 0;
		}
		
		else if(a[mid] > x){
			right=mid-1;
		}
		else if(a[mid] < x){
			left=mid+1;
		}
		
	}
	
	
	if(a[mid] < x){
		for(int i=mid;i<n;i++)
			if(a[i] >= x)return i;
		return n;
		
	}
	else {
		for(int i=mid;i>=0;i--)
			if(a[i] <= x)return i+1;
		return 0;
	}

	
}


int upper_bound(int a[],int x,int n){
	int left=0;
	int right=n-1;
	int mid;
	
	while(left<=right){
	
		mid=(left+right)/2;
		
		if(a[mid]==x){
			for(int i=mid;i<n;i++)
				if(a[i]!=x)return i;
			return n;
		}
		
		else if(a[mid] > x){
			right=mid-1;
		}
		else if(a[mid] < x){
			left=mid+1;
		}
		
	}
	
	
	if(a[mid] < x){
		for(int i=mid;i<n;i++)
			if(a[i] > x)return i;
		return n;
		
	}
	else {
		for(int i=mid;i>=0;i--)
			if(a[i] <= x)return i+1;
		return 0;
	}

	
}



int main(){
	int n,i,q,x;
	int a[N];
	
	scanf("%d",&n);
	for(i=0;i<n;i++)scanf("%d",&a[i]);
	
	scanf("%d",&q);
	for(i=0;i<q;i++){
	
		scanf("%d",&x);
		printf("%d %d\n",lower_bound(a,x,n),upper_bound(a,x,n));
	
	}
	
	return 0;
	
}

すっきり書けてよかったです。