BZOJ 3527: [ZJOI2014]力(FFT)

时间:2022-03-19 08:23:43

题意

给出\(n\)个数\(q_i\),给出\(Fj\)的定义如下:

\[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\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_1x+...+A_nx^n
\]

\[B_0+B_1x+...+B_nx^n
\]

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

发现\(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;
}