NOIP2011选择客栈

选择客栈

【题目分析】
这是一道比较好玩的题目，情景感比较强。而问题的实质就是：
1、将同种颜色的客栈进行处理。
2、对于一个客栈，它有两种情况，费用超过p、费用不超过p。
3、超过p的找之后相同颜色且费用不超过p的。
4、不超过p的与之后相同颜色的随意组合。
既然是批量处理，自然想到了前缀和。（处理3的时候用到了后缀和）
想清楚了情况，那就开始模拟吧。

【代码】

#define N 52
#define M 200005
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int cnt[N],L[M],R[M];
int n,k,p;
long long ans=0;
struct node{
    int c,val;
}a[M];
int main(){
    int i,j;
    scanf("%d %d %d",&n,&k,&p);
    for(i=1;i<=n;i++)scanf("%d %d",&a[i].c,&a[i].val);
    for(j=0;j<k;j++){
        int cnt=0;
        L[0]=R[n+1]=0;
        for(i=1;i<=n;i++){
            L[i]=L[i-1];
            if(a[i].c==j)L[i]++;
        }
        for(i=n;i>=1;i--){
            if(a[i].c==j)cnt++;
            R[i]=R[i+1];
            if(a[i].val<=p)R[i]=cnt;
        }
        for(i=1;i<=n;i++){
            if(a[i].c!=j)continue;
            if(a[i].val<=p)ans+=L[n]-L[i];
            else ans+=R[i+1];   
        }
    }cout<<ans<<endl;
    return 0;
}