BZOJ1293 || 洛谷P2564 [SCOI2009]生日礼物【单调队列】

Description

小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有N个,分为K种。简单的说,可以将彩带考虑为x轴,每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠,但多个彩珠也可以出现在同一个位置上。 小布生日快到了,于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮,小西希望这一段彩带中能包含所有种类的彩珠。同时,为了方便,小西希望这段彩带尽可能短,你能帮助小西计算这个最短的长度么?彩带的长度即为彩带开始位置到结束位置的位置差。

Input

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

Output

应包含一行,为最短彩带长度。

HINT

有多种方案可选,其中比较短的是1~5和5~8。后者长度为3最短。
【数据规模】
对于50%的数据, N≤10000;
对于80%的数据, N≤800000;
对于100%的数据,1≤N≤1000000,1≤K≤60,0≤彩珠位置<2^31。


题目分析

将所有彩珠按坐标升序排序
用cnt数组记录当前每个颜色出现了多少次

从头开始遍历每一个彩珠
先将当前彩珠从队尾入队
然后检查队首颜色出现次数,若出现次数大于1则不断弹出队首

若当前队列中恰好每种颜色都有
则计算队尾与队首颜色坐标之差,更新ans

由于颜色范围很小,连离散化都不用


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

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int maxn=1000010;
int n,m;
struct node{int x,num;}rem[maxn];
int cnt[maxn],sum,tot;
int q[maxn],ll,rr;
int ans=2e9;
bool cmp(node a,node b){return a.x<b.x;}

int main()
{
    n=read();m=read();
    for(int i=1;i<=m;++i)
    {
        int k=read();
        while(k--) 
        rem[++tot].x=read(),rem[tot].num=i;
    }

    sort(rem+1,rem+1+n,cmp); ll=rr=1;
    for(int i=1;i<=n;++i)
    {
        if(cnt[rem[i].num]==0)sum++;
        cnt[rem[i].num]++; q[rr++]=i;
        while(ll<rr&&cnt[rem[q[ll]].num]>1) --cnt[rem[q[ll]].num],++ll;
        if(sum==m) ans=min(ans,rem[q[rr-1]].x-rem[q[ll]].x);
    }
    printf("%d",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/niiick/article/details/80819316