选择客栈noip2011

哈，没想到吧。今天居然有两篇（算什么，厕所读物吗

选择客栈

本题的更优解请跳转zt

这题11年，刚改2day。

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

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

对于100%的数据，有 2 ≤n ≤200,000，0<k ≤50，0≤p ≤100 ， 0 ≤最低消费≤100。

好的，直接枚举左边的客栈，枚举右边的客栈，再判断是否合法就OK了。

数组开大一点还多过了luogu的一个点（垃圾数据

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
inline void in(int &p,char c=getchar())
{
    while(c<'0' or c>'9')
        c=getchar();
    p=0;
    while(c>='0' and c<='9')
        p=p*10+c-'0',c=getchar();
}
struct hotel
{
    int color;
    int money;
}h[1000001];
long long ans;
int main()
{
    int n,m,maxmoney;
    in(n);in(m);in(maxmoney);
    for(int i=1;i<=n;i++)
        in(h[i].color),in(h[i].money);
    for(int i=1;i<=n;i++)
    {   
        bool you=(h[i].money<=maxmoney);//这是“有”，不是“右”
        for(int j=i+1;j<=n;j++)
        {
            if(h[j].money<=maxmoney)
                you=1;
            if(you and i!=j and h[i].color==h[j].color)
                ans++;
        }
    }
    cout<<ans;
    return 0;
}

看了题解，我受到启发。决定小题大做，用线段树

toby_ztだらお：为什么要用线段树？

对于两个客栈之间，只要有一个咖啡厅符合条件，那整段都符合条件，很有线段树的感觉。

这题用线段树的话，只需要建树

void build(int rt,int l,int r)
{
    if(l==r)
    {
        tree[rt]=di[l];
        return ;
    }
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    tree[rt]=(tree[rt<<1] or tree[rt<<1|1]);
}

和查询

bool getans(int rt,int L,int R,int l,int r)
{
    if(L<=l and R>=r)
        return tree[rt];
    int m=(l+r)>>1;
    bool aans=0;//我外面定义一个计答案的ans，为了程序的严谨性，嗯变量名丑一点也是可以原谅的
    if(L<=m and !aans)
        aans=getans(rt<<1,L,R,l,m);
    if(R>m and !aans)//!aans的目的是减少不必要的查询。若已经有左子树ok，右子树什么样已经不重要了
        aans=getans(rt<<1|1,L,R,m+1,r);
    return aans;
}

两种操作。

那怎么枚举求的区间呢？或者说，怎么枚举两个人住的两家酒店？

为了减少枚举量，我们可以按颜色（色调）枚举。

使用vector <int> 存每种颜色的酒店。

输入的时候就顺手预处理一下：

vector <int> c[52];//0-50的话，最多有51种，所以开52就够啦
...
for(int i=1;i<=n;i++)
{
    int x,y;
    in(x);in(y);//in(int &p)就是读入优化
    c[x].push_back(i);
    if(y<=maxm)//maxm是两个人最高的消费
        di[i]=1;//标记一下，给一会建树用
}

枚举的时候就是这样：

for(int i=0;i<m;i++)
    for(unsigned int j=0;j<c[i].size()-1;j++)//注意<size()-1
        for(unsigned int k=j+1;k<c[i].size();k++)
        {
            bool flag=0;//如果这c[i][j]~~c[i][k]这段ok的话，那后面的也都ok啦
            flag=getans(1,c[i][j],c[i][k],1,n);
            if(flag)
            {
                ans+=c[i].size()-k;//大家随便画个图体会一下为什么是size()-k
                break;
            }
        }

最后输出ans。

ans需不需要开long long我并没有测试，不过还是开了保险。

完整版：

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
using namespace std;
inline void in(int &p,char c=getchar())
{
    while(c<'0' or c>'9')
        c=getchar();
    p=0;
    while(c>='0' and c<='9')
        p=p*10+c-'0',c=getchar();
}
vector <int> c[52];
bool tree[200001<<2],di[200001];
long long ans;
void build(int rt,int l,int r)
{
    if(l==r)
    {
        tree[rt]=di[l];
        return ;
    }
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    tree[rt]=(tree[rt<<1] or tree[rt<<1|1]);
}
bool getans(int rt,int L,int R,int l,int r)
{
    if(L<=l and R>=r)
        return tree[rt];
    int m=(l+r)>>1;
    bool aans=0;
    if(L<=m and !aans)
        aans=getans(rt<<1,L,R,l,m);
    if(R>m and !aans)
        aans=getans(rt<<1|1,L,R,m+1,r);
    return aans;
}
int main()
{
    int n,m,maxm;
    in(n);in(m);in(maxm);
    for(int i=1;i<=n;i++)
    {
        int x,y;
        in(x);in(y);
        c[x].push_back(i);
        if(y<=maxm)
            di[i]=1;
    }
    build(1,1,n);
    for(int i=0;i<m;i++)
        for(unsigned int j=0;j<c[i].size()-1;j++)
            for(unsigned int k=j+1;k<c[i].size();k++)
            {
                bool flag=0;
                flag=getans(1,c[i][j],c[i][k],1,n);
                if(flag)
                {
                    ans+=c[i].size()-k;
                    break;
                }
            }
    cout<<ans;
    return 0;
}

转载于:https://www.cnblogs.com/syhien/p/7754997.html