hdu(5289)——Assignment（muliset的使用)

该博客介绍了如何利用STL中的muliset数据结构解决一道编程题目，题意涉及将具有特定技能值差异限制的人分组。通过比较muliset中最大值和最小值的差值来判断是否满足条件，并使用头尾指针l和r进行分组操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：

现在有n个人，然后每个人都有一个技能值a[i]。如果几个人是在同一组的话，那么他们之间的技能值的差值小于k并且标号连续。

然后问你可以进行分组的方式有几种。

思路：

1）因为数据范围过大，所以不可能直接暴力，for两遍

2）用STL中的multiset，我们每次往后找并与集合中的max和min值比较差值，若差值小于k，那么则是成立的。并且设置两个头尾指针l，r来表示。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<set>
using namespace std;
#define maxn 100010
#define inf 99999999
typedef __int64 ll;
int a[maxn];
multiset<int> ms;
int main(){
	int T;
	int n,k;
	scanf("%d",&T);
	while(T--){
		memset(a,0,sizeof(a));
		ms.clear();
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		ll ans=n;
		int l=1,r=2;
		int smax,smin;
		ms.insert(a[1]);
		while(1){
			if(ms.size()){
				smax=*ms.rbegin();			//注意这个东西只有当ms.size()不为0的时候才有意义，所以要写在里面； 
				smin=*ms.begin();
				if(abs(a[r]-smax)<k&&abs(a[r]-smin)<k){
					ans+=(ll)ms.size();				//每次都加上当前的size 
					ms.insert(a[r]);
					r++;
					if(r==n+1) break;
				}
				else{
					ms.erase(ms.find(a[l]));	//这里删除的只是一个数，因为ms.find()相当于一个迭代器指针，指向那个数
					l++;
				}
			}
			else{
				l=r;
				ms.insert(a[r]);
				r++;
				if(r==n+1) break;
			}
		}
		printf("%I64d\n",ans);
	}
}
/*
1
6 2
1 1 3 3 3 4
*/