bzoj1293 [SCOI2009] 生日礼物 队列

题目描述

小西有一条很长的彩带，彩带上挂着各式各样的彩珠。已知彩珠有N个，分为K种。简单的说，可以将彩带考虑为x轴，每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠，但多个彩珠也可以出现在同一个位置上。

小布生日快到了，于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮，小西希望这一段彩带中能包含所有种类的彩珠。同时，为了方便，小西希望这段彩带尽可能短，你能帮助小西计算这个最短的长度么？彩带的长度即为彩带开始位置到结束位置的位置差。

输入格式：

第一行包含两个整数N, K，分别表示彩珠的总数以及种类数。接下来K行，每行第一个数为Ti，表示第i种彩珠的数目。接下来按升序给出Ti个非负整数，为这Ti个彩珠分别出现的位置。

输出格式：

输出应包含一行，为最短彩带长度。

先按坐标排序，然后把每个点放进队列里，在保证能拿到所有种类的珠子的情况下不断缩短长度，进行入队和出队操作。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 1000005
int b[];//每个种类珠子的个数
struct node{
    int x,y;//x为位置，y是种类
}a[maxn],f[maxn];
int n,m,t,cnt,l,r,sum,ans;
inline int read(){
    int x=,f=;char ch=getchar();
    while(ch>''||ch<''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
bool cmp(node a,node b){
    return a.x<b.x;
}
void in(node a){
    r++;
    f[r]=a;
    b[a.y]++;
    if(b[a.y]==)sum++;//sum是此时队列中种类的个数
}
void out(){
    b[f[l].y]--;
    if(b[f[l].y]==)sum--;
    l++;
}
int main(){
    n=read();m=read();
    for(int i=;i<=m;i++){
        t=read();
        for(int j=;j<=t;j++){
            cnt++;
            a[cnt].x=read();
            a[cnt].y=i;
        }
    }
    sort(a+,a+cnt+,cmp);
    l=;r=;ans=1e9;
    for(int i=;i<=cnt;i++){
        in(a[i]);
        while(sum==m){
            ans=min(ans,f[r].x-f[l].x);
            out();
        }
    }
    printf("%d",ans);
    return ;
}