[学习笔记]拉格朗日插值

拉格朗日插值法(图文详解)

自我感觉挺实用的一个算法。

也为一些题目提供了解决的思路。

插值:给一些散点,求满足这些个散点的函数(多项式),即求出这些系数

一般求一个点值,都要先得到系数,再O(n)算。求系数,高斯消元,是O(n^3)的。

但是,如果只要一个点值,这样岂不是血亏。

拉格朗日这个人比较厉害,他发明的算法,可以在不用求出具体系数的情况下,O(n^2)的计算一个位置的点值。

思想类似于互质的CRT,

对于给定n+1个点值,这个多项式最多n次的。而且,把每个横坐标带进去,xi自己的一项得到yi,别的由于分子有xi-xi,都是0

所以,这个多项式一定和实际上的多项式是一个多项式。

然后我们把要求的x带进去,就得到了函数值。

 

有什么用?

如果证明一个式子的函数是n次多项式的话,那么可以尝试得到n+1个点值,然后弄出这个公式,就可以计算比较大的答案。

https://blog.csdn.net/xyz32768/article/details/81233900

这个题,很大数据范围的k次方和,第一没有办法反演。第二没有规律可以找。

猜这个求和函数是一个k+1次多项式。然后带点求值,然后对目标答案的计算进行化简。

从O(n)到O(k^2)到O(klogk)(然鹅这个logk是因为点值的快速幂,后面的计算不是瓶颈23333)

突破口

1.想到是一个多项式

2.点值的取值是有讲究的,1~k+2这样连续的整点有助于预处理减少复杂度(跟自己干嘛要过不去23333)

所以,拉格朗日插值这个公式其实很整齐,

如果点值横坐标给的很好的话(支持递推),那么可以在O(n)时间求出一个值。已经非常不错了。

 

CF622F The Sum of the k-th Powers

代码:

#include<bits/stdc++.h>
#define reg register int
#define il inline
#define numb (ch^'0')
using namespace std;
typedef long long ll;
il void rd(int &x){
    char ch;x=0;bool fl=false;
    while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);
    for(x=numb;isdigit(ch=getchar());x=x*10+numb);
    (fl==true)&&(x=-x);
}
namespace Miracle{
const int N=1e6+5;
const int mod=1e9+7;
int n,k;
int qm(int x,int y){
    int ret=1;
    while(y){
        if(y&1) ret=(ll)ret*x%mod;
        x=(ll)x*x%mod;
        y>>=1;
    }
    return ret;
}
ll pre[N],bac[N];
ll fu[N];
ll y[N];
ll sol(){
    ll ret=0;
    for(reg i=1;i<=k+2;++i){
        //cout<<" i "<<i<<" : "<<y[i]<<" "<<pre[i-1]<<" "<<pre[k+2-i]<<" "<<bac[k+2]<<" "<<n-i<<endl;
        if(n>k+2) ret=(ret+y[i]*qm(pre[i-1]*(fu[k+2-i])%mod,mod-2)%mod*(bac[k+2]*qm((n-i),mod-2)%mod)%mod)%mod;
        else {
         if(n!=i) ret=(ret+y[i]*qm(pre[i-1]*(fu[k+2-i])%mod,mod-2)%mod*(bac[k+2]*qm(((n-i)+mod)%mod,mod-2)%mod)%mod)%mod;
         else ret=(ret+y[i])%mod;
        } 
    //    cout<<" ret "<<ret<<endl;
    }
    return ret;
}
int main(){
    rd(n);rd(k);
    pre[0]=1;
    bac[0]=1;
    fu[0]=1;
    y[0]=0;
    for(reg i=1;i<=k+2;++i){
        pre[i]=pre[i-1]*i%mod;
        fu[i]=fu[i-1]*(mod-i)%mod;
        bac[i]=(bac[i-1]*(n-i)%mod+mod)%mod;
        y[i]=(y[i-1]+qm(i,k))%mod;
    }
    printf("%lld",sol());
    return 0;
}

}
signed main(){
    Miracle::main();
    return 0;
}

/*
   Author: *Miracle*
   Date: 2019/1/16 15:31:40
*/
View Code

 

 


upda:2019.2.18

如果要找到真正的系数:

可以快速插值O(nlogn),非常难写

一个比较实用的是O(N^2)的背包:

考虑拉格朗日插值的公式的每一个f(i)项对每个系数的贡献

$f(i) \frac{\Pi_{j!=i}(x-x_j)}{\Pi_{j!=i}(x_i-x_j)}$

分母是定值,$f(i)$是定值

分子不同之间差不多,

计算:$\Pi(x-x_j)$再每次O(N)除以$(x-x_i)$

计算方法:

对每一项的贡献就是选择k个x,剩下n-k个选择$-x_j$这样

所以背包:$f[i][j]$表示前i个“括号”,x次幂是j的权值之和

$f[i][j]=f[i-1][j]*(-x_i)+f[i-1][j-1]$

O(N^2)

 

还有一种更容易实现的方法:

$f*(x-x_i)=f*x-f*x_i$也就是f平移一位,然后减掉自己原来的xi倍

递推即可实现。

还原时候除法,就是倒着回退一次即可。

 

posted @ 2018-12-21 20:24  *Miracle*  阅读(1504)  评论(0编辑  收藏  举报