[BZOJ 2655]calc

Description

题库链接

给出 $A,n,p$ ，让你在模 $p$ 意义下求所有序列 $a$ 满足“长度为 $n$ 且 $a_i\in[1,A]$ ，并且对于 $i\neq j,a_i\neq a_j$”的价值和。

一个序列的价值定义为 $\prod\limits_{i=1}^n a_i$ 。

$1\leq A\leq 10^9,1\leq n\leq 500,p\leq 10^9$ 并且 $p$ 为素数， $p>A>n+1$ 。

Solution

考虑朴素的 $\text{DP}$ 。

记 $f_{i,j}$ 为长度为 $j$ 的序列只含 $[1,i]$ 内的数的价值。

转移是考虑是否加上 $i$ 这个元素并且如果加上放在哪一位，那么

$$f_{i,j}=f_{i-1,j-1}\times j\times i+f_{i-1,j}$$

这样转移是 $O(An)$ 的，过不了...

据说 $f_{i,j}$ 是一个关于 $i$ 有关的 $2j$ 次的多项式。

具体证明的话，可以差分之后用数学归纳法来证。

那么我们考虑 $\text{DP}$ 出 $f_{i,n},i\in[0,2n]$ 的值，用拉格朗日插值求出唯一解即可...

Code

#include <bits/stdc++.h>
using namespace std;
const int N = 1000+5;

int f[N][N], a, n, p, x[N], y[N], ifac[N], inv[N];

int quick_pow(int a, int b) {
    int ans = 1;
    while (b) {
        if (b&1) ans = 1ll*ans*a%p;
        b >>= 1, a = 1ll*a*a%p;
    }
    return ans;
}
int lagrange(int n, int *x, int *y, int xi) {
    int ans = 0, s1 = 1;
    for (int i = 0; i <= n; i++) {
        s1 = 1ll*s1*(xi-x[i])%p;
        inv[i] = quick_pow(xi-x[i], p-2);
    }
    ifac[1] = ifac[0] = 1;
    for (int i = 2; i <= n; i++) ifac[i] = -1ll*p/i*ifac[p%i]%p;
    for (int i = 2; i <= n; i++) ifac[i] = 1ll*ifac[i]*ifac[i-1]%p;
    for (int i = 0; i <= n; i++)
        (ans += 1ll*y[i]*s1%p*inv[i]%p*ifac[i]%p*(((n-i)&1) ? -1 : 1)*ifac[n-i]%p) %= p;
    return (ans+p)%p;
}
void work() {
    scanf("%d%d%d", &a, &n, &p);
    for (int i = 0; i <= n*2; i++) f[i][0] = 1;
    for (int i = 1; i <= n*2; i++)
        for (int j = 1; j <= n; j++)
            f[i][j] = (1ll*f[i-1][j-1]*i%p*j%p+f[i-1][j])%p;
    if (a <= 2*n) {printf("%d\n", f[a][n]); return; }
    for (int i = 0; i <= 2*n; i++) x[i] = i, y[i] = f[i][n];
    printf("%d\n", lagrange(2*n, x, y, a));
}
int main() {work(); return 0; }