洛谷P2564 [SCOI2009]生日礼物(队列模拟)

[SCOI2009]生日礼物

题目描述
小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有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 。

代码

#include <cstdio>
#include <algorithm>
#include <queue>
#define N 1000005
using namespace std;

struct arr
{
    int num,pos;
}a[N];
int f[100],n,k;
queue<int> q;

int minf(int x, int y) {return x<y?x:y;}
int cmp(arr x, arr y) {return x.pos<y.pos;}
int main()
{
    int tot = 0;
    scanf("%d%d", &n, &k);
    for (int i = 1; i <= k; i++)
    {
        int x;
        scanf("%d", &x);
        for (int j = 1; j <= x; j++)
        {
            scanf("%d", &a[++tot].pos);
            a[tot].num = i;
        }
    }
    sort(a + 1, a + tot + 1,cmp);
    int cnt = 0;
    int i = 1;
    int ans = 1e9;
    for (int i = 1; i <= n; i++)
    {
        q.push(i);
        if (!f[a[i].num]) cnt++;
        f[a[i].num]++;
        while (q.size() && cnt == k)
        {
            ans = minf(ans, a[i].pos - a[q.front()].pos);
            int now = q.front();
            q.pop();
            f[a[now].num]--;
            if (!f[a[now].num]) cnt--;
        }
    }
    printf("%d", ans);
}

猜你喜欢

转载自blog.csdn.net/zhanghaoxian1/article/details/81783434
今日推荐