HDU 5793-A Boring Question（数学化简，快速幂，逆元）

HDU 5793-A Boring Question

本篇文章部分参考

[https://www.cnblogs.com/Sunshine-tcf/p/5737627.html]

题目原址

[http://acm.hdu.edu.cn/showproblem.php?pid=5793]

题意

给你两个数 n m ，求解它们带入公式后的答案。

题解

公式十分复杂，刚开始甚至没看懂是什么意思，但知道需要找规律或者化简。
后来去看了一下其他博客的题解(链接在上)，找到了官方题解，果真是化简，就是有点点复杂。。。

1. 公式本身：本质上是很多个(懒得想多少个) m 个组合数的乘积 的和，给出样例的表达
   $$f(x)=C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=0,k_3=0)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=1,k_2=0,k_3=0)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=1,k_3=0)}+$$$$C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=1,k_2=1,k_3=0)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=0,k_3=1)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=1,k_2=0,k_3=1)}+$$$$C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=1,k_3=1)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=1,k_2=1,k_3=1)}$$但其实可以发现，当且仅当$k_1,k_2,k_3$是广义降序的时候乘积和不为0，上面式子可以删去一些项得到$$C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=0,k_3=0)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=0,k_3=1)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=0,k_2=1,k_3=1)}+C_{k_2}^{k_1}\cdot {C_{k_3}^{k_2}}_{(k_1=1,k_2=1,k_3=1)}$$既然是要广义降序，那么公式本身就可以化简为
   $$\sum _{0\le k_1,k_2,\cdot \cdot \cdot k_m\le n}\prod _{1\le j<m}{C}_{k_{j+1}}^{k_j}=\sum _{k_m=0}^n\sum _{k_{m-1}=0}^{k_m}\cdot \cdot \cdot \sum _{k_1=0}^{k_2}\prod _{1\le j<m}{C}_{k_{j+1}}^{k_j}$$
2. 巧提公因式：很显然，前面的几项如$k_m,k_{m-1}$确定后，后面的每项联乘中的$C_{k_m}^{k_{m-1}}$都确定了，那么就可以单独提出来，用样例做示例：
   $${C_{k_3}^{k_2}}_{(k_2=0,k_3=0)}({C_{k_2}^{k_1}}_{(k_1=0)})+{C_{k_3}^{k_2}}_{(k_2=0,k_3=1)}({C_{k_2}^{k_1}}_{(k_1=0)})+{C_{k_3}^{k_2}}_{(k_2=1,k_3=1)}({C_{k_2}^{k_1}}_{(k_1=0)}+{C_{k_2}^{k_1}}_{(k_1=1)})$$对于公式来说，提公因式后：
   $$\sum _{k_m=0}^n\sum _{k_{m-1}=0}^{k_m}(C_{k_m}^{k_{m-1}}\sum _{k_{m-2}=0}^{k_{m-1}}(C_{k_{m-1}}^{k_{m-2}}\cdot \cdot \cdot (\sum _{k_1=0}^{k_2}{C}_{k_2}^{k_1})\cdot \cdot \cdot ))$$所以我觉得上面的官方图解有点小瑕疵？我也不深究了，接下来可以发现最内项为$2^{k_2}$，化简成：
   $$\sum _{k_2=0}^{k_3}{C}_{k_3}^{k_2}\cdot 2^{k_2}=3^{k_3}$$如此下去直到：
   $$\sum _{k_m=0}^n{m}^{k_m}=1+m^1+m^2+\cdot \cdot \cdot +m^n=\frac{m^{n+1}-1}{m-1}$$简洁得不行，用快速幂算一下分子，算一下分母的逆元即可。

实现

#include <stdio.h>
typedef long long ll;
int i, j, k;
int x, y;
const int mod = (int)1e9 + 7;
ll q_pow(int a, int b) {
    ll ans = 1;
    ll p = a;
    while (b) {
        if (b & 1)
            ans = ans * p%mod;
        p = p * p%mod;
        b >>= 1;
    }
    return ans;
}
int inv(int a,int b,int&x,int&y) {
    if (b == 0) {
        x = 1; y = 0;
        return a;
    }
    int ans = inv(b, a%b, y, x);
    y -= a / b * x;
    return ans;
}
ll solve(int n, int m) {
    ll ans = 1;
    inv(m - 1, mod, x, y);
    x = ((x%mod) + mod) % mod;//将x化为最小非负整数解
    ans = (q_pow(m, n + 1) - 1)*x % mod;
    return ans;
}
int main() {
    int t, n, m;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &m);
        printf("%lld\n", solve(n, m));
    }
}