Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others)
题意是给出一个数组,求区间的个数,每个区间里的最大值和最小值的差值要小于K,区间长度不限
Problem Description
Tom owns a company and he is the boss. There are n staffs which are numbered from 1 to n in this company, and every staff has a ability. Now, Tom is going to assign a special task to some staffs who were in the same group. In a group, the difference of the ability of any two staff is less than k, and their numbers are continuous. Tom want to know the number of groups like this.
Input
In the first line a number T indicates the number of test cases. Then for each case the first line contain 2 numbers n, k (1<=n<=100000, 0<k<=10^9),indicate the company has n persons, k means the maximum difference between abilities of staff in a group is less than k. The second line contains n integers:a[1],a[2],…,a[n](0<=a[i]<=10^9),indicate the i-th staff’s ability.
Output
For each test,output the number of groups.
Sample Input
2 4 2 3 1 2 4 10 5 0 3 4 5 2 1 6 7 8 9
Sample Output
5 28HintFirst Sample, the satisfied groups include:[1,1]、[2,2]、[3,3]、[4,4] 、[2,3]
题意是给出一个数组,求区间的个数,每个区间里的最大值和最小值的差值要小于K,区间长度不限
这道题主要是用两个优先列队来做,用两个优先列队来维护区间的最大值和最小值,然后每次把数组的中数据放进去都要判断一次最大值和最小值,符合题意的就加上,不符合题意的就删掉,其中用multiset可以删除指定元素。(嘛~都是听学长给我讲的,这种用法现在还是挺陌生的,得去看看有关STL的书才行)
#include<iostream>
#include<set>
#include<queue>
using namespace std;
int num[100010];
int main()
{
int t;
cin>>t;
while(t--)
{
int n,k;
cin>>n>>k;
for(int i=0;i<n;i++) cin>>num[i];
multiset<pair<int,int> > q1,q2;
long long ans=0;
int nut=0;
for(int i=0;i<n;i++)
{
q1.insert(make_pair(num[i],i));
q2.insert(make_pair(-num[i],i));
while(-(*q1.begin()).first-(*q2.begin()).first>=k)
{
q1.erase(make_pair(num[nut],nut));
q2.erase(make_pair(-num[nut],nut));
nut++;
}
ans+=i-nut+1;
}
cout<<ans<<endl;
}
return 0;
}