TLE 求助
查看原帖
TLE 求助
547908
NightTide楼主2022/4/4 19:57

RT,在洛谷过了,但是在校内 OJ 被卡掉了,求助大神卡常(但其实有一个点将近两秒,不知道为什么)

#include<bits/stdc++.h>
#define MAXN 500010
using namespace std;
struct node{
    int a,b,p,id;
};
struct ques{
    int l,r,id;
};
int n,q;
stack<node> s;
ques ask[MAXN];
node data[MAXN];
int ans[MAXN],sum[MAXN];
bool cmp_node(node a,node b){
    return a.p < b.p;
}
bool cmp_ques(ques a,ques b){
    if(a.l == b.l) return a.r < b.r;
    return a.l < b.l;
}
int low_bit(int x){
    return x&(-x);
}
void updata(int pos,int val){
    while(pos <= n){
        sum[pos] += val;
        pos += low_bit(pos);
    }
}
int query(int pos){
    int res = 0;
    while(pos >= 1){
        res += sum[pos];
        pos -= low_bit(pos);
    }
    return res;
}
int main(){
    // freopen("stack.in","r",stdin);
    // freopen("stack.out","w",stdout);
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++) data[i].id = i;
    for(int i=1;i<=n;i++) scanf("%d",&data[i].a);
    for(int i=1;i<=n;i++) scanf("%d",&data[i].b);
    for(int i=1;i<=q;i++) scanf("%d%d",&ask[i].l,&ask[i].r), ask[i].id = i;
    for(int i=1;i<=n;i++){
        while(!s.empty() && !( (s.top().a != data[i].a) && (s.top().b > data[i].b) )) s.pop();
        data[i].p = s.empty() ? 0 : s.top().id; s.push(data[i]);
    }
    sort(ask+1,ask+q+1,cmp_ques);
    sort(data+1,data+n+1,cmp_node);
    int pos = 1;
    for(int i=1;i<=q;i++){
        while(data[pos].p < ask[i].l){
            updata(data[pos].id,1);
            pos++;
        }
        ans[ask[i].id] = query(ask[i].r) - query(ask[i].l - 1);
    }
    for(int i=1;i<=q;i++) printf("%d\n",ans[i]);
    return 0;
}
/*
10 4
3 1 3 1 2 3 3 2 1 1
10 10 2 9 7 5 4 7 6 1
1 4
7 8
7 10
1 8
*/
2022/4/4 19:57
加载中...