【dp】cf1188C-Array Beauty

题意

给定一个序列，一个序列的美丽值为这个序列中差值最小的那个数

现在给你一个长度为n数组，让你求这个数组所有长度为k的子序列的美丽值的和

数组中每个数最大为1e5

题解

首先考虑到美丽值与序列的顺序无关，我们先把数组排一个序

然后一般做法就是枚举，这里显然不行

想到每个数最大为1e5，那就说明不同的美丽值个数最多1e5个

现在问题就是，给你一个美丽值，问你能不能求出有多少个序列满足这个要求

这样还是特别抽象，我们考虑转化一下答案，我们求每个美丽值的个数乘以美丽值再加起来

不如求美丽值大于某个数的序列个数，然后再加起来。

现在问题就是，给你一个美丽值，问你有多少个序列的美丽值大于等于这个美丽值

考虑dp一下，dp【i】【j】表示长度为i，以第j个数结尾且满足美丽值大于要求的序列个数

那么dp【i】【j】=dp【i-1】【d】，d为所有满足要求的位置，我们可以用一个前缀和来优化

再考虑d在外层循环中是不减的，所以可以o（1）的求出符合条件的d

所以子问题的答案就是所有的dp【k】【i】的和

一次dp 的复杂度，kn，要dp多少次呢？最多1e5次，但是考虑一下，差值最大为1e5，长度为k，所以美丽值最大为1e5/（k-1），复杂度就是（1e5n）

代码

#include <bits/stdc++.h>

using namespace std;
const int maxn=1000+10;
const int p=998244353;
int dp[maxn][maxn];//dp[i][j]长度为i，以j结尾的数量

int n,k;
int a[maxn];
int s[maxn][maxn];
int ans;
int solve(int x)
{
    for(int i=1;i<=k;i++){
        for(int j=1;j<=n;j++){
            dp[i][j]=0;
            s[i][j]=0;
        }
    }
    for(int j=1;j<=n;j++)dp[1][j]=1;
    for(int i=1;i<=k;i++){
        int l=1;
        for(int j=1;j<=n;j++){
            while(a[j]-a[l]>=x)l++;
            dp[i][j]=(dp[i][j]+s[i-1][l-1])%p;
            s[i][j]=(s[i][j-1]+dp[i][j])%p;
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)ans=(ans+dp[k][i])%p;
    return ans;
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    sort(a+1,a+n+1);
    for(int i=1;i<=(a[n])/(k-1);i++){
        ans=(ans+solve(i))%p;
        //cout<<ans<<endl;
    }
    cout<<ans<<endl;
    return 0;
}