算法:双指针
先把物体的位置和种类记录在一个结构体中,按位置从小到大排序
用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; }