2011年noipc++提高组 第2题

题目描述
丽江河边有
n家很有特色的客栈，客栈按照其位置顺序从
1 到 n 编号。每家客栈都按照某一种色调进行装饰（总共 k 种，用整数 0~ 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
说明
【输入输出样例说明】

2 人要住同样色调的客栈，所有可选的住宿方案包括：住客栈①③，②④，②⑤，④⑤，但是若选择住
45 号客栈的话，
45 号客栈之间的咖啡店的最低消费是 4，而两人能承受的最低消费是 3元，所以不满足要求。因此只有前 3 种方案可选。
【数据范围】
对于 30% 的数据，有 n≤100 ；
对于50% 的数据，有n≤1,000 ；
对于100% 的数据，有 2≤n≤200,000,0

看到这道题目，我首先想到的是用数组f[i][j]来存贮从i到j的最小消费
由此可以得到动态转移方程
f[i][j]=min(f[i][j-1],b[j]); ps:b[j]表示第j家咖啡店的最低消费
从而可以到代码

#include <iostream>

using namespace std;


int a[200000],b[200000];
int f[20000][20000];
int main()
{
    int n,k,p;
    cin>>n>>k>>p;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i]>>b[i];
    }
    for(int i=1;i<=n;i++)
    {
        f[i][i]=b[i];
        for(int j=i+1;j<=n;j++)
        {
          f[i][j]=min(f[i][j-1],b[j]);
        }
    }
    int num=0;
    for(int i=1;i<=n;i++)
       for(int j=i+1;j<=n;j++)
       {
        if(a[i]==a[j]&&f[i][j]<=p)
        {
            num++;
           }
       }
       cout<<num<<endl;
    return 0;

}

但是根据数据范围和题目要求，这个代码是只能得60分的 1：时间超时 2：f 数组内存不够。所以要进行优化
观察样例 ： 0 1 0 1 1 之后发现只要枚举1遍 客栈i 记录离他最接近的合法点 记录在这个点之前有多少个相同颜色的客栈
那么 方案数=目前的方案数+相同颜色客栈个数

#include <iostream>

using namespace std;


int a[200055],b;
int f[200005][55];
int zui[200055];
int main()
{
    int n,k,p;
    cin>>n>>k>>p;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i]>>b;
        for(int j=0;j<k;j++)
        {
            f[i][j]=f[i-1][j];
        }
        f[i][a[i]]++;
        if(b<=p)
        {
            zui[i]=i;
        }else{
            zui[i]=zui[i-1];
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        ans+=f[zui[i]][a[i]];
        if(zui[i]==i) ans--;
    }

    cout<<ans<<endl;
    return 0;

}