HDU 4417 Super Mario

离线处理的策略也很重要！

AC：

#include <stdlib.h>
#include <cstdio>
#include <iostream>
#include <queue>
#include <stack>
#include <string.h>
#include <math.h>
#include <algorithm>

#define mem(t, v)   memset ((t) , v, sizeof(t))
#define INF 0x3f3f3f3f
#define eps 1e-8
typedef long long ll;

using namespace std;

const int MAX=100010;
int t,n,m;
int f[MAX];
int ref[MAX];
int mark[MAX];
struct s{
    int val,id;
}in[MAX];
struct z{
    int x,y,id,h,ans;
}q[MAX];
struct node{
    int l,r;
    int sum;
}tree[MAX<<2];

bool cmp(s a,s b){
    return a.val<b.val;
}

bool _cmp(s a,s b){
    return a.id<b.id;
}

bool cmpq(z a,z b){
    return a.h<b.h;
}

bool _cmpq(z a,z b){
    return a.id<b.id;
}

void Build(int i,int l,int r){
    tree[i].l=l,tree[i].r=r,tree[i].sum=0;
    if(l==r)return ;
    int mid=(l+r)>>1;
    Build(i<<1,l,mid);
    Build(i<<1|1,mid+1,r);
}

void Update(int i,int x,int k){
    if(tree[i].l==tree[i].r){
        tree[i].sum+=k;
        return ;
    }
    int mid=(tree[i].r+tree[i].l)>>1;
    if(mid>=x)Update(i<<1,x,k);
    if(mid+1<=x)Update(i<<1|1,x,k);
    tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum;
}

int Search(int i,int l,int r){
    if(tree[i].l>=l&&tree[i].r<=r){
        return tree[i].sum;
    }
    if(tree[i].l>r||tree[i].r<l)return 0;
    int s=0;
    if(tree[i<<1].r>=l)s+=Search(i<<1,l,r);
    if(tree[i<<1|1].l<=r)s+=Search(i<<1|1,l,r);
    return s;
}

int main(){
    scanf("%d",&t);
    for(int k=1;k<=t;k++){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&in[i].val);
            in[i].id=i;
        }

        sort(in+1,in+n+1,cmp);
//        for(int i=1;i<=top;i++)
//            printf("%d:%d++\n",i,f[i]);

        for(int i=0;i<m;i++){
            scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].h);
            q[i].id=i;
        }
        sort(q,q+m,cmpq);
        Build(1,1,n);
        for(int i=0,j=1;i<m;i++){
            while(j<=n&&in[j].val<=q[i].h){
                Update(1,in[j++].id,1);
            }
            q[i].ans=Search(1,q[i].x+1,q[i].y+1);
        }
        sort(q,q+m,_cmpq);
        printf("Case %d:\n",k);
        for(int i=0;i<m;i++)
            printf("%d\n",q[i].ans);
    }
    return 0;
}

 

TLE： 

#include <stdlib.h>
#include <cstdio>
#include <iostream>
#include <queue>
#include <stack>
#include <string.h>
#include <math.h>
#include <algorithm>

#define mem(t, v)   memset ((t) , v, sizeof(t))
#define INF 0x3f3f3f3f
#define eps 1e-8
typedef long long ll;

using namespace std;

const int MAX=100010;
int t,n,m;
int f[MAX];
int ref[MAX];
int mark[MAX];
struct s{
    int val,id;
}in[MAX];
struct z{
    int x,y,id,h,ans;
}q[MAX];
struct node{
    int l,r;
    int sum;
}tree[MAX<<2];

bool cmp(s a,s b){
    return a.val<b.val;
}

bool _cmp(s a,s b){
    return a.id<b.id;
}

bool cmpq(z a,z b){
    if(a.x<b.x)return true;
    else if(a.x==b.x&&a.y<b.y)return true;
    else return false;
}

bool _cmpq(z a,z b){
    return a.id<b.id;
}

void Build(int i,int l,int r){
    tree[i].l=l,tree[i].r=r,tree[i].sum=0;
    if(l==r)return ;
    int mid=(l+r)>>1;
    Build(i<<1,l,mid);
    Build(i<<1|1,mid+1,r);
}

void Update(int i,int x,int k){
    if(tree[i].l==tree[i].r){
        tree[i].sum+=k;
        return ;
    }
    int mid=(tree[i].r+tree[i].l)>>1;
    if(mid>=x)Update(i<<1,x,k);
    if(mid+1<=x)Update(i<<1|1,x,k);
    tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum;
}

int Search(int i,int l,int r){
    if(tree[i].l>=l&&tree[i].r<=r){
        return tree[i].sum;
    }
    if(tree[i].l>r||tree[i].r<l)return 0;
    int s=0;
if(tree[i<<1].r>=l)s+=Search(i<<1,l,r);
    if(tree[i<<1|1].l<=r)s+=Search(i<<1|1,l,r);
    return s;
}

int main(){
    scanf("%d",&t);
    for(int k=1;k<=t;k++){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&in[i].val);
            in[i].id=i;
        }

        sort(in+1,in+n+1,cmp);
        int top=0;
        in[0].val=-1;
        for(int i=1;i<=n;i++){
            if(in[i].val!=in[i-1].val){
                f[++top]=in[i].val;
            }
            ref[in[i].id]=top;
        }
        sort(in+1,in+1+n,_cmp);
//        for(int i=1;i<=top;i++)
//            printf("%d:%d++\n",i,f[i]);

        for(int i=0;i<m;i++){
            scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].h);
            q[i].id=i;
        }
        sort(q,q+m,cmpq);

        Build(1,1,top);
        mem(mark,0);
        for(int i=0,j;i<m;i++){
            j=1;
            for(;j<=q[i].x;j++){
                int t=ref[in[j].id];
                if(mark[j]){
                    Update(1,t,-1);
                    mark[j]=0;
                }
            }
//            printf("s:%d ",j);
 for(;j<=q[i].y+1;j++){
                int t=ref[in[j].id];
                if(!mark[j]){
                    Update(1,t,1);
                    mark[j]=1;
                }
            }
//            printf("e:%d |||\n",j);
            for(;j<=n;j++){
                int t=ref[in[j].id];
                if(mark[j]){
                    Update(1,t,-1);
                    mark[j]=0;
                }
            }
//            for(int o=1;o<=n;o++)
//                printf("%d*",mark[o]);
//            printf("\n");
            int h=upper_bound(f+1,f+top+1,q[i].h)-f-1;
            q[i].ans=Search(1,1,h);
//            printf("%d %d %d %d %d**\n",q[i].x,q[i].y,q[i].id,h,q[i].ans);
        }
        sort(q,q+m,_cmpq);
        printf("Case %d:\n",k);
        for(int i=0;i<m;i++)
            printf("%d\n",q[i].ans);
    }
    return 0;
}