日常降智。
不过还是第一次和 2700 的题正解这么近呢……
由于排序后不影响答案,而且直觉告诉我们排序后会更好做,不妨排个序。
直觉告诉我们,变成求最小差 (ge v) 的方案数会比最小差 (=v) 的方案数好算。
问题就变成如何求最小差 (ge v) 的方案数。
令 (f_{i,j}) 表示前 (i) 个数中选了 (j) 个,且 (i) 被选了的方案数。有 (f_{i,1}=1)。
转移:(f_{i,j}=sumlimits_{a_i-a_kge v}f_{k,j-1})。
很明显可以前缀和+双指针优化。
时间复杂度 (O(nka_max))。然后我就自闭了。
%了一发 wqy 的题解,太神了吧……
其实是最小差的最大值达不到 (a_max-a_min),而只有 (frac{a_max-a_min}{k-1})。(抽屉原理)
复杂度立刻降到 (O(nkfrac{a_max}{k-1})=O(na_max))。
看来……会很多的 DP 套路优化,发掘不了性质,还是只能被吊打……
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int maxn=1111,mod=998244353;
#define MP make_pair
#define PB push_back
#define lson o<<1,l,mid
#define rson o<<1|1,mid+1,r
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(int i=(a);i>=(b);i--)
#define MEM(x,v) memset(x,v,sizeof(x))
inline ll read(){
char ch=getchar();ll x=0,f=0;
while(ch<'0' || ch>'9') f|=ch=='-',ch=getchar();
while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
return f?-x:x;
}
int n,k,a[maxn],f[maxn][maxn],s[maxn][maxn],ans;
int main(){
n=read();k=read();
FOR(i,1,n) a[i]=read();
sort(a+1,a+n+1);
FOR(x,1,(a[n]-a[1])/(k-1)){
FOR(i,0,n) FOR(j,0,k) f[i][j]=s[i][j]=0;
FOR(i,1,n) f[i][1]=1,s[i][1]=i;
FOR(j,2,k){
int cur=0;
FOR(i,1,n){
while(cur<i && a[i]-a[cur]>=x) cur++;
if(cur && a[i]-a[cur]<x) cur--;
f[i][j]=s[cur][j-1];
s[i][j]=(s[i-1][j]+f[i][j])%mod;
}
}
ans=(ans+s[n][k])%mod;
}
printf("%d
",ans);
}