[ZJOI2014]力 题解

题目描述
给出n个数qi，给出Fj的定义如下：

$$F_j = \sum_{i<j}\frac{q_iq_j}{(i-j)^2} -\sum_{i>j}\frac{q_iq_j}{(i-j)^2}$$
令Ei=Fi/qi，求Ei.
输入输出格式
输入格式：
第一行一个整数n。
接下来n行每行输入一个数，第i行表示qi。
输出格式：
n行，第i行输出Ei。
与标准答案误差不超过1e-2即可。
说明
对于30%的数据，n≤1000。
对于50%的数据，n≤60000。
对于100%的数据，n≤100000，0 < <script type="math/tex" id="MathJax-Element-3987"><</script>qi < <script type="math/tex" id="MathJax-Element-3988"><</script>1000000000。

对式子变形，再利用多项式卷积。FFT。

$$\begin{equation}\begin{split} E_i&= \sum_{j<i}\frac{q_j}{(i-j)^2} - \sum_{j>i}\frac{q_j}{(i-j)^2} \\ &= \sum_{j=1}^{i-1}\frac{q_j}{(i-j)^2} - \sum_{j=i+1}^{n}\frac{q_j}{(i-j)^2} \end{split}\end{equation}$$
令 $$\begin{equation}\begin{split}g_i = \begin{cases}&\frac{1}{i^2}(i\neq0)\\&0 (i=0)\end{cases} \end{split}\end{equation}$$
有
$$E_i = \sum_{j=1}^{i-1}{q_j\cdot g_{i-j}} - \sum_{j=i+1}^{n}{q_j \cdot a_{j-i}}$$
不难发现， $\sum_{j=1}^{i-1}{q_j\cdot g_{i-j}}$ 卷积一下就好了，接下来讨论右边的。
为了表示方便，令 $$E_i = \varphi_i - \omega_t$$
其中 $\varphi = Q \times A$
接下来求 $\omega$。
$$\begin{equation}\begin{split} \omega_t &=\sum_{i \lt j \le n } {q_j \cdot g_{j-i}} \\ &=\sum_{i \lt j+i \le n} {q_{j+i} \cdot g_j }\\ &=\sum_{0 \lt j \le n-i } { q_{j+i} \cdot g_j }\\ &=\sum_{0 \lt n-i-j \le n-i } { q_{n-j} \cdot g_{n-i-j} }\\ &=\sum_{0\le j\le n-i}{q_{n-j}\cdot g_{n-i-j}}\\ &=\sum_{0 \le j \le t}{q_{n-j}\cdot g_{t-j}} \quad(t=n-i) \\ &=\sum_{0 \le j \le t}{p_j \cdot g_{t-j}} \quad(p_i = q_{n-i})\\ &=P\times G \end{split}\end{equation}$$

然后把$\varphi$和$\omega$给求出来就OK了。

#include <bits/stdc++.h>
using namespace std ;
const int maxn = 400005 ;
const double PI = acos(-1.0) ;
int n, m, dep, rev[maxn], len ;
double q[maxn], g[maxn], p[maxn], ans[maxn] ;
complex <double> a[maxn], b[maxn] ;
void DFT ( complex <double> a[], int len, int t ) {
    int i, j, k ;
    for ( i = 1 ; i < len ; i ++ ) if ( i<rev[i] ) swap ( a[i], a[rev[i]] ) ;
    for ( j = 1 ; j < len ; j <<= 1 ) {
        complex <double> dw( cos(2*PI/(j<<1)), sin(2*t*PI/(j<<1)) ) ;
        for ( i = 0 ; i < len ; i += (j<<1) ) {
            complex <double> w(1,0), t0, t1 ;
            for ( k = 0 ; k < j ; k ++, w *= dw ) {
                t0 = a[i+k] ;
                t1 = w*a[i+k+j] ;
                a[i+k] = t0+t1 ;
                a[i+j+k] = t0-t1 ;
            }
        }
    }
    for ( i = 0 ; i <= len && t == -1 ; i ++ ) a[i] /= len ;
}
void FFT ( double x[], double y[], complex<double> a[], complex<double> b[], complex <double> c[], int len ) {
    int i ;
    for ( i = 0 ; i < len ; i ++ ) a[i] = x[i] ;
    for ( i = 0 ; i < len ; i ++ ) b[i] = y[i] ;
    DFT ( a, len, 1 ) ; DFT ( b, len, 1 ) ;
    for ( i = 0 ; i < len ; i ++ ) c[i] = a[i]*b[i] ;
    DFT ( c, len, -1 ) ;
}
void init() {
    int i, j, k = 0 ;
    len = n<<1|1 ;
    for ( k = 1 ; k <= len ; k <<= 1 ) ++ dep ;
    len = k ;
    for ( i = 1 ; i < len ; i ++ ) rev[i] = (rev[i>>1]>>1)|((i&1)<<dep-1) ;
}
int main() {
    int i, j, k ;
    scanf ( "%d", &n ) ;
    for ( i = 1 ; i <= n ; i ++ ) scanf ( "%lf", q+i ) ;
    for ( p[0] = q[n], i = 1 ; i <= n ; i ++ ) g[i] = 1.0/(i*i), p[i] = q[n-i] ;
    init() ;
    FFT ( q, g, a, b, a, len ) ;
    for ( i = 0 ; i <= n ; i ++ ) ans[i] = a[i].real() ;
    FFT ( p, g, a, b, a, len ) ;
    for ( i = 0 ; i <= n ; i ++ ) ans[i] -= a[n-i].real() ;
    for ( i = 1 ; i <= n ; i ++ ) printf ( "%.3lf\n", ans[i] ) ;
    return 0 ;
}