hust oj 1628 LowerBound

题目

查询一个区间比v大的最小的数

和hdu4417的相似，可以用线段树离线，或者划分树+二分，或者分块。。。

划分树+二分：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100010
int tree[20][N],sa[N],sum[20][N];
int t,n,m,op,l,r;

inline void build(int d,int l,int r)
{
    int m=(l+r)>>1;
    int lless=m-l+1;
    for(int i=l;i<=r;i++)
    {
        if(tree[d][i]<sa[m])
            lless-=1;
    }
    int lp=l,rp=m+1;
    for(int i=l;i<=r;i++)
    {
        if(l==i) sum[d][i]=0;
        else  sum[d][i]=sum[d][i-1];
        if(tree[d][i]<sa[m])
        {
            tree[d+1][lp++]=tree[d][i];
            sum[d][i]++;
        }
        else if(tree[d][i]>sa[m])
        {
            tree[d+1][rp++]=tree[d][i];
        }
        else
        {
            if(lless)
            {
                sum[d][i]++;
                tree[d+1][lp++]=tree[d][i];
                lless--;
            }
            else
            {
                tree[d+1][rp++]=tree[d][i];
            }
        }
    }
    if(l==r) return;
    build(d+1,l,m),build(d+1,m+1,r);
}


inline int query(int l,int r,int x,int y,int k,int d)
{
    if(x==y) return tree[d][x];
    int ss,s,m=(l+r)>>1;
    ss=(l==x?0:sum[d][x-1]);
    s=sum[d][y]-ss;
    if(s>=k)
    {
        int xx=l+ss;
        int yy=l+ss+s-1;
        return query(l,m,xx,yy,k,d+1);
    }
    else
    {
        int ll=x-1-l+1-ss;
        int rr=y-x+1-s;
        int xx=m+1+ll;
        int yy=m+1+ll+rr-1;
        return query(m+1,r,xx,yy,k-s,d+1);
    }
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&sa[i]);
            tree[0][i]=sa[i];
        }
        sort(sa+1,sa+n+1);
        build(0,1,n);
        while(m--)
        {
            scanf("%d%d%d",&op,&l,&r);
            int s=1,t=l-op+1;
            int ans=-1;
            while(s<=t)
            {
                int mid=(s+t)>>1;
                int temp=query(1,n,op,l,mid,0);
                if(temp>r) ans=temp;
                if(temp<=r) s=mid+1;
                else  t=mid-1;
            }
            if(ans==-1)
                puts("not exist");
            else
                printf("%d\n",ans);
        }
    }
}

分块：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

#define N 100010
#define INF ~0U>>1
#define S 350

inline int input(){
    int ret=0;
    char c=getchar();
    while(c<'0' || c>'9'){
        c=getchar();
    }
    while(c>='0' && c<='9'){
        ret=ret*10+c-'0';
        c=getchar();
    }
    return ret;
}
int a[N],block[N/S+1][S];
int cnt,num;
int t,n,m;
int l,r,v;

inline int query(int l,int r,int v){
    int ll=l/S,lr=r/S;
    int ans=INF;
    if(ll == lr){
        for(int i=l;i<=r;i++){
            if(a[i]>v && a[i]<ans){
                ans=a[i];
            }
        }
    }
    else{
        for(int i=l;i<(ll+1)*S;i++){
            if(a[i]>v && a[i]<ans){
                ans=a[i];
            }
        }
        for(int i=lr*S;i<=r;i++){
            if(a[i]>v && a[i]<ans){
                ans=a[i];
            }
        }
        for(int i=ll+1;i<lr;i++){
            if(block[i][0] > v){
                ans=min(ans,block[i][0]);
            }
            else if(block[i][S-1] <= v){
                continue;
            }
            else{
                int s=0,t=S-1,tag=INF;
                while(s<=t){
                    int mid=(s+t)>>1;
                    if(block[i][mid]>v) tag=mid;
                    if(block[i][mid]<=v) s=mid+1;
                    else t=mid-1;
                }
                ans=min(ans,block[i][tag]);
            }
        }
    }
    return ans;
}
int main()
{
    t=input();
    while(t--){
        n=input(),m=input();
        cnt=num=0;
        for(int i=0;i<n;i++){
            a[i]=input();
            block[cnt][num]=a[i];
            num++;
            if(num == S){
                sort(block[cnt],block[cnt]+num);
                num=0;
                cnt++;
            }
        }
        if(num) sort(block[cnt],block[cnt]+num);
        while(m--){
            l=input(),r=input(),v=input();
            l--,r--;
            int ans=query(l,r,v);
            if(ans == INF){
                printf("not exist\n");
            }
            else{
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}

线段树离线:

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

inline int input(){
    int ret=0;
    char c=getchar();
    while(c<'0' || c>'9'){
        c=getchar();
    }
    while(c>='0' && c<='9'){
        ret=ret*10+c-'0';
        c=getchar();
    }
    return ret;
}
#define N 100010
#define INF ~0U>>1
#define L t<<1
#define R t<<1|1
struct node{
    int l,r,val;
}root[N<<2];
inline void build(int t,int l,int r){
    root[t].l=l,root[t].r=r;
    root[t].val=INF;
    if(l!=r){
        int m=(l+r)>>1;
        build(L,l,m);
        build(R,m+1,r);
    }
}
inline void modefiy(int t,int x,int val){
    int l=root[t].l,r=root[t].r;
    if(l==r){
        root[t].val=val;return;
    }
    int m=(l+r)>>1;
    if(x<=m) modefiy(L,x,val);
    else  modefiy(R,x,val);
    root[t].val=min(root[L].val,root[R].val);
}
inline int query(int t,int x,int y){
    int l=root[t].l,r=root[t].r;
    if(l==x&&r==y){
        return root[t].val;
    }
    int m=(l+r)>>1;
    int ans=INF;
    if(x<=m) ans=min(ans,query(L,x,min(m,y)));
    if(y>m) ans=min(ans,query(R,max(m+1,x),y));
    return ans;
}


struct Point{
    int id,val;
}p[N];
bool cmp1(Point i,Point j){
    if(i.val == j.val) return i.id<j.id;
    return i.val>j.val;
}
struct Query{
    int l,r,v,id;
}q[N];

bool cmp2(Query i,Query j){
    if(i.v == j.v){
        if(i.l == j.l) {
            return i.r<j.r;
        }
        return i.l<j.l;
    }
    return i.v>j.v;
}

int t,n,m,ans[N];

int main(){
    t=input();
    while(t--){
        n=input(),m=input();
        for(int i=1;i<=n;i++){
            p[i].id=i,p[i].val=input();  
        }
        sort(p+1,p+n+1,cmp1);
        for(int i=1;i<=m;i++){
            q[i].id=i,q[i].l=input(),q[i].r=input(),q[i].v=input();
            
        }
        sort(q+1,q+m+1,cmp2);
        int j=1;
        build(1,1,n);
        for(int i=1;i<=m;i++){
            while(j<=n && p[j].val>q[i].v){
                modefiy(1,p[j].id,p[j].val),j++;
            }
            int tag=query(1,q[i].l,q[i].r);
            if(tag <= q[i].v) tag=INF;
            ans[q[i].id]=tag;
            //  printf("%d %d %d %d \n",q[i].id,ans[q[i].id],q[i].v,tag);
        }
        for(int i=1;i<=m;i++){
            if(ans[i] == INF){
                puts("not exist");
            }
            else{
                printf("%d\n",ans[i]);
            }
        }
    }
    return 0;
}