BZOJ 3527: [ZJOI2014]力（FFT）

题意

给出 $n$ 个数 $q_i$ ,给出 $Fj$ 的定义如下：

F_{j} = \sum_{i < j} \frac{q_{i} q_{j}}{(i - j)^{2}} - \sum_{i > j} \frac{q_{i} q_{j}}{(i - j)^{2}} .

$F_j=\sum \limits _ {i < j} \frac{q_iq_j}{(i-j)^2}-\sum \limits _{i >j} \frac{q_iq_j}{(i-j)^2}.$

令 $E_i=F_i/q_i$ ，求 $E_i$ .

题解

一开始没发现求 $E_i$ ... 其实题目还更容易想了...

E_{i} = \sum_{j < i} \frac{q_{j}}{(i - j)^{2}} - \sum_{j > i} \frac{q_{j}}{(i - j)^{2}}

$E_i=\sum\limits _{j<i}\frac{q_j}{(i-j)^2}-\sum\limits _{j>i}\frac{q_j}{(i-j)^2}$

这个东西就是转换成求两个一样的东西就行了。

就是求 $sum_i=\sum \limits_{j<i} \frac{q_j}{(i-j)^2}$ .

这个就是可以转换成求一个卷积形式就行了。

注意多项式乘法格式是这样的：

A_{0} + A_{1} x + . . . + A_{n} x^{n}

$A_0+A_1x+...+A_nx^n$

B_{0} + B_{1} x + . . . + B_{n} x^{n}

$B_0+B_1x+...+B_nx^n$

令 $A$ 与 $B$ 的卷积为 $C$ ,则$$C_i=\sum \limits {j \le i}A_j*B$$.

发现 $i-j$ 那个形式似乎就可以满足本题的形式。

所以令 $B_i=\frac{1}{i^2}$ 就行了，然后 $A_i=q_i$ .

对于这个求两边卷积就行了23333

注意有的细节要处理一下，就是要清空一些数组，

注意一下下标（思维要清楚），而且也要令 $A_0=B_0=0$ 。

而且之前求 $B_i$ 的时候， $i^2$ 会爆long long ！

代码

/**************************************************************
    Problem: 3527
    User: zjp_shadow
    Language: C++
    Result: Accepted
    Time:3688 ms
    Memory:32012 kb
****************************************************************/
#include <bits/stdc++.h>
#define For(i, l, r) for(register int i = (l), _end_ = (int)(r); i <= _end_; ++i)
#define Fordown(i, r, l) for(register int i = (r), _end_ = (int)(l); i >= _end_; --i)
#define Set(a, v) memset(a, v, sizeof(a))
using namespace std;
 
bool chkmin(int &a, int b) {return b < a ? a = b, 1 : 0;}
bool chkmax(int &a, int b) {return b > a ? a = b, 1 : 0;}
 
inline int read() {
    int x = 0, fh = 1; char ch = getchar();
    for (; !isdigit(ch); ch = getchar() ) if (ch == '-') fh = -1;
    for (; isdigit(ch); ch = getchar() ) x = (x<<1) + (x<<3) + (ch ^ '0');
    return x * fh;
}
 
void File() {
#ifdef zjp_shadow
    freopen ("P3527.in", "r", stdin);
    freopen ("P3527.out", "w", stdout);
#endif
}
 
struct Complex {
    double re, im;
};
 
inline Complex operator + (const Complex &lhs, const Complex &rhs) {
    return (Complex) {lhs.re + rhs.re, lhs.im + rhs.im};
}
 
inline Complex operator - (const Complex &lhs, const Complex &rhs) {
    return (Complex) {lhs.re - rhs.re, lhs.im - rhs.im};
}
 
inline Complex operator * (const Complex &lhs, const Complex &rhs) {
    return (Complex) {lhs.re * rhs.re - lhs.im * rhs.im, lhs.re * rhs.im + rhs.re * lhs.im};
}
 
const int N = 1 << 19;
int n_, n;
double f[N], g[N];
const double Pi = acos(-1.0);
 
int r[N];
 
void FFT(Complex P[], int opt) {
    For (i, 0, n - 1) if (i < r[i]) swap(P[i], P[r[i]]);
    for (int i = 2; i <= n; i <<= 1) {
        Complex Wi = (Complex) {cos(2 * Pi / i), opt * sin(2 * Pi / i)};
        int p = i / 2;
        for (int j = 0; j < n; j += i) {
            Complex x = (Complex) {1.0, 0.0};
            For (k, 0, p - 1) {
                Complex u = P[j + k], v = x * P[j + k + p];
                P[j + k] = u + v;
                P[j + k + p] = u - v;
                x = x * Wi;
            }
        }
    }
}
 
int m;
void Mult(Complex a[], Complex b[]) {
    int cnt = 0;
    for (n = 1; n <= m; n <<= 1) ++ cnt;
    For (i, 1, n - 1)
        r[i] = (r[i >> 1] >> 1) | ((i & 1) << (cnt - 1) );
    FFT(a, 1); FFT(b, 1);
    For (i, 0, n - 1) a[i] = a[i] * b[i];
    FFT(a, -1);
    For (i, 0, n - 1) a[i].re = a[i].re / n;
}
 
double ans[N];
Complex a[N], b[N];
 
int main () {
    //int n1 = read(), n2 = read(),
    File();
    n_ = read();
    m = n_ + n_;
    For (i, 1, n_) {
        scanf("%lf", &f[i]);
        g[i] = (double)1.0 / ((long long)i * (long long)i);
    }
    For (i, 0, n_) a[i].re = f[i], a[i].im = 0;
    For (i, 0, n_) b[i].re = g[i], b[i].im = 0; 
    Mult(a, b);
    For (i, 1, n_)
        ans[i] += a[i].re;
 
    reverse(f + 1, f + 1 + n_);
    For (i, 0, n - 1) a[i].re = f[i], a[i].im = 0;
    For (i, 0, n - 1) b[i].re = g[i], b[i].im = 0;
 
    Mult(a, b);
 
    For (i, 1, n_)
        ans[n_ - i + 1] -= a[i].re;
 
    For (i, 1, n_)
        printf ("%.4lf\n", ans[i]);
    return 0;
}

__EOF__

本文作者：zjp_shadow
本文链接：https://www.cnblogs.com/zjp-shadow/p/8435930.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！