#队列#codevs 2422 洛谷 2564 生日礼物

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sugar_free_mint/article/details/82263139

题目

在一个数轴上选定一段区间,使区间包含所有种类的数,且区间长度最短


分析

准确来说,这道题其实是暴力+队列,特别明显的是,按数的大小排序,然后模拟,时间复杂度 O ( n l o g n + n )


代码

#include <cstdio>
#include <queue>
#include <algorithm>
struct rec{int x,rk;}a[1000001];
int n,m,kind[61]; std::queue<int>q;
int in(){
    int ans=0; char c=getchar();
    while (c<48||c>57) c=getchar();
    while (c>47&&c<58) ans=ans*10+c-48,c=getchar();
    return ans;
}
bool cmp(rec x,rec y){return x.x<y.x;}
int main(){
    n=in(); m=in(); int tot=0,ans=2147483647;
    for (register int i=1;i<=m;i++){
        int k=in();
        for (register int j=1;j<=k;j++)
            a[++tot]=(rec){in(),i};
    }
    std::sort(a+1,a+1+n,cmp); int sum=0;//快排
    for (register int i=1;i<=n;i++){
        q.push(i);
        kind[a[i].rk]++;
        if (kind[a[i].rk]==1) sum++;//新的种类
        while (sum==m){
            int k=q.front();
            if (ans>a[q.back()].x-a[k].x) ans=a[q.back()].x-a[k].x;//求最小值
            q.pop(); kind[a[k].rk]--;
            if (kind[a[k].rk]==0) sum--;//少了一种
        }
    }
    printf("%d",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/sugar_free_mint/article/details/82263139