洛谷 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;}
}
}