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

目录

  传送门
  加强版传送门
  解题思路
  AC代码

传送门

加强版传送门

---

解题思路

首先很显然，对于同一种色调的客栈来说，从小到大枚举左端点 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

作者：尹昱钦
欢迎任何形式的转载，但请务必注明出处。
限于本人水平，如果文章和代码有表述不当之处，还请不吝赐教。