长沙学院20级训练赛—商品出售

题目描述：

         在璃月庆典期间，派蒙碰到了一个穿着怪异的神秘商人，当派蒙好奇地查看神秘商人摊位上的东西时，神秘商人却说，如果你能回答这到题，那么这些东西就全送个你了。说完派蒙就决定试一试。

         题目是这样的，一共有n个商品，一共有A,B,C,D类，每类至少有1个，但不知道具体数量（即数量情况需考虑在内）。其中如果A类商品没有卖完客户就不会买B类商品，其中如果C类商品没有卖完客户就不会买D类商品。庆典只有k天，第一天一定能至少卖出一个商品（当然是托啦），除此以外每天卖出商品的数量可以为零，问所有商品在k天内卖完一共有多少情况，可以提前卖完(一天内卖出的顺序不算多种情况，如一天内卖出ABCD 和ACBD 为同种情况)。当然结果可能很大，请对1e9+7取模。

         这下派蒙犯了难，于是派蒙决定向你请教，你能告诉派蒙题目的答案吗？

输入描述：

         输入n和k(4<=n<=1e6,0<=k<=1e2)

输出描述：

         输出所有情况对1e9+7取模的结果。

示例1：

输入：

         4 2

输出：

         8

样列解释：

                  (ABCD) ( )

                  (A) (BCD)

                (AB) (CD)

                  (AC) (BD)

                  (ABC) (D)

                  (ACD) (B)

                  (C) (ABD)

                  (CD) (AB)

示例2：

输入：

         4 3

输出：

         27

---

题解：

        该题目可以考虑将AB分为一类，CD分为一类，那么设AB的数量设为i（2<=i<=n-2），那么CD的数量为n-i。再考虑内部的情况，AB数量情况一共有i-1种，CD数量情况一共有n-i-1种，所以当AB数量为i时ABCD的数量情况一共有(i-1)*(n-i-1)种，则可以用O(n)的时间复杂度求出所有ABCD的数量情况。

         然后就是考虑将AB和CD分配给k天，当AB数量为i时，因为A一定在B前出售，所以顺序唯一确定，可以直接使用公式C(i+k-1,k-1)(球同 盒不同 可为空同理)求出，CD同理可得C(n-i+k-1,k-1)。且每一天的出售顺序不考虑在内，则将AB和CD分配给k天分别独立，所以当AB数量为i时可以求得所有情况为C(i+k-1,k-1)*C(n-i+k-1,k-1)*(i-1)*(n-i-1)。这里求取组合数使用的是预处理，所以总时间复杂度为O(nlogn)，加上前面的O(n),时间复杂度为O(n+nlogn)，本题n<=1e6不会超时。

        而题目中第一天一定卖出一个商品则可考虑只出售k-1天，那么第一天一定不出售商品的情况就求出来了，然后用当前所有情况减去即可。

---

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll mod=1e9+7;
const int N=1e6+7;
int fact[N],infact[N];
int n,k;
ll ans;
int qmi(int a, int k, int p)    // 快速幂模板
{
    int res = 1;
    while (k)
    {
        if (k & 1) res = (ll)res * a % p;
        a = (ll)a * a % p;
        k >>= 1;
    }
    return res;
}
// 预处理阶乘的余数和阶乘逆元的余数
void yu()
{
    fact[0] = infact[0] = 1;
    for (int i = 1; i < N; i ++ )
    {
        fact[i] = (ll)fact[i - 1] * i % mod;
        infact[i] = (ll)infact[i - 1] * qmi(i, mod - 2, mod) % mod;
    }
}
int C(int a,int b,int p)
{
    int res=(ll)fact[a]*infact[a-b]%p*infact[b]%p;
    return res;
}
int get(int a,int k)
{
    int res=(ll)C(a+k-1,k-1,mod)*C(n-a+k-1,k-1,mod)%mod*((a-1)%mod)%mod*((n-a-1)%mod)%mod;
    return res;
}

int main()
{
    yu();
    scanf("%d%d,",&n,&k);
    for(int i=2;i<=n-2;i++)
        ans=(ans+get(i,k)-get(i,k-1)+mod)%mod;
    printf("%lld\n",ans);
    system("pause");
    return 0;
}