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。
思路: 用stl的sort排序, 再用双指针遍历寻找。
过程中出现的错误:本题用双指针时应从l = 0, r = 0开始,这时数组可以从node[ 1 ],开始起步,否则很难处理第一个结点上的小球的种类。
源代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 1e6 + 7;
const int inf = 0x3f3f3f3f;//表示无穷大
struct Node {
int x, col;//x表示球的位置,col代表球的种类
bool operator<(const Node& A)const {
return x < A.x;
}
}node[N];
int main() {
int n, k, num;
int tota = 1 , q;
int nu[66] = { 0 };
int ans = inf;
cin >> n >> k;
for (int i = 1; i <= k; i++) {
cin >> num; //每种种类的球的个数
for (int j = 0; j < num; j++) {
cin >> q; //输入种球的位置分布
node[tota++] = { q,i }; //结构体赋值的方法(以前没见过)
}
}
sort(node + 1, node + n + 1);//根据结构体内的bool类型进行sort排序得到得到x递增的node[]数组
int l = 0, r = 0, tep = 0;
while (l < n) {
l++;
if (l > 0) {
nu[node[l - 1].col] --;//在[ l-1 , r ]中要少了一个node[l-1]颜色小球的个数,所以要-1
if (nu[node[l - 1].col] == 0)tep--;//如果该颜色的小球数量变成0了那颜色种类就要-1
}
while (tep < k && r < n) {
r++;
if (nu[node[r].col] == 0)tep++;//如果r没有右移之前[ l , r ]上该种颜色的小球个数为0,那么右移后小球颜色种类+1
nu[node[r].col]++; //无论小球种类有无变化,r++后的r所在的小球颜色的个数都要+1;
}
if(k==tep)ans = min(ans, node[r].x-node[l].x);//如果取长度的较小值
}
cout << ans << '\n';
}