少见地秒了这道1700,要是以后都这样就好了....
Problem - F - Codeforces
题意:
给定一个数列,让你在这个数列里找一个大小为M的子集,使得极差不超过M
思路:
子集,不是子序列,说明和顺序无关,因此可以考虑排序
观察一下样例可知,排序后我们可以双指针一下,然后方案数就是区间map之积
Code:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mxn=2e5+10;
const int mxe=2e5+10;
const int mod=1e9+7;
map<int,int> mp;
int N,M;
int len=0;
int a[mxn],b[mxn],c[mxn],pre[mxn];
int ksm(int a,int b,int mod){
int res=1ll;
while(b){
if(b&1) res=(res*a)%mod;
a=(a*a)%mod;
b>>=1;
}
return res;
}
void solve(){
mp.clear();
len=0;
cin>>N>>M;
set<int> S;
for(int i=1;i<=N;i++){
cin>>a[i];
S.insert(a[i]);
mp[a[i]]++;
}
for(auto it:S) b[++len]=it;
for(int i=1;i<=len;i++) c[i]=mp[b[i]];
pre[0]=1;
for(int i=1;i<=len;i++) pre[i]=pre[i-1]*c[i]%mod;
int r=1;
int ans=0;
for(int l=1;l<=len;l++){
while(r<=len&&b[r]-b[l]<M&&r-l+1<=M) r++;
if(r-1-l+1==M&&b[r-1]-b[l]<M) ans+=pre[r-1]*ksm(pre[l-1],mod-2,mod)%mod;
}
cout<<ans%mod<<'\n';
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int __=1;cin>>__;
while(__--)solve();return 0;
}