HDU1121(拉格朗日插值法)
题意:
给定你一个序列,要求算出这个序列的后C个数字
思路:
拉格朗日插值法也是希望找出n个数背后的通项公式。
差分:
1 2 4 7 11 16 22.这是原始序列,称为0阶差分,
用前一个数减去后一个数得到:1 2 3 4 5 6 ,这种操作进行一次,我们称为1阶差分.
在一阶差分基础上再进行差分运算。。。。。循环下去,进行n-1次,得到n-1阶差分。
通过差分定义发现,i阶差分序列的个数等于i-1阶差分的数目减一。 等于 n-i。 n是原始序列,0阶差分序列的个数。
所以求差分的公式如下:
for(int i=0;i<n;i++)
{
scanf("%d",&s[0][i]);//0阶
}
for(int i=1;i<n;i++)
{
for(int j=0;j<n-i;j++)
{
s[i][j]=s[i-1][j+1]-s[i-1][j];
}
}
对于n-1阶差分,由于只有一个数,而题目中要求求出后面的m个数。所以需要对n-1阶差分进行补齐操作。也就是n-1阶差分,是一个由相同数字组成的序列。
for(int i=1;i<=m;i++)
{
s[n-1][i]=s[n-1][0];
}
最后进行倒推操作,求出0阶差分序列的m个数。
当满足数据全都是一个常数时,就可以往回进行一个序列的累加推导,就可以求出我们想到得出的序列后C位数字了
AC代码:
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const ll maxn=110;
int num[maxn][maxn];
int main()
{
int t,n,c;
cin>>t;
while(t--)
{
cin>>n>>c;
for(int i=0;i<n;i++)
cin>>num[0][i];
for(int i=1;i<n;i++)
{
for(int j=0;j<n-i;j++)
{
num[i][j]=num[i-1][j+1]-num[i-1][j];
}
}
for(int i=1;i<n+c;i++)
{
num[n-1][i]=num[n-1][0];
}
for(int i=n-2;i>=0;i--)
{
for(int j=n-i;j<n+c;j++)
{
num[i][j]=num[i+1][j-1]+num[i][j-1];
}
}
cout<<num[0][n];
for(int i=1;i<c;i++)
cout<<" "<<num[0][n+i];
cout<<endl;
}
return 0;
}