经典算法题：选择客栈问题

来源：选择客栈_牛客博客

 题目描述：

丽江河边有 n 家很有特色的客栈，客栈按照其位置顺序从 1 到n 编号。每家客栈都按照某一种色调进行装饰（总共 k 种，用整数 0 ~ k-1 表示） ，且每家客栈都设有一家咖啡店，每家咖啡店均有各自的最低消费。
两位游客一起去丽江旅游，他们喜欢相同的色调，又想尝试两个不同的客栈，因此决定分别住在色调相同的两家客栈中。晚上，他们打算选择一家咖啡店喝咖啡，要求咖啡店位于两人住的两家客栈之间（包括他们住的客栈） ，且咖啡店的最低消费不超过 p。
他们想知道总共有多少种选择住宿的方案，保证晚上可以找到一家最低消费不超过 p 元的咖啡店小聚。

输入

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

输出

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

样例输入

5 2 3

0 5

1 3

0 2

1 4

1 5

样例输出

3

提示

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


【数据范围】
对于 30%的数据，有 n ≤ 100；
对于 50%的数据，有 n ≤ 1,000；
对于 100%的数据，有 2 ≤ n≤ 200,000，0 < k ≤ 50，0 ≤ p ≤ 100， 0 ≤ 最低消费 ≤ 100。

思路和代码：

其实是贪心算法：从左往右遍历，记录每一种color对应的计数 count[color] 和最后一次出现的索引 last[color]，因此只有当前color和之前color的计数才可能增加组合数count[color]，问题是从last[color]及之后是否出现过最低消费least小于等于p的情况（前提条件），如果出现则更新sum[color] = count[color]，表示之前出现的count[color]和当前color的索引i 可以 增加组合数count[color]，否则的话直接增加sum[color]，因为last[color]后没有最小消费小于等于p的情况。

#include<iostream>
#include<vector>
using namespace std;
int main(){
    int n,k,p;
    cin >> n >> k >> p;
    vector<int> last(k), sum(k), count(k);
    int record = -1, ans = 0;
    int color, least;
    for(int i = 0; i < n; i++)
    {
        cin >> color >> least;
        if(least <= p)
            record = i;
        if(record >= last[color])
            sum[color] = count[color];
        ans += sum[color];
        last[color] = i;
        count[color]++;
    }
    cout << ans << endl;
}

真是 神奇 Amazing！

只考虑同一色调color的情况：

第一种情况：当前索引i 色调color 之前的计数count[color]

...      last[color] ...     record ...     i

record记录最近出现的满足最低消费的索引record那么如果大于等于last[color]，则可以更新

sum[color] = count[color]，并增加组合数sum[color]（count[color]可以作为第一家客栈的树木）；

第二种情况：当前索引i 色调color 之前的计数count[color]

...      record  ...      last[color]  ...     i

record后所有的最低消费都不能满足要求

如果record小于last[color]，则sum[color]不变，因为last[color]之后的计数并不能用于第一家客栈，只能用于第二家；