NOIP2011 选择客栈

题目

vijos1737

题解

用ans[i]记录1~i中能与i组合的客栈总数

那么对于色调相同且距离最近的i,j (i < j)，若i~j中有可去的咖啡店，那么ans[j]即为j以前色调与j相同的客栈总数；若i~j中无可去的咖啡店，那么ans[]=ans[i]

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath> 
#define N 200005
using namespace std;

int n,k,p,out;
int color,cost;
int last[55];//last[i]记录颜色i上一次出现的位置 
int sum[N];//将可以去的咖啡店记为1，不能去的记为0，sum[i]记录前缀和 
int ans[N];//ans[i]记录1~i中能与i组合的客栈总数 
int num[55];//num[i]记录颜色为i的客栈在之前出现过几次 

int main()
{
    scanf("%d%d%d",&n,&k,&p);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&color,&cost);
        if(cost<=p) sum[i]=sum[i-1]+1;
        else sum[i]=sum[i-1]; 
        if(!last[color]) last[color]=i;
        else
        {
            if(sum[i]-sum[last[color]-1]) ans[i]=num[color];
            else ans[i]=ans[last[color]];
            last[color]=i;
        }
        num[color]++;
    }
    for(int i=1;i<=n;i++) out+=ans[i];
    printf("%d",out);
    return 0;
}

转载于:https://www.cnblogs.com/XYZinc/p/7551312.html