Cod sursa(job #850698)

Utilizator Luncasu_VictorVictor Luncasu Luncasu_Victor Data 8 ianuarie 2013 20:04:33
Problema Cautare binara Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.89 kb
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <cmath>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <deque>
#include <set>
#include <stack>
#include <cassert>
using namespace std;

#define PRO "cautbin"
void OpenFiles(int EVAL)
{
	if(EVAL)
	{
		char input[100] = PRO, output[100] = PRO;
		freopen(strcat(input, ".in"),"r",stdin);
		freopen(strcat(output,".out"),"w",stdout);
	} else
	{
		freopen("test.in","r",stdin);
		freopen("test.out","w",stdout);
	}
}

#define MAX 100001
#define INF 0xffffff

int a[MAX],n;

int BSearch(int val)
{
	int m,l=1,r=n;
	while(r-l>1)
	{
		m=(l+r)/2;
		if(a[m]<=val)l=m; else r=m-1;
	}
	return a[r]==val?r:a[l]==val?l:-1;
}

int BSearch1(int val)
{
	int m,l=1,r=n;
	while(r-l>1)
	{
		m=(l+r)/2;
		if(a[m]<=val)l=m; else r=m-1;
	}
	return a[r]<=val?r:l;
}


//0 x - cea mai mare pozitie pe care se afla un element cu valoarea x sau -1 daca aceasta valoare nu se gaseste in sir
//1 x - cea mai mare pozitie pe care se afla un element cu valoarea mai mica sau egala cu x in sir. Se garanteaza ca cel mai mic numar al sirului este mai mic sau egal decat x 
//2 x - cea mai mica pozitie pe care se afla un element cu valoarea mai mare sau egala cu x in sir. Se garanteaza ca cel mai mare numar din sir este mai mare sau egal decat x

int BSearch2(int val)
{
	int m,l=1,r=n;
	while(l<r)
	{
		m=(l+r)/2;
		if(a[m]<val)l=m+1; else r=m;
	}
	return l;
}

int main(int argv,char *args[])
{
	OpenFiles(argv==0);	
	// start
	int op,m,x;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]);
		scanf("%d",&m);
		for(int i=1;i<=m;i++)
		{
			scanf("%d %d",&op,&x);
			switch(op)
			{
			case 0: printf("%d\n",BSearch(x)); break;
			case 1: printf("%d\n",BSearch1(x)); break;
			case 2: printf("%d\n",BSearch2(x)); break;
			}
		}
	return 0;
}