[agc009c]Division into Two

来源:互联网 发布:音频合并时间软件 编辑:程序博客网 时间:2024/03/29 07:40

前言

DP题,还可以,不是很难。

题目大意

把一堆互不相同的数分成两个集合X和Y,其中一个集合X任意两数之差不小于A,另一集合Y任意两数之差不小于B。
问方案数,集合可以为空。

DP

把这些数排序。
接着添加第0个数无穷小以及第n+1个数无穷大。
我们假设A>=B。
设f[i]表示分配好了前i个数,第i个数去X集合的方案。
转移显然要找到上一个位置j。
第一种情况j=i-1,只要a[i]-a[i-1]>=A就能转移。
其余情况,首先a[i]-a[j]>=A,容易发现可以维护一个指针来维护最大的j。
其次可以维护一个指针k表示(k,i)这一段可以全部分到Y集合。
那么可以得到转移区间。
当然仍然还有一些要考虑,比如上一个X两边的Y。
首先因为A>=B,因此上一个X左边的如果无法选入Y,一定也不能选入X,因此我们只需要判断能否选入Y。
对于左右无法同时选入Y的决策j,即a[j+1]-a[j-1]

#include<cstdio>#include<algorithm>#define fo(i,a,b) for(i=a;i<=b;i++)using namespace std;typedef long long ll;const int maxn=100000+10,mo=1000000007;ll a[maxn],A,B;int tree[maxn*4],f[maxn];int i,j,k,l,r,t,n,m,ans;void change(int p,int l,int r,int a,int b){    if (l==r){        (tree[p]+=b)%=mo;        return;    }    int mid=(l+r)/2;    if (a<=mid) change(p*2,l,mid,a,b);else change(p*2+1,mid+1,r,a,b);    tree[p]=(tree[p*2]+tree[p*2+1])%mo;}int query(int p,int l,int r,int a,int b){    if (a>b) return 0;    if (l==a&&r==b) return tree[p];    int mid=(l+r)/2;    if (b<=mid) return query(p*2,l,mid,a,b);    else if (a>mid) return query(p*2+1,mid+1,r,a,b);    else return (query(p*2,l,mid,a,mid)+query(p*2+1,mid+1,r,mid+1,b))%mo;}int main(){    scanf("%d%lld%lld",&n,&A,&B);    if (A<B) swap(A,B);    fo(i,1,n) scanf("%lld",&a[i]);    a[0]=-A;    a[n+1]=a[n]+A;    f[0]=1;    change(1,0,n,0,f[0]);    l=0;r=0;    fo(i,1,n+1){        while (a[i]-a[r+1]>=A) r++;        if (i>=2&&a[i-1]-a[i-2]<B) l=i-2;        if (a[i]-a[i-1]>=A) f[i]=f[i-1];        (f[i]+=query(1,0,n,l,min(i-2,r)))%=mo;        if (i<=n&&a[i+1]-a[i-1]>=B) change(1,0,n,i,f[i]);    }    ans=f[n+1];    (ans+=mo)%=mo;    printf("%d\n",ans);}
原创粉丝点击