生日礼物
Description
小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有N个,分为K种。简单的说,可以将彩带考虑为x轴,每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠,但多个彩珠也可以出现在同一个位置上。 小布生日快到了,于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮,小西希望这一段彩带中能包含所有种类的彩珠。同时,为了方便,小西希望这段彩带尽可能短,你能帮助小西计算这个最短的长度么?彩带的长度即为彩带开始位置到结束位置的位置差。
Input
第一行包含两个整数N, K,分别表示彩珠的总数以及种类数。接下来K行,每行第一个数为Ti,表示第i种彩珠的数目。接下来按升序给出Ti个非负整数,为这Ti个彩珠分别出现的位置。
Output
应包含一行,为最短彩带长度。
Sample Input
6 3
1 5
2 1 7
3 1 3 8
Sample Output
3
HINT
有多种方案可选,其中比较短的是1~5和5~8。后者长度为3最短。
【数据规模】
对于50%的数据, N≤10000;
对于80%的数据, N≤800000;
对于100%的数据,1≤N≤1000000,1≤K≤60,0≤彩珠位置<2^31。
自己思考结果止步于80%的数据,O(nlogn+nlogk)……
正解解直接O(nlogk)复杂度碾压……
但是貌似咱的复杂度在BZOJ上可过?毕竟总时限……
思路:
不成熟的做法:
考虑到k很小,维护一个桶,离散化坐标后枚举右端点,维护最小左端点……
算了反正不成熟就不细说了。
正确的姿势:
维护一个根据位置从小到大排序的堆,初始时把每种颜色坐标最小的一个球扔进堆中。
然后不停弹出队首,每次弹出后,将被与弹出的球颜色相同的,坐标比当前坐标大的下一个球加入队列中,即保证队列中时刻都有k个元素,且这k个元素的颜色不同。
这样维护一个堆内最右的坐标,每次执行弹出加入后用最右坐标减去队首坐标得到的长度去更新答案即可~
为什么是对的呢?
首先这样的答案一定是合法的。
其次最优解在这种移动方式下一定会被访问到。
因此这样可以找到最优解。
#include<iostream>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int N=1000009;
const int K=69;
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0' || '9'<ch)ch=getchar();
while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
return x;
}
struct pr
{
int pos,c;
bool operator < (pr b)const
{
return pos>b.pos;
}
};
inline void chkmax(int &a,int b){if(a<b)a=b;}
inline void chkmin(int &a,int b){if(a>b)a=b;}
bool cmp(int a,int b){return a>b;}
int n,k,ans,r;
queue<int> vec[K];
priority_queue<pr> q;
int main()
{
n=read();
k=read();
for(int i=1,t;i<=k;i++)
{
t=read();
for(int j=1;j<=t;j++)
vec[i].push(read());
}
r=0;
for(int i=1;i<=k;i++)
{
q.push((pr){vec[i].front(),i});
chkmax(r,vec[i].front());
vec[i].pop();
}
ans=r-q.top().pos;
//printf("%d %d\n",q.top().pos,r);
for(int i=k,pos,col;i<=n;i++)
{
pr cur=q.top();
q.pop();
pos=cur.pos;
col=cur.c;
//printf("pop %d %d\n",pos,col);
if(vec[col].empty())
goto end;
chkmax(r,vec[col].front());
q.push((pr){vec[col].front(),col});
//printf("push %d %d\n",vec[col].front(),col);
vec[col].pop();
chkmin(ans,r-q.top().pos);
//printf("ans %d %d %d\n",ans,q.top().pos,r);
}
end:;
printf("%d\n",ans);
return 0;
}