题目描述
小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有N个,分为K种。简单的说,可以将彩带考虑为x轴,每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠,但多个彩珠也可以出现在同一个位置上。
小布生日快到了,于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮,小西希望这一段彩带中能包含所有种类的彩珠。同时,为了方便,小西希望这段彩带尽可能短,你能帮助小西计算这个最短的长度么?彩带的长度即为彩带开始位置到结束位置的位置差。
输入格式:
第一行包含两个整数N, K,分别表示彩珠的总数以及种类数。接下来K行,每行第一个数为Ti,表示第i种彩珠的数目。接下来按升序给出Ti个非负整数,为这Ti个彩珠分别出现的位置。
输出格式:
输出应包含一行,为最短彩带长度。
先按坐标排序,然后把每个点放进队列里,在保证能拿到所有种类的珠子的情况下不断缩短长度,进行入队和出队操作。
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> using namespace std; #define maxn 1000005 int b[65];//每个种类珠子的个数 struct node{ int x,y;//x为位置,y是种类 }a[maxn],f[maxn]; int n,m,t,cnt,l,r,sum,ans; inline int read(){ int x=0,f=1;char ch=getchar(); while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } bool cmp(node a,node b){ return a.x<b.x; } void in(node a){ r++; f[r]=a; b[a.y]++; if(b[a.y]==1)sum++;//sum是此时队列中种类的个数 } void out(){ b[f[l].y]--; if(b[f[l].y]==0)sum--; l++; } int main(){ n=read();m=read(); for(int i=1;i<=m;i++){ t=read(); for(int j=1;j<=t;j++){ cnt++; a[cnt].x=read(); a[cnt].y=i; } } sort(a+1,a+cnt+1,cmp); l=1;r=0;ans=1e9; for(int i=1;i<=cnt;i++){ in(a[i]); while(sum==m){ ans=min(ans,f[r].x-f[l].x); out(); } } printf("%d",ans); return 0; }