Noip[2011]Codevs[1135]洛谷[1311]选择客栈 清流解法，二分+线段树+树状数组

题目链接https://www.luogu.org/problem/show?pid=1311#sub

题目描述

丽江河边有n 家很有特色的客栈，客栈按照其位置顺序从 1 到n 编号。每家客栈都按照某一种色调进行装饰（总共 k 种，用整数 0 ~ k-1 表示），且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。

两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈），且咖啡店的最低消费不超过 p 。

他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过 p元的咖啡店小聚。

输入输出格式

输入格式：
输入文件hotel.in，共n+1 行。

第一行三个整数n ，k ，p，每两个整数之间用一个空格隔开，分别表示客栈的个数，色调的数目和能接受的最低消费的最高值；

接下来的n 行，第 i+1 行两个整数，之间用一个空格隔开，分别表示 i 号客栈的装饰色调和i 号客栈的咖啡店的最低消费。

输出格式：
输出文件名为hotel.out 。

输出只有一行，一个整数，表示可选的住宿方案的总数。

输入输出样例

输入样例#1：
5 2 3
0 5
1 3
0 2
1 4
1 5
输出样例#1：
3
说明

【输入输出样例说明】

2 人要住同样色调的客栈，所有可选的住宿方案包括：住客栈①③，②④，②⑤，④⑤，但是若选择住4 、5 号客栈的话，4 、5 号客栈之间的咖啡店的最低消费是4 ，而两人能承受的最低消费是3 元，所以不满足要求。因此只有前 3 种方案可选。

【数据范围】

对于30% 的数据，有 n $≤$100；

对于50% 的数据，有 n $≤$1,000；

对于100%的数据，有 2 ≤n ≤200,000,0 < <script type="math/tex" id="MathJax-Element-137"><</script>k≤50，0$≤$p $≤$100

这题正解$O(nk)$的模拟…
但其实可以用$O(n\log(n)^2)$卡过，卡时多刺激啊2333

枚举每一个起点$l$，二分查找出右端点$r$，使$l$~$r$这一段区间最小值小于k，则l对答案的贡献为r到n所有和l颜色相同的点的数量(r右面所有和l颜色相同的点，都可以与l构成一组解)

区间最小和可以用线段树维护，点的数量可以由树状数组统计..

除luogu外的oj基本上都可以过的，luogu稳定T3个点（垃圾评测机）

代码如下：

#include<cstdio>
using namespace std;
int n,m,k,ans,t=1,x,y;
int s[52][200050];
int read(){
    char c=getchar();
    int x=0;
    while(!(c>='0' && c<='9')) c=getchar();
    while(c>='0' && c<='9'){x=(x<<3)+(x<<1)+c-'0';c=getchar();}
    return x;
}
inline int min(int a,int b){return a<b?a:b;}
struct Data{
    int k,x;
    inline void scan(){x=read();k=read();}
}a[200050];
inline void add(int x,int p,int v){
    for(;p<=n;p=p+(p&-p))
        s[x][p]+=v;
}
inline int sum(int x,int p){
    int tmp=0;
    for(;p;p=p-(p&-p))
        tmp=tmp+s[x][p];
    return tmp;
}
struct Node{
    int l,r,minn;
    Node *ls,*rs;
}q[1200050];
void maketree(int l,int r,Node *k){
    k->l=l;k->r=r;
    if(l==r){
        k->minn=a[l].k;
        return;
    }
    int mid=(l+r)>>1;
    k->ls=&q[++t];
    k->rs=&q[++t];
    maketree(l,mid,k->ls);maketree(mid+1,r,k->rs);
    k->minn=min(k->ls->minn,k->rs->minn);
}
int Query_min(int x,int y,Node *k){
    if(k->l>=x && k->r<=y) return k->minn;
    int mid=(k->l+k->r)>>1;
    if(x>mid) return Query_min(x,y,k->rs);
    if(y<=mid) return Query_min(x,y,k->ls);
    return min(Query_min(x,y,k->ls),Query_min(x,y,k->rs));
}
inline bool check(int l,int r){
    int t=Query_min(l,r,&q[1]);
    return t<=k?true:false;
}
int main(){
    n=read();m=read();k=read();
    for(int i=1;i<=n;i++){
        a[i].scan();
        add(a[i].x,i,1);
    }
    maketree(1,n,&q[1]);
    for(int i=n-1;i>=1;i--){
        int l=i+1,r=n,tmp=0;
        while(l<=r){
            int mid=(l+r)>>1;
            if(check(i,mid)) {tmp=mid;r=mid-1;}
            else l=mid+1;
        }
        if(!tmp) continue;
        ans=ans+sum(a[i].x,n)-sum(a[i].x,tmp-1);
    }
    printf("%d",ans);
return 0;
}

codevs的评测记录

luogu的评测记录

某小oj的评测记录