Gym 100703I 尺取法

点击打开链接

题意：给n组和每组有多少人，现在要求你从每组中选出一个人，然后这些人的最大值与最小值的差最小

思路：将所有的人按照权值从大到小排序，然后尺取知道每一组都出现一个人，然后比较差值即可，尺取的话直接改一改挑战程序设计的例题就可以了

#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=200010;
int B[210][1010],A[210],cnt[maxn],vis[maxn];
struct edge{
    int val,id,pos;
}num[maxn],ans[210];
bool cmp(const edge &a,const edge &b){return a.val<b.val;}
bool cmp1(const edge &a,const edge &b){return a.id<b.id;}
int main(){
    int n;
    while(scanf("%d",&n)!=-1){
        for(int i=0;i<n;i++) scanf("%d",&A[i]);
        for(int i=0;i<=n;i++) cnt[i]=0,vis[i]=0;
        int len=0;
        for(int i=0;i<n;i++) for(int j=0;j<A[i];j++) scanf("%d",&B[i][j]),num[len].val=B[i][j],num[len].id=i,num[len++].pos=j;
        sort(num,num+len,cmp);
        int max1=inf,s=0,t=0,sum=0,pos1,pos2;
        while(1){
            while(t<len&&sum<n){
                cnt[num[t].id]++;
                if(cnt[num[t].id]==1) sum++;
                t++;
            }
            if(sum<n) break;
            if(max1>num[t-1].val-num[s].val){
                max1=num[t-1].val-num[s].val;
                pos1=s;pos2=t-1;
            }
            if(max1==0) break;
            cnt[num[s].id]--;
            if(cnt[num[s].id]==0) sum--;s++;
        }
        int len1=0;
        for(int i=pos1;i<=pos2;i++){
            if(vis[num[i].id]==0){
                vis[num[i].id]=1;
                ans[len1].id=num[i].id;
                ans[len1++].pos=num[i].pos;
            }
        }
        sort(ans,ans+len1,cmp1);
        printf("%d\n",max1);
        for(int i=0;i<len1-1;i++) printf("%d ",ans[i].pos+1);
        printf("%d\n",ans[len1-1].pos+1);
    }
    return 0;
}