NOIP2011 选择客栈

题目：http://www.luogu.org/problem/show?pid=1311#
分析：
用sum[i][j]记录前i个中颜色为j的有多少个，可以O(nk)预处理出来。
当w[i]<=p时，i对答案的贡献就是sum[i-1][color[i]]。
当w[i]>p时，i对答案的贡献就是sum[last][color[i]]，last是最后一个满足w[i]<=p的元素序号。
总时间复杂度是O(n+nk)

代码：

#include <cstdio>
#include <algorithm>
using namespace std;
const int Tmax=200005,Tmax2=55;
int n,m,p,sum[Tmax][Tmax2],w[Tmax],f,last,color[Tmax];
void work()
{
    int i;
    for(i=1;i<=n;i++)
    {
        if(w[i]<=p)
        {
            last=i;
            f+=sum[i-1][color[i]];
        }
        else f+=sum[last][color[i]];

    }
    printf("%d",f);
    return;
}
int main()
{
    int i,j,a,b;
    scanf("%d%d%d",&n,&m,&p);
    for(i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        w[i]=b;
        color[i]=a;
        sum[i][a]++;
        for(j=0;j<m;j++)
          sum[i][j]+=sum[i-1][j];
    }
    work();
    return 0;
}