AtCoder Grand Contest 009 C - Division into Two 动态规划

题意

给出n个数S[1..n]和两个数A，B。问有多少种把n个数分成两个集合的方案，满足第一个集合中任意两个数差的绝对值不小于A，第二个集合中任意两个数差的绝对值不小于B。
n<=100000,S[i],A,B<=10^18

分析

一开始想的线段树优化建图+2-SAT求方案数。。。
首先要来找一波性质：
先把S从小到大排序，然后不妨设$A<B$。
若存在相邻$i>2$使得$S[i]-S[i-2]<A$，则无解。因为这三个数显然不能在同一个集合里面，然而只有两个集合，故方案数为0。
也就是说，只有相邻的两个数可能会受到第一个集合的限制。
这就可以开始dp了：
设f[i]表示前i个数，且第i个数必然在第二个集合中的方案数。
考虑f[i]可以由哪些位置转移过来。
假设f[j]可以转移到f[i]，必然有$s[i]-s[j]>=B$，且$[j+1,i-1]$这一段中的数全放在第一个集合不会产生冲突。
显然能转移到的必然是连续的一段，只要用前缀和优化一下就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=100005;
const int MOD=1000000007;

int n,f[N],g[N];
LL a,b,s[N];

int main()
{
    scanf("%d%lld%lld",&n,&a,&b);
    if (a>b) swap(a,b);
    for (int i=1;i<=n;i++) scanf("%lld",&s[i]);
    sort(s+1,s+n+1);
    for (int i=3;i<=n;i++) if (s[i]-s[i-2]<a) {puts("0");return 0;}
    f[0]=g[0]=1;int p=0,lim=0;
    for (int i=1;i<=n;i++)
    {
        while (p<i&&s[i]-s[p+1]>=b) p++;
        if (lim<=p) f[i]=(g[p]+MOD-(lim?g[lim-1]:0))%MOD;
        g[i]=(g[i-1]+f[i])%MOD;
        if (i>1&&s[i]-s[i-1]<a) lim=i-1;
    }
    int ans=0;
    for (int i=n;i>=0;i--)
    {
        (ans+=f[i])%=MOD;
        if (i<n&&s[i+1]-s[i]<a) break;
    }
    printf("%d",ans);
    return 0;
}