生日礼物(尺取法、双指针)
题目描述:
小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有N个,分为K种。简单的说,可以将彩带考虑为x轴,每一个彩珠有一个对应的坐标(即位置)。某些坐标上可以没有彩珠,但多个彩珠也可以出现在同一个位置上。 小布生日快到了,于是小西打算剪一段彩带送给小布。为了让礼物彩带足够漂亮,小西希望这一段彩带中能包含所有种类的彩珠。同时,为了方便,小西希望这段彩带尽可能短,你能帮助小西计算这个最短的长度么?彩带的长度即为彩带开始位置到结束位置的位置差。
Input:
第一行包含两个整数N, K,分别表示彩珠的总数以及种类数。
接下来K行,每行第一个数为Ti,表示第i种彩珠的数目。接下来按升序给出Ti个非负整数,为这Ti个彩珠分别出现的位置。
Output:
应包含一行,为最短彩带长度。
Example:
input:
6 3
1 5
2 1 7
3 1 3 8
output:
3
思路:尺取法,通过双指针标记满足条件时左右端点。记录每个位置上彩珠的种类,然后对位置排序,利用一个数组记录当前区间每个颜色的出现次数,然后我们不断推进右端点,当左右端点区间颜色种类达到 k 的时候,再推进左端点,一直推进到左右端点区间颜色种类数恰好不是 k 个,一直重复此过程。
#include<bits/stdc++.h> #include<iostream> #include<cctype> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #include<queue> #include<stack> using namespace std; typedef unsigned long long ull; typedef long long ll; typedef pair<ll,ll> pi; #define IOS std::ios::sync_with_stdio(false) #define ls p<<1 #define rs p<<1|1 #define mod 1000000000 + 7 #define PI acos(-1.0) #define INF 1e18 #define N 2000000 + 5 /*********************Code*********************/ ll n,k,sum[N],m,cnt; struct node{ ll col,x; bool operator< (const node &a)const{ return x < a.x; } }e[N]; int main(void){ IOS; cin>>n>>k; for(ll i = 1;i <=k;i++){ cin>>m; for(ll j = 0;j < m;j++){ cin>>e[++cnt].x; e[cnt].col = i; } } sort(e+1,e+cnt+1); ll l = 1,ans = INF,res = 0; for(ll i = 1;i <=cnt;i++){ if(!sum[e[i].col]) res++; sum[e[i].col]++; while(res==k){ ll temp = e[i].x - e[l].x; ans = min(ans,temp); sum[e[l].col]--; if(!sum[e[l].col]) res--; l++; } } cout<<ans<<endl; return 0; }