【59.61.214.20:3001/No.1354】生日礼物 题解

题目描述

小西有一条很长的彩带，彩带上挂着各式各样的彩珠。已知彩珠有N个，分为K种。简单的说，可以将彩带考虑为x轴，每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠，但多个彩珠也可以出现在同一个位置上。 小布生日快到了，于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮，小西希望这一段彩带中能包含所有种类的彩珠。同时，为了方便，小西希望这段彩带尽可能短，你能帮助小西计算这个最短的长度么？彩带的长度即为彩带开始位置到结束位置的位置差。

输入

第一行包含两个整数N, K，分别表示彩珠的总数以及种类数。接下来K行，每行第一个数为Ti，表示第i种彩珠的数目。接下来按升序给出Ti个非负整数，为这Ti个彩珠分别出现的位置。

输出

包含一行，为最短彩带长度。

样例输入

6 3
1 5
2 1 7
3 1 3 8

样例输出

3

数据范围限制

对于50%的数据， N≤10000；
对于80%的数据， N≤800000；
对于100%的数据，1≤N≤1000000，1≤K≤60，0≤彩珠位置<2^31。

提示

 

​【样例解释】
有多种方案可选，其中比较短的是1~5和5~8。后者长度为3最短。

 最直观的的想法是：把彩珠位置存入数组的下标里，从位置1一直往下找，直到所有种类都包含进去，记录答案，再从位置2一直往下找……最后选出最优答案。但这个思路漏洞百出：一个位置可以系上好多个彩珠，也就是一个数组元素可能对应多个种类彩珠，要么开二维数组，要么用链表。而且彩珠位置可达到2^31，数组开那么大肯定爆炸。还有一个问题就是每次从头开始找，需两重循环，数据大点，时间会超限。时间上的问题我们先撇开不谈，怎么优化空间呢？

我们可以用离散化的思想，依据所有彩珠位置的大小顺序给它们编个号。举个例子，假设样例中三种彩珠分别为彩珠1、彩珠2、彩珠3，彩珠1在位置5，彩珠2位置1和位置7各有一个，彩珠3分布在位置1、3和8，位置从小到大排序后，是1（有2个彩珠在这个位置）、3、5、7、8，分别编上1、2、3、4、5，这就把这些位置映射到数组上了。然后我们注意到彩珠位置对应的种类还没处理，就再用一个数组（或把两个数组封装在结构体里）存储排序后的彩珠对应的种类，也就是（2、3）、3、1、2、3。一山不容二虎，位置1两个种类2、3不能挤在一个数组元素里，我们可以用链表存储。毕竟指针在竞赛中有一定风险，又发现k值不超过60，此时，用二进制数来“存储”这些信息，太合适了。种类数为i，就把二进制数的第i位设为1。举个例子，种类2、3，用110这个二进制数来表示（二进制数越往左位数越大，本来最右边是第0位，但我看着不顺眼，所以就当最右边是第1位了）。具体怎么实现，参见代码。

现在只剩下时间太慢的问题了。我们仔细观察，能够得知两重循环其实做了一些不必要的工作。当我们找到一种符合题意的情况时，不用从下一个位置重新开始找，因为中间那些位置我们已经找过了。如样例，到位置5的彩珠时，种类全了，由于同种类的彩珠太多可能浪费了长度，我们先把位置1给舍弃，看看种类全不全，结果不全，这时我们只把一个彩珠2和一个彩珠3舍弃了，其它的彩珠还在，也就是说从位置5往下找，当种类再次齐全时，再次重复以上步骤即可。

补充一个猥琐的细节：对于1<<k，当k>31时，要写成1LL<<k，否则会错。我因为这个错误浪费了两个小时，惨绝人寰啊。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
struct Orap{
	long long pos,ord;
}a[1000001],h[1000001];
long long head,ans;
int s[101];
int n,k,t,cnt,ph,p;
int cmp(Orap x,Orap y)
{
	if(x.pos!=y.pos) return x.pos<y.pos;
	else return x.ord<y.ord;
}
int main()
{
	freopen("gift.in","r",stdin);
	freopen("gift.out","w",stdout);
	cin>>n>>k;
	n=0;
	ans=1e18;
	ph=1;
	for(int i=1;i<=k;i++)
	{
		cin>>t;
		for(int j=1;j<=t;j++)
		{
			cin>>a[++n].pos;
			a[n].ord=i;
		}	
	}
	sort(a+1,a+1+n,cmp);
	for(int i=1;i<=n;i++)
	{
		if(a[i].pos!=a[i-1].pos)
			h[++cnt].pos=a[i].pos;
		h[cnt].ord=h[cnt].ord|(1LL<<a[i].ord-1);
	}
	for(int i=1;i<=cnt;i++)
	{
		head=h[ph].pos;
		if(p!=2)
			for(int j=0;j<k;j++)
				if((h[i].ord&(1LL<<j)))
					s[j+1]++;
		p=0;
		for(int j=1;j<=k;j++)
			if(!s[j])
				p=1;
		if(!p||h[i].pos-head>ans)
		{
			for(int j=0;j<k;j++)
				if((h[ph].ord&(1LL<<j)))
					s[j+1]--;
			ph++;
			ans=min(ans,h[i].pos-head);
			i--;
			p=2;
		}
	}
	cout<<ans;
}