【gzoj8.12综合三】选择客栈【数学处理】

最新推荐文章于 2024-05-27 19:36:05 发布

ssllyr

最新推荐文章于 2024-05-27 19:36:05 发布

阅读量125

点赞数

分类专栏：数学文章标签：贪心算法

本文链接：https://blog.csdn.net/dglyr/article/details/126311727

版权

数学专栏收录该内容

56 篇文章 0 订阅

订阅专栏

在这里插入图片描述

分析

看到数据范围，大概是一种 $n * l o g n$ 或者 $O (n)$ 的做法。

因为左右端点(客栈)只能枚举一个，所以考虑求出每一个右端点对答案的贡献。

我们枚举右端点，不妨贪心地想，他们去的咖啡店一定是在最靠近右端点的符合条件的咖啡店，位置记为mpos，因为这样左边能供选择的客栈才会尽可能多。mpos的位置是可以维护的。然后左端点(客栈)一定在mpos的左边，并且与右端点色调相同。第一个想法就是直接循环判断，但是不行，所以直接上空间换时间，用一个 $n u m [i]$ 记录在mpos之前的第 $i$ 种颜色出现的数量，然后如果有新的符合条件的咖啡店就更新mpos以及新旧mpos之间这段区间的num值，以及累加答案，如果加入的咖啡店不符合条件，就直接累加客栈数就行。

这样子其实是 $O (2 n)$ 的复杂度，没问题。具体细节不懂看代码吧

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

long long n,k,p,a[2000010],num[110000],ans,mpos,cost;

int main()
{
	cin>>n>>k>>p;
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&a[i],&cost);
		if(cost<=p)
		{
			for(int j=i;j>mpos;j--)
			{
				num[a[j]]++;
			}
			mpos=i;
			ans+=num[a[mpos]]-1;//-1是因为不能把自己统计进去 
		}
		else
		{
			ans+=num[a[i]];
		}
	} 
	cout<<ans;
	return 0;
}