[BZOJ3048][Usaco2013 Jan]Cow Lineup（单调队列）

题目描述

传送门

题解

首先离散化。
类似于单调队列的思想，一个头指针一个尾指针，向后移动的时候维护区间的种类数和它们的个数，保证区间种数-k=1。
边做边维护最大值。

代码

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

const int max_n=1e5+5;

int n,k,head,tail,kinds,Max;
int a[max_n],b[max_n],sum[max_n];
bool vis[max_n];

int main(){
//  freopen("lineup.in","r",stdin);
//  freopen("lineup.out","w",stdout); 
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;++i){
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    sort(b+1,b+n+1);
    int cnt=unique(b+1,b+n+1)-b-1;
    for (int i=1;i<=n;++i)
      a[i]=lower_bound(b+1,b+n+1,a[i])-b;

    head=1,tail=1;
    sum[a[1]]=1,kinds=1,vis[a[1]]=true;
    while (tail<n){
        while (vis[ a[tail+1] ]||(!vis[ a[tail+1] ]&&kinds-1<k)){
            tail++;
            if (!vis[ a[tail] ]){
                vis[ a[tail] ]=true;
                kinds++;
                sum[ a[tail] ]=1;
                Max=max(Max,sum[ a[tail] ]);
            }
            else{
                sum[ a[tail] ]++;
                Max=max(Max,sum[ a[tail] ]);
            }
            if (tail==n) break;
        }

        if (sum[ a[head] ]==1){
            kinds--;
            vis[ a[head] ]=false;
        }
        sum[ a[head] ]--;
        head++;
    }

    printf("%d\n",Max);
}

总结