【NOIP2011】选择客栈

题目描述

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

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

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

输入输出格式

输入格式:

\(n+1\)行。

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

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

输出格式:

一个整数,表示可选的住宿方案的总数。

输入输出样例

输入样例#1:

5 2 3 
0 5 
1 3 
0 2 
1 4 
1 5 

输出样例#1:

3

分析

判断一对客栈的可行性只与它们之间的是否存在最低消费不大于\(p\)的客栈,因此我们用\(last\)变量记录上一个满足条件的客栈,\(pos[i]\)记录上一个色调为\(i\)的客栈,\(cnt[i]\)记录色调\(i\)的客栈数量。只要\(last\)\(pos[i]\)的后面,则当前客栈与之前的所有客栈的组合都可行,并以\(bfe[]\)数组记下此时的\(cnt[]\)。否则,当前客栈与上次\(bfe[]\)记录的客栈中间的客栈是不可行的,这时\(bfe\)数组便派上用场了,直接加上\(bfe[i]\)即可。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#define il inline
#define re register
#define maxn 200011
#define mymax(a,b) a>b?a:b
#define mymin(a,b) a<b?a:b
#define tie0 cin.tie(0),cout.tie(0)
#define fastio ios::sync_with_stdio(false)
using namespace std;
typedef long long ll;

template<typename T>inline void read(T &x){
    T f=1;x=0;char c;
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())x=x*10+(c^48);
    x*=f;
}

int n,k,p;
int last,sum;
int cnt[51],pos[51],bfe[51];

int main(){
    fastio,tie0;
    int a,b;
    cin>>n>>k>>p;
    for(int i=1;i<=n;++i){
        cin>>a>>b;
        if(b<=p) last=i;
        if(last>=pos[a]) bfe[a]=cnt[a];
        sum+=bfe[a];
        pos[a]=i;
        cnt[a]++;
    }
    printf("%d",sum);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/hlw1/p/11228750.html