其他分享
首页 > 其他分享> > 洛谷 P1311 [NOIP2011 提高组] 选择客栈 & P6032 选择客栈 加强版(双指针)

洛谷 P1311 [NOIP2011 提高组] 选择客栈 & P6032 选择客栈 加强版(双指针)

作者:互联网

传送门

加强版传送门


解题思路

首先很显然,对于同一种色调的客栈来说,从小到大枚举左端点 L,符合要求的客栈 R~N 一定是满足 R 单调递增的。
而且选择的咖啡店也是随着 L 的增加间断着递增。
所以就可以双指针(准确说是多指针)将时间复杂度降至O(n)。
双倍经验双倍快乐

AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=2e6+5;
int n,k,p,a[maxn],b[maxn],c[maxn],mid=n+1,r[maxn];
long long ans;
vector<int> num[10005];
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>k>>p;
    for(int i=0;i<=k;i++) num[i].push_back(0),r[i]=1;
    for(int i=1;i<=n;i++){
    	cin>>a[i]>>b[i];
    	num[a[i]][0]++;
    	num[a[i]].push_back(i);
	}
	for(int i=1;i<=n;i++) if(b[i]<=p){
		mid=i;
		break;
	}
    for(int i=1;i<=n;i++){
    	while(mid<=n&&(mid<i||b[mid]>p)) mid++;
    	if(mid>n) break;
    	while(r[a[i]]<=num[a[i]][0]&&i>=num[a[i]][r[a[i]]]) r[a[i]]++;
    	while(r[a[i]]<=num[a[i]][0]&&num[a[i]][r[a[i]]]<mid) r[a[i]]++;
    	ans+=num[a[i]][0]-r[a[i]]+1;
   	}
   	cout<<ans;
    return 0;
}

//NOIP2011提高组Day1 t2

标签:洛谷,加强版,int,客栈,mid,num,maxn,include
来源: https://www.cnblogs.com/yinyuqin/p/15111520.html