Cod sursa(job #3165941)

Utilizator tudor_costinCostin Tudor tudor_costin Data 7 noiembrie 2023 10:44:11
Problema Zoo Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.71 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#include <climits>
using namespace std;
ifstream fin("zoo.in");
ofstream fout("zoo.out");
using ll=long long;
const int Nmax=16005;
pair<ll,ll> a[Nmax];
ll b[Nmax];
vector<ll> aint[4*Nmax];
void build(int nod,int st,int dr)
{
    for(int i=st;i<=dr;i++)
    {
        aint[nod].push_back(b[i]);
    }
    sort(aint[nod].begin(),aint[nod].end());
    if(st<dr)
    {
        int mij=(st+dr)/2;
        build(2*nod,st,mij);
        build(2*nod+1,mij+1,dr);
    }
}
ll query(int nod,int st,int dr,int poz1,int poz2,int val1,int val2)
{
    ll sol;
    if(poz1<=st && dr<=poz2)
    {
        int p1=lower_bound(aint[nod].begin(),aint[nod].end(),val1)-aint[nod].begin();
        int p2=upper_bound(aint[nod].begin(),aint[nod].end(),val2)-aint[nod].begin();
        sol=p2-p1;
        return sol;
    }
    int mij=(st+dr)/2;
    if(mij>=poz2) return query(2*nod,st,mij,poz1,poz2,val1,val2);
    if(mij<poz1) return query(2*nod+1,mij+1,dr,poz1,poz2,val1,val2);
    return (query(2*nod,st,mij,poz1,mij,val1,val2)+query(2*nod+1,mij+1,dr,mij+1,poz2,val1,val2));

}
int main()
{
    int n;
    fin>>n;
    for(int i=1;i<=n;i++)
    {
        fin>>a[i].first>>a[i].second;
    }
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)
    {
        b[i]=a[i].second;
    }
    build(1,1,n);
    int m;
    ll x1,x2,y1,y2;
    fin>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>x1>>y1>>x2>>y2;
        pair<ll,ll> lb={x1,0};
        int poz1=lower_bound(a+1,a+n+1,lb)-a;
        lb={x2,LLONG_MAX};
        int poz2=upper_bound(a+1,a+n+1,lb)-a-1;
        fout<<query(1,1,n,poz1,poz2,y1,y2)<<'\n';
    }
    return 0;
}