Cod sursa(job #2332020)
Utilizator | Data | 30 ianuarie 2019 12:00:42 | |
---|---|---|---|
Problema | Cautare binara | Scor | 100 |
Compilator | cpp-64 | Status | done |
Runda | Arhiva educationala | Marime | 2.38 kb |
#include <fstream>
#include <algorithm>
using namespace std;
ifstream fin("cautbin.in");
ofstream fout("cautbin.out");
int n,v[100001],i,x,m,task,p,u,ok1,poz,j,mm;
int main()
{
fin>>n;
for (i=1;i<=n;i++)
{
fin>>v[i];
}
fin>>mm; //taskuri
for (i=1;i<=mm;i++)
{
fin>>task>>x;
ok1=0;
if (task==0)
{
p=1;
u=n;
while (p<=u)
{
m=(p+u)/2;
if (v[m]<x)
{
p=m+1;
}
else
if (v[m]==x)
{
ok1=1;
poz=m;
break;
}
else
{
u=m-1;
}
}
if (ok1==0)
{
fout<<"-1"<<'\n'; //nu exista
}
else
{ //caut cea mai mare pozitie in care e x
for (j=poz;j<=n;j++)
{
if (v[j]!=x)
{
break;
}
}
j--;
fout<<j<<'\n';
}
}
else
if (task==1)
{
poz=upper_bound(v+1,v+n+1,x)-v; //return pozitia cea mai mare in care e x
//sau elem imediat urmator
if (v[poz]>x) //daca e elem urmator
{
poz--;
}
if (poz>n) //daca nu exista x si iese e > decat ultimul elem si iese
{
poz--;
}
fout<<poz<<'\n';
}
else if(task==2)
{
p=1;
u=n;
//cauta primul x daca exista
//sau cel mai mic nr mai mare decat x
while (p<=u)
{
m=(p+u)/2;
if (v[m]<x)
{
p=m+1;
}
else
{
u=m-1;
}
}
//prima jumatate si in ex de tip 1 2 4 5 6 si cauta 3 ia pe 2 nu pe 4
m=(p+u)/2;
if (v[m]<x)
{
m++;
}
fout<<m<<'\n';
}
}
return 0;
}