NOIP 2011 选择客栈

评测传送

第一次做的时候读错题目，以为咖啡店也和他们住的客栈同一色调，全WA（哭），惨痛的教训！

对于每一种色调，我们先求一遍后缀和。
然后对于不能作为咖啡店的这种色调的客栈，它可以与离它最近的咖啡店后面的所有同一色调的客栈组成一种方案。而可以作为咖啡店的这种客栈，则可以与它后面所有的相同客栈组成一种方案。
对于每一种色调都这样处理就好了。
时间复杂度：O(n*k)

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#define LL long long
using namespace std;
const int N=200009;
int n,k,p;
LL sl[N],sr[N],ans; 
int c[N],a[N];
int main()
{
    freopen("hotel.in","r",stdin);
    scanf("%d%d%d",&n,&k,&p);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&c[i],&a[i]);
    }
    for(int co=0;co<k;co++)
    {
        for(int i=n;i>=1;i--)
        {
            sr[i]=sr[i+1];
            if(c[i]==co) sr[i]++;
        }
        LL s=0,ss=0;
        for(int i=1;i<=n;i++)
        {
            if(a[i]<=p)
            {
                ss+=s*sr[i];
                if(c[i]==co)
                {
                    ss+=sr[i]-1;
                }
                s=0;
            }
            else
            if(c[i]==co) s++;

        }
        ans+=ss;
    }
    printf("%lld\n",ans);
    return 0;
}