[BZOJ4402]Claris的剑（组合数学）

=== ===

这里放传送门

=== ===

题解

这题比较关键的一点就是注意到题目里说两把剑本质不同当且仅当存在一种数字的出现次数不同。也就是说这个东西跟数字的出现顺序是无关的。那么我们可以考虑确定一种排序方式让每一种本质不同的解都唯一对应一种看起来比较好做的序列。

因为每两个元素相差必须是1，通过多画几个试试观察可以发现每一种合法解都可以排列成一种这样的序列：设序列中最大元素是k，首先把1..k按照顺序排成一串，然后在中间的k-1个空挡里插入形如(i+1,i)这样的数对。例如1 2 3 2 1 2 3 2 1这组解就可以排列成1 2 1 2 1 2 3 2 3这个样子。
注意到这一点就有思路了。如果固定了长度n和最大出现的数字k，那么实际上就是要在k-1个空位置中插入(n-k)/2个数对。插入的数具体是什么是由它的位置决定的。那么这就是一个经典的插板法可以解决的问题了。
有一个问题是如果n-k是奇数怎么办，实际上多出来的那个数字是可以把它固定被放在最后一个位置上的。比如说是1 2 3 2 1 2 3 2 1 2，按照上面的方法“排序”以后可以变成1 2 1 2 1 2 3 2 3 2。可以发现这种变换方式对于每一种本质不同的数列是唯一的。所以直接下取整就可以了。
这样的话我们就有了一种$O(nk)$的求解方式，就是先枚举数列长度再枚举出现的最大数字。也就是下面代码里面注释掉的那部分。

接下来考虑优化这个做法。一个比较好想到的思路就是因为在做组合数的时候是连续枚举的，可以考虑利用公式$C(n,m)=C(n-1,m)+C(n-1,m-1)$来进行合并。然后通过把表打出来看一看观察可以发现当m相同的时候n的变化是有规律的。。然后反正凑一凑然后凑出$\sum\limits_{i=m}^rC(i,m)$这样能连续合并的东西大概就是这个样子就可以$O(n)$了。。
具体那个规律是啥ATP找的规律看起来非常naive。。就不好意思拿出来误导人了QAQ。。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const long long Mod=1e9+7;
int n,m,Lim;
long long ans,mul[2000020],inv[2000020];
long long C(int n,int m){
    if (n<m) return 0;
    long long tmp=mul[n];
    tmp=tmp*inv[m]%Mod;
    tmp=tmp*inv[n-m]%Mod;
    return tmp;
}
/*long long get(int n,int m){return C(n+m-1,m-1);}
long long calc(int n){
    long long sum=0;
    for (int i=2;i<=m;i++){
        if (n<=i) break;
        sum=(sum+get((n-i)/2,i-1));
    }
    printf("\n");
    return sum;
}*/
int main()
{
    scanf("%d%d",&n,&m);
    if (m==0){printf("0\n");return 0;}
    Lim=max(n,m)+10;
    mul[0]=1;inv[1]=1;inv[0]=1;
    for (int i=1;i<=Lim;i++) mul[i]=mul[i-1]*i%Mod;
    for (int i=2;i<=Lim;i++) inv[i]=(Mod-Mod/i)*inv[Mod%i]%Mod;
    for (int i=1;i<=Lim;i++) inv[i]=inv[i]*inv[i-1]%Mod;
    Lim=min(m-2,n-2);
    for (int i=0;i<=Lim;i++){
        int lim=(n-i)/2,Max;
        Max=i+lim-1;
        ans=(ans+C(Max+1,i+1)*2)%Mod;
        if ((n-i)%2==0) ans=(ans-C(Max,i))%Mod;
    }
//  for (int i=0;i<=Lim;i++) ans=(ans-C(i,i))%Mod;
    ans=(ans-(Lim+1)+min(n,m))%Mod;
    ans=(ans+Mod)%Mod;
    printf("%I64d\n",ans);
    return 0;
}