[ZJOI2014]力

题目描述

给出n个数(q[i])，给出(F[i])的定义如下：

(F[i]=sum_{j=1}^{i-1}frac{q[i]*q[j]}{(i-j)^2}-sum_{j=i+1}^{j=n}frac{q[i]*q[j]}{(i-j)^2})

令(E[i]=frac{F[i]}{q[i]})，求(E[i]).

输入输出格式

输入格式：

第一行一个整数n。

接下来n行每行输入一个数，第i行表示qi。

输出格式：

n行，第i行输出Ei。

与标准答案误差不超过1e-2即可。

输入输出样例

输入样例#1：

5
4006373.885184
15375036.435759
1717456.469144
8514941.004912
1410681.345880

输出样例#1：

-16838672.693
3439.793
7509018.566
4595686.886
10903040.872

说明

对于30%的数据，n≤1000。

对于50%的数据，n≤60000。

对于100%的数据，n≤100000，0<qi<1000000000。

---

第一道FFT的题

看完题后感觉一脸懵逼

(？？？)

之后瞎化了一下然而并没有卵用

开心的看了题解

因为我们要求(E[i])

而(E[i] = frac{F[i]}{q[i]})

所以(E[i] = sum_{j=1}^{i-1}frac{q[j]}{(i-j)^2} - sum_{j=i+1}^{n}frac{q[j]}{(i-j)^2})

然后我们设(g[i]=frac{1}{i^2})

这样(E[i]=sum_{j=1}^{i-1}{q[j]*g[i-j]} - sum_{j=i+1}^{n}{q[j]*g[j-i]})

我们可以发现第一项满足卷积

所以第一项就直接把(q)和(g)卷起来取前n项就好辣

现在考虑第二项怎么化

设(j=i+k)

所以右边的式子就可以化成(sum_{k=1}^{n-i}{q[i+k]*g[k]})

然后我们可以把q数组翻转一下,设为(revq[i])

这样式子就成了(sum_{k=1}^{n-i}{revq[n-i-k]*g[k]})

满足卷积的基本形式,我们把(revq)和(g)卷起来取前n项就好辣

但是这次卷积的结果储存在n-i+1上,所以我们要从n开始反着与第一项作差

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
const int M = 400005 ;
using namespace std ;
const double Pi = acos(-1.0) ;

struct Complex {
    double x , y ;
    Complex (double Tx = 0 , double Ty = 0) { x = Tx , y = Ty ; }
}a[M] , b[M] ;
Complex operator + (Complex a , Complex b) {
    return Complex (a.x + b.x , a.y + b.y) ;
}
Complex operator - (Complex a , Complex b) {
    return Complex (a.x - b.x , a.y - b.y) ;
}
Complex operator * (Complex a , Complex b) {
    return Complex (a.x * b.x - a.y * b.y , a.x * b.y + a.y * b.x) ;
}
int n ;
int l , r[M] , digital = 1 ;
double q[M] , fir[M] , sec[M] ;
inline void FFT(Complex *A , int unit) {
    for(int i = 0 ; i < digital ; i ++)
        if(r[i] > i)
            swap(A[i] , A[r[i]]) ;
    for(int mid = 1 ; mid < digital ; (mid <<= 1)) {
        Complex W (cos(Pi / mid) , unit * sin(Pi / mid)) ;
        int R = (mid << 1) ;
        for(int j = 0 ; j < digital ; j += R) {
            Complex w (1 , 0) ;
            for(int k = 0 ; k < mid ; k ++ , w = w * W) {
                Complex x = A[j + k] , y = w * A[j + k + mid] ;
                A[j + k] = x + y ; A[j + k + mid] = x - y ;
            }
        }
    }
}
inline void Solve() {
    while(digital <= n + n) 
      digital <<= 1 , ++l ;
    for(int i = 0 ; i < digital ; i ++)
      r[i] = (r[i>>1]>>1) | ((i&1)<<(l - 1)) ;
    FFT(a , 1) ; FFT(b , 1) ;
    for(int i = 0 ; i <= digital ; i ++) a[i] = a[i] * b[i] ;
    FFT(a , -1) ;
}
int main() {
    scanf("%d",&n) ;
    for(int i = 1 ; i <= n ; i ++) scanf("%lf",&q[i]) ;
    for(int i = 1 ; i <= n ; i ++) {
        a[i].x = q[i] ;
        b[i].x = 1.0 / (1.0 * i * i) ;
    }
    Solve() ;
    for(int i = 1 ; i <= n ; i ++) fir[i] = a[i].x / ( 1.0 * digital );
    digital = 1 ; l = 0 ;
    memset(a , 0 , sizeof(a)) ; memset(b , 0 , sizeof(b)) ;
    for(int i = 1 ; i <= (n>>1) ; i ++) swap(q[i] , q[n - i + 1]) ;
    for(int i = 1 ; i <= n ; i ++) {
        a[i].x = q[i] ;
        b[i].x = 1.0/(1.0 * i * i) ;
    }
    Solve() ;
    for(int i = 1 ; i <= n ; i ++) sec[i] = a[n - i + 1].x / ( 1.0 * digital );
    for(int i = 1 ; i <= n ; i ++) printf("%.8lf
",fir[i] - sec[i]) ;
    return 0 ;
}