Bzoj4352 Tower

题目描述

给定N个数字，可以组成排列{a_i}.有N！种排列，求这其中满足对于i(2<=i<=N)有{a_i-a_i-1<=d}的排列方案数;

30分打法

暴力搜索，开long long，30分；

AC打法

设f[i]为第i块的方案数；首先将长度升序排序，考虑当前要放的砖块长度为i，那么如果将该砖块放在最低层，那么肯定有f[i-1]种方案(乘法原理:f[i-1]*1);接下来考虑放在其他砖块上的方案数，找出满足a[i]-a[j]<=d(也就是说:a[j]>=a[i]-d)的sum{j}，那么这里又增加了f[i-1]*sumj种方案(乘法原理),在这里可以先用O(n)预处理出sumj，或者可以用lower_bound求出pos，然后sumj=n-pos；则f[i]=f[i-1]*(i-pos+1);

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

const int Maxn=600000+10;
const int Mod=1000000009;
typedef long long LL;
int a[Maxn];
LL ans=1;
int n,d;

int main()
{
    scanf("%d%d",&n,&d);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)
    {
        int pos=lower_bound(a+1,a+i,a[i]-d)-a;
        ans=ans*(i-pos+1)    //这里用ans代替了f[i-1]，以节省空间
        ans=ans%Mod;    //这里要处理一下，防止爆long long
    }
    printf("%lld\n",ans);
    return 0;
}