队列尺取法 -- F Groundhog Looking Dowdy

F Groundhog Looking Dowdy

题意：
有n天， 每天选一件衣服，有k件衣服可供选择，每件衣服都有个权值。在n天中有m天你要外出约会，求这m天中你衣服权值的极差（衣服权值的 （最大值 - 最小值））的最小值。

思路：
题意要求极差最小，这意味着m天衣服的权值间隙要尽可能的小，（比如最佳的情况是连续的），那我们就开个结构体数组，记录每件衣服的权值和所属日期。按衣服权值从小到大排序。依次将数据压道队列中，当刚好压到有m个不同日期的衣服时，把该天衣服的权值 - 队首第一个不同日期的最后一件衣服的权值（这句话有点绕，因为队列中同一天的衣服可能会连续存在，要贪心取该天的最后一件衣服的权值），遍历完数组ans取最小值就是答案。

code：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
struct node{
    int res, day;
} pi[maxn * 2];
int vis[maxn];
bool cmp(struct node a, struct node b){
    if(a.res == b.res) return a.day < b.day;
    return a.res < b.res;
}

int main(){
    int n, m, k;
    int cnt = 0;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++){
        scanf("%d", &k);
        for(int j = 1; j <= k; j++){
            scanf("%d", &pi[cnt].res);
            pi[cnt++].day = i;
           } 
    }
    
    if(m == 1) printf("0\n");   //m = 1，极差为0
    else{
   	
       sort(pi, pi + cnt, cmp);
    
       for(int i = 1; i <= n; i++) vis[i] = 0;
       queue<struct node> qu;
       while(qu.size()) qu.pop(); 
       int opt = 0;  //记录队列内衣服所属日期的种数
       int ans = 1 << 30;
    
       for(int i = 0; i < cnt; i++){
			if(opt < m) {
				qu.push(pi[i]);
				if(vis[pi[i].day] == 0) opt++, vis[pi[i].day]++;
				else vis[pi[i].day]++;
				
				if(opt == m){
					struct node p = qu.front();
					while(qu.size() && p.day == pi[i].day){
						vis[p.day]--;
						qu.pop();
					}
					
					do{
						p = qu.front();
						vis[p.day]--;
						if(vis[p.day] == 0) opt--;
						qu.pop();
					} while(qu.size() && p.day == qu.front().day);
					
					ans = min(ans, pi[i].res - p.res);
				}
			}			

			
		
        }
    
        printf("%d\n", ans);
    
    }
}