P1311 选择客栈

Posted 2021-01-08 kafuuchino

P1311 选择客栈

RMQ+二分（当然这并不是正解）

对于每一个客栈，二分满足条件的范围，在RMQ上查询。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cctype>
using namespace std;
template <typename T> inline T min(T &a,T &b) {return a<b ? a:b;}
template <typename T> inline void read(T &x){
    char c=getchar(); x=0;
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=getchar();
}
int n,k,p,mnd[18][200002],log[200002]; long long ans;
vector <int> g[52];
inline int query(int l,int r){
    int k=log[r-l+1];
    return min(mnd[k][l],mnd[k][r-(1<<k)+1]);
}
int main(){
    read(n); read(k); read(p); int q; log[0]=-1;
    for(int i=1;i<=n;++i){
        read(q); read(mnd[0][i]);
        g[q].push_back(i);
        log[i]=log[i>>1]+1;
    }
    for(int i=1;(1<<i)<=n;++i)
        for(int j=1;j+(1<<(i-1))<=n;++j)
            mnd[i][j]=min(mnd[i-1][j],mnd[i-1][j+(1<<(i-1))]);
    for(int i=0;i<k;++i)
        for(int j=0;j<g[i].size()-1;++j){
            if(query(g[i][j],g[i][g[i].size()-1])>p) break; //已经无法满足条件就跳出
            int l=j+1,r=g[i].size()-1,mid;
            while(l<r){ //二分确定范围
                mid=l+((r-l)>>1);
                if(query(g[i][j],g[i][mid])>p) l=mid+1;
                else r=mid;
            }ans+=g[i].size()-l;
        }
    printf("%lld",ans);
    return 0;
}