Atcoder2292 Division into Two 【动态规划优化】

题目描述：

有一个包含N个不同整数的集合，其中第i小的元素是Si。我们希望把它划分成两个集合X和Y，使得：

集合X中任意两个相邻元素的差不小于A。
集合Y中任意两个相邻元素的差不小于B。

共有多少种不同的划分方式呢？输出答案对10^9+7取模后的结果。注意集合X或Y可以为空集。

1≤N≤10^5

解题思路：

先排序。
很好想到$O(n^2)$的dp，$f[i][j]$表示X集合最后一个元素为$s_i$，Y集合最后一个元素为$s_j$的方案数，考虑优化。

不妨假设 $A<B$，那么如果存在 $a_i-a_{i-2}<A$ 则无解（这一定要判，不然dp是错的）。
那么现在我们只用记录$f[i]$表示Y集合最后一个元素是$s_i$的方案数，考虑哪些$f[j]$可以转移过来，即是满足：

$$s_i-s_j\ge B$$ $$s_k-s_{k-1}\ge A，k\in[j+2,i-1]$$

相当于把$[j+1,i-1]$一段全部放入集合X。

注意可转移的范围是一个随 $i$ 右移的区间，用前缀和优化即可，时间复杂度 $O(n)$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll getint()
{
    ll i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=100005,mod=1e9+7;
int n,ans,f[N],g[N];
ll a[N],A,B;
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),A=getint(),B=getint();
    if(A>B)swap(A,B);
    for(int i=1;i<=n;i++)a[i]=getint();
    sort(a+1,a+n+1);
    for(int i=3;i<=n;i++)if(a[i]-a[i-2]<A){puts("0");return 0;}
    f[0]=g[0]=1;int l=0,r=0;
    for(int i=1;i<=n;i++)
    {
        while(r<i-1&&a[i]-a[r+1]>=B)++r;
        if(l<=r)f[i]=(g[r]+mod-(l?g[l-1]:0))%mod;
        g[i]=(g[i-1]+f[i])%mod;
        if(i>1&&a[i]-a[i-1]<A)l=i-1;
    }
    for(int i=n;i>=0;i--)
    {
        ans=(ans+f[i])%mod;
        if(i<n&&a[i+1]-a[i]<A)break;
    }
    cout<<ans<<'\n';
    return 0;
}