描述
小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有N个,分为K种。简单的说,可以将彩带考虑为x轴,每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠,但多个彩珠也可以出现在同一个位置上。 小布生日快到了,于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮,小西希望这一段彩带中能包含所有种类的彩珠。同时,为了方便,小西希望这段彩带尽可能短,你能帮助小西计算这个最短的长度么?彩带的长度即为彩带开始位置到结束位置的位置差。
输入
输入文件gift.in,第一行包含两个整数N, K,分别表示彩珠的总数以及种类数。接下来K行,每行第一个数为Ti,表示第i种彩珠的数目。接下来按升序给出Ti个非负整数,为这Ti个彩珠分别出现的位置。
输出
输出文件gift.out应包含一行,为最短彩带长度。
样例输入 [复制]
6 3
1 5
2 1 7
3 1 3 8
样例输出 [复制]
3
提示
有多种方案可选,其中比较短的是15和58。后者长度为3最短。 【数据规模】 对于50%的数据, N≤10000; 对于80%的数据, N≤800000; 对于100%的数据,1≤N≤1000000,1≤K≤60,0≤Ti<231。
简单分析
看到区间条件和最小,不难想到队列;
最开始直接用二分+队列验证,复杂度 O((30+logn)*n) 不知怎么,没有过。。
updated 11.12 1e6的数据nlogn显然不一定能过得去。。
后来发现可以贪心优化
基于一个本质:队列里面一定要有所有种类,对于一个满足条件的情况,可以优化:
删除队尾或队首多余的颜色
注:1.多余:队中此颜色出现数大于1 2.先按照位置大小排序
相当于不断地扩展队列,如果满足情况看是否能优化解 。
#include<bits/stdc++.h>
using namespace std;
#define gc getchar()
#define in rd()
inline int rd(){
int num=0,f=1;
char c=gc;
for(;!isdigit(c);c=gc)if(c=='-')f=-1;
for(;isdigit(c);c=gc)num=(num<<1)+(num<<3)+(c^48);
return num*f;
}
const int maxn=1e6+10;
struct atom{
int loc,col;
bool operator <(const atom &t) const{
return loc<t.loc;
}
}a[maxn],q[maxn];
int n,top=0,head,tail,K;
int amt[100];
signed main(){
n=in,K=in;
for(int i=1;i<=K;++i){
int t;t=in;
for(int j=1;j<=t;++j){
++top;
a[top].loc=in;
a[top].col=i-1;
//这个i或者i-1都不影响
}
}
sort(a+1,a+n+1);
head=0,tail=-1;
top=0;
int ans=(1<<31)-1;
for(int i=1;i<=n;++i){
q[++tail]=a[i];
if(!amt[a[i].col])++top;
++amt[a[i].col];
if(top==K){
while(amt[q[head].col]>1){
--amt[q[head].col];
++head;
}
ans=min(ans,q[tail].loc-q[head].loc);
}
}
printf("%d",ans);
return 0;
}