Captain obvious and the Rabbit-Man 谜之数学题

题面

这是一个中文版的简化题面，原文请–> 戳这里 <–。

思路

注：下文中的$Fib_i$表示原文中的F(i)，即从$Fib_1=1$、$Fib_2=2$开始的斐波那契数列。

发现$\{a_k\}$是未知的，高斯消元$O(n^3)$求出所有a可以得到55分。

考虑满分做法，时间复杂度大概为$O(n^2)$。

根据二项式定理的经验，感性理解一下可以得出，$p(k+1)$可以表示为$p(1)..p(k)$的线性组合。

其中$p(t)=\sum_{i=1}^k a_i \cdot Fib_i^t$。

不妨设$p(k+1)=t_1 \cdot p(1) + t_2 \cdot p(2) + .. + t_k \cdot p(k)$。

这样我们就有:

$\sum_{i=1}^k a_i \cdot Fib_i^{k+1}=\sum_{i=1}^k t_i \cdot \sum_{j=1}^k a_j \cdot Fib_j^i$

$=\sum_{i=1}^k \sum_{j=1}^k t_i \cdot a_j \cdot Fib_j^i=\sum_{j=1}^k \sum_{i=1}^k t_i \cdot a_j \cdot Fib_j^i=\sum_{j=1}^k a_j \sum_{i=1}^k t_i \cdot Fib_j^i$

交换一下右边的循环变量i和j。

$=\sum_{i=1}^k a_i \sum_{j=1}^k t_j \cdot Fib_i^j$

总的写一下:

$\sum_{i=1}^k a_i \cdot Fib_i^{k+1}=\sum_{i=1}^k a_i \sum_{j=1}^k t_j \cdot Fib_i^j$

$\forall i: a_i \cdot Fib_i^{k+1}=a_i \sum_{j=1}^k t_j \cdot Fib_i^j$为上式的充分条件。

$\forall i: a_i \cdot Fib_i^{k+1}=a_i \sum_{j=1}^k t_j \cdot Fib_i^j \Leftrightarrow \forall i: Fib_i^{k+1} = \sum_{j=1}^k t_j\cdot Fib_i^j$

这样$\{t_k\}$就变得和$\{a_k\}$无关了。

如果我们能构造一个$\{t_k\}$使得$\forall i: Fib_i^{k+1} = \sum_{j=1}^k t_j\cdot Fib_i^j$，那么它显然也能使得$p(k+1)=t_1 \cdot p(1) + t_2 \cdot p(2) + .. + t_k \cdot p(k)$。

$\forall i: Fib_i^{k+1} = \sum_{j=1}^k t_j\cdot Fib_i^j \Leftrightarrow \forall i: -Fib_i^{k+1}+ \sum_{j=1}^k t_j\cdot Fib_i^j=0$

用变量$x$替换上式中的$Fib_i$，上式转化为：

$\forall i \in \{1,2,..,k\}:f(Fib_i)=0$,

其中:$f(x)=t_1\cdot x^1+t_2 \cdot x^2+.. t_k \cdot x^k +(-x^{k+1})$

也就是说，前k个斐波那契数为这个k+1次函数的k个零点，且最高次项系数为-1。这样的函数显然这个函数是很容易构造的：$f(i)=-x \cdot (x - Fib_1) \cdot (x-Fib_2) \cdot .. \cdot (x - Fib_k)$

这个多项式的i次项系数即为$t_i$，用系数表示法描述多项式，暴力卷积即可通过本题所有数据。

但是我认为这道题可以通过分治FFT进行优化，但对模数M必须进行进一步的约束（或者是使用任意模数的NTT解决，但是可惜我不会啊！！）。

代码

这道题的一种$O(n^2)$的做法，其实我觉得并不完美。如果能结合分治套任意模数NTT的话可以优化成$O(n\log ^2n)$。

#include <cstdio>
#include <cstring>
typedef long long LLint;
const int maxn=4000+5;

struct vec{
    int len; LLint a[maxn];
    void set(int a0,int a1){
        a[0]=a0;a[1]=a1;len=2;
    }
    void outp(){
        printf("{ ");
        for(int i=0;i<len-1;i++){
            printf("%lld, ",a[i]);
        }
        printf("%lld } ",a[len-1]);
    }
};

void mul(const vec& a,const vec& b,vec& c,LLint M){
    c.len=a.len+b.len-1;
    memset(c.a,0x00,sizeof(LLint)*(c.len+1));
    for(int i=0;i<a.len;i++){
        for(int j=0;j<b.len;j++){
            c.a[i+j] = (c.a[i+j] + a.a[i]*b.a[j]%M)%M;
        }
    }
}

void copy(const vec& a,vec& b){
    b.len=a.len;
    memcpy(b.a,a.a,sizeof(LLint)*(a.len+1));
}

int M;
void operator *=(vec& a,const vec& b){
    vec c; mul(a,b,c,M); copy(c,a);
}

LLint fib[maxn]; vec base,step;

int main(){
    freopen("num.in","r",stdin);
    freopen("num.out","w",stdout);
    int k; scanf("%d%lld",&k,&M);
    fib[0]=fib[1]=1;
    for(int i=2; i<=k+1; i++){
        fib[i] = (fib[i-1] + fib[i-2]) %M;
    }
    base.len=base.a[0]=1;
    for(int i=1; i<=k; i++){
        step.set(-fib[i],1);
        base *= step;
    }
    LLint sum=0;
    for(int i=1;i<=k;i++){
        int p; scanf("%d",&p);
        sum = ((sum - base.a[i-1]*p % M)%M + M) %M;
    }
    printf("%lld\n",sum);
    return 0;
}