【P1311】选择客栈

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

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

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

输入输出格式
输入格式：
共n+1 行。

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

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

输出格式：
一个整数，表示可选的住宿方案的总数。

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

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

【数据范围】

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

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

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

思路：两个人要住在两家颜色相同的酒店，且两家酒店之间至少有一家咖啡店可以让他们相聚，所以只需统计每家咖啡店之前最近的可相聚咖啡店的位置和每家咖啡店之前各种颜色客栈的数量。最后在总计总数的时候，i作为被选择的咖啡店较后的咖啡店，被选择另一家店的方案数即为i之前最近的可相聚咖啡店之前所有同色咖啡店的数量。
特别的，如果i本身就可以作为相聚地点，就需要将统计的同颜色的咖啡店数量减一，因为两个人无法住在同一家店中。

代码：

#include<bits/stdc++.h>
using namespace std;
struct dd{
    int price,color;
    //用于记录在那之前各种颜色的客栈出现的次数
    int icolor[55];
    //用于记录上一个可行消费咖啡店的位置
    int point;
}q[200005];
int main()
{
    int n,p,k,sum=0;
    scanf("%d %d %d",&n,&k,&p);
    //第一家咖啡店特殊处理
    scanf("%d %d",&q[1].color,&q[1].price);
    for(int i = 0; i < k; i++)
        q[1].icolor[i]=0;
    q[1].icolor[q[1].color]=1;
    if(q[1].price<=p)
        q[1].point=1;
    else
        q[1].point=0;
    //剩下的咖啡店
    for(int i = 2; i <= n; i++)
    {
        scanf("%d %d",&q[i].color,&q[i].price);
        for(int j = 0; j < k; j++)
            q[i].icolor[j]=q[i-1].icolor[j];
        q[i].icolor[q[i].color]++;
        if(q[i].price<=p)
            q[i].point=i;
        else
            q[i].point=q[i-1].point;
    }
    for(int i = 1; i <= n; i++)
    {
        //sum用于统计总数量
        if(q[i].point == i)
            sum = sum + q[i].icolor[q[i].color]-1;
        else
            sum = sum + q[q[i].point].icolor[q[i].color];
    }
    printf("%d\n",sum);
    return 0;
}