生日礼物(尺取法、双指针)

题目链接

题目描述:

小西有一条很长的彩带,彩带上挂着各式各样的彩珠。已知彩珠有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;
}

 

posted @ 2021-02-20 15:20  阿涅—Rachel  阅读(59)  评论(0)    收藏  举报