算法:双指针
先把物体的位置和种类记录在一个结构体中,按位置从小到大排序
用l,r分别代表两个指针,在[l,r]的物品数量区间内看是否满足cnt==k
代码
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp(aa,bb) make_pair(aa,bb)
#define _for(i,b) for(int i=(0);i<(b);i++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,b,a) for(int i=(b);i>=(a);i--)
#define mst(abc,bca) memset(abc,bca,sizeof abc)
#define X first
#define Y second
#define lowbit(a) (a&(-a))
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
typedef long double ld;
const int N=1000010,K=65;
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-6;
struct node{
ll place;
int kind;
}a[N];
int st[K];
bool cmp(node x,node y){
return x.place<y.place;
}
void solve(){
int n,k;cin>>n>>k;
int cnt=0;
rep(i,1,k){
int t;cin>>t;
rep(j,1,t){
cin>>a[cnt].place;
a[cnt++].kind=i;
}
}
sort(a,a+n,cmp);
cnt=0;
ll ans=(1ll<<31)+1;
int l=0,r=-1;
while(r<n-1){
if(!st[a[++r].kind]) cnt++;
st[a[r].kind]++;
if(cnt==k){
ans=min(ans,a[r].place-a[l].place);
// cout<<a[r].place<<" "<<a[l].place<<"\n";
while(cnt==k){
st[a[l].kind]--;
if(!st[a[l++].kind]) cnt--;
if(cnt==k) ans=min(ans,a[r].place-a[l].place);
// cout<<a[r].place<<" "<<a[l].place<<"\n";
}
}
}
cout<<ans<<"\n";
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
#ifdef DEBUG
freopen("F:/laji/1.in", "r", stdin);
// freopen("F:/laji/2.out", "w", stdout);
#endif
// int t;cin>>t;while(t--)
solve();
return 0;
}

京公网安备 11010502036488号