洛谷 P2564 [SCOI2009]生日礼物

洛谷 P2564 [SCOI2009]生日礼物

---

题目

题目背景

四川2009NOI省选

题目描述

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

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

输入输出格式

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

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

输入输出样例

输入样例#1：

6 3
1 5
2 1 7
3 1 3 8

输出样例#1：

3

说明

【样例说明】

有多种方案可选，其中比较短的是1~5和5~8。后者长度为3最短。

【数据规模】

对于50%的数据， N≤10000；

对于80%的数据， N≤800000；

对于100%的数据，1≤N≤1000000，1≤K≤60，0≤Ti<2^31。

---

题解

开个队列和hash，分别记录入队的元素和队列中各颜色出现的次数，然后就是一顿枚举直至cnt不能达到m就好了

---

代码

#include<cstdio>
#include<cstdlib>
using namespace std;

int n,m,q,tot,l,r,cnt,ans;
int a[1000005],b[1000005],hash[65];

int readln()
{
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x;
}

int min(int x,int y){return x<y?x:y;}

void qsort(int l,int r)
{
    int i=l,j=r,mid=a[rand()%(r-l+1)+l],t;
    do {
        while (a[i]<mid) i++;
        while (a[j]>mid) j--;
        if (i<=j) {
            t=a[i];a[i]=a[j];a[j]=t;
            t=b[i];b[i]=b[j];b[j]=t;
            i++;j--;
        }
    } while (i<=j);
    if (i<r) qsort(i,r);
    if (l<j) qsort(l,j);
}

int main()
{
    n=readln();m=readln();
    for (int i=1;i<=m;i++)
    {
        q=readln();
        for (int j=1;j<=q;j++) a[++tot]=readln(),b[tot]=i;
    }
    qsort(1,tot);
    ans=0x7fffffff;l=1;
    while (r<=tot)
    {
        while (cnt!=m&&r<tot)
        {
            hash[b[++r]]++;
            if (hash[b[r]]==1) cnt++;
        }
        if (cnt==m)
        {
            ans=min(ans,a[r]-a[l]);
            hash[b[l]]--;
            if (hash[b[l]]==0) cnt--;
            l++;
        }
        else {printf("%d",ans);return 0;}
    }
}