[题解] [ZJOI2014] 力
作者:互联网
题面
题解
恩, 我们首先有这两个关系
\[
\displaystyle\begin{aligned} F_j &= \sum_{i < j}\frac{q_iq_j}{(i - j)^2} - \sum_{i > j}\frac{q_iq_j}{(i - j)^2}\\ &= q_j\cdot(\sum_{i < j}\frac{q_i}{(i - j)^2}-\sum_{i>j}\frac{q_i}{(i - j)^2})\end{aligned}
\]
然后由
\[
\displaystyle E_i = \frac{F_i}{q_i}
\]
我们可以推出这样的式子
\[
\displaystyle\begin{aligned}
E_i&=\frac{F_i}{q_i}\\&=\sum_{j<i}\frac{q_j}{(i - j)^2}-\sum_{j>i}\frac{q_j}{(j - i)^2}\\
\end{aligned}
\]
我们来看一下 n = 3 的情况
我们将红色的线的两端乘起来, 再把三条红线的和相加就得到了\(E_0\), 同理我们通过蓝线, 绿线可以得到\(E_1\), \(E_2\)
我们定义\(a_i = p_i\), \(b_i = \frac{1}{i ^ 2}\), 特别的, 我们定义\(b_{-i} = -\frac{1}{i ^ 2}\), 先别管为什么下标为负
\[
\displaystyle\begin{aligned}
E_0&=a_0b_0+a_1b_{-1}+a_2b_{-2}\\
E_1&=a_0b_1+a_1b_0+a_2b_{-1}\\
E_2&=a_0b_2+a_1b_1+a_2b_0
\end{aligned}
\]
恩, 我们来讨论数组下标小于零的问题, 整体平移就行了嘛, 注意到\(b\)最小的下标为\((-3 + 1)\), 所以我们平移\((3 - 1)\)位就可以了, 所以有:
\[
\begin{aligned}
E_0&=a_0b_2+a_1b_1+a_2b_0\Leftrightarrow E_0 = \sum_{i+j=2}a_ib_j\\
E_1&=a_0b_3+a_1b_2+a_2b_1\Leftrightarrow E_1 = \sum_{i+j=3}a_ib_j\\
E_2&=a_0b_4+a_1b_3+a_2b_2\Leftrightarrow E_2 = \sum_{i+j=4}a_ib_j
\end{aligned}
\]
恩, 我们来看一下卷积的形式
\[
c_k = \sum_{i + j = k}a_ib_j
\]
我们会发现这里的\(E\)数组的下标离可以卷积差了一个\((3 - 1)\), 用\(E_2\)代替\(E_0\)就可以用\(a\)与\(b\)的卷积来算了
我们将特殊情况扩展到一般情况, 设\(E\)共有\(n\)项, 自己瞎猜一下可以发现\(b\)数组下标平移了\(n - 1\)位, \(E\)数组的下标平移了\(n - 1\)位, 所以, 最后我们输出\(E\)的第\([n - 1, (n - 1) + n - 1]\)项即可
这篇题解就是写给我们这种蒟蒻看的, 至于卷积, 我猜各位是为了练习FFT才来做这道题的吧
Code
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#define itn int
#define reaD read
#define N 1000005
const double pi = acos(-1);
using namespace std;
int n, m, cnt, rev[N];
struct complex
{
double x, y;
complex(double xx = 0, double yy = 0) { x = xx; y = yy; }
complex operator + (complex p) { return complex(x + p.x, y + p.y); }
complex operator - (complex p) { return complex(x - p.x, y - p.y); }
complex operator * (complex p) { return complex(x * p.x - y * p.y, x * p.y + y * p.x); }
} a[N], b[N];
inline int read()
{
int x = 0, w = 1; char c = getchar();
while(c < '0' || c > '9') { if (c == '-') w = -1; c = getchar(); }
while(c >= '0' && c <= '9') { x = x * 10 + c - '0'; c = getchar(); }
return x * w;
}
void FFT(complex* p, int opt)
{
for(int i = 0; i < m; i++) if(i < rev[i]) swap(p[i], p[rev[i]]);
for(int i = 1; i < m; i <<= 1)
{
complex cur = complex(cos(pi / i), opt * sin(pi / i));
for(int j = 0; j < m; j += (i << 1))
{
complex w = complex(1, 0);
for(int k = 0; k < i; k++, w = w * cur)
{
complex l = p[j + k], r = w * p[i + j + k];
p[j + k] = l + r; p[i + j + k] = l - r;
}
}
}
}
int main()
{
n = read();
for(int i = 0; i < n; i++) scanf("%lf", &a[i].x);
for(int i = 1; i < n; i++)
{
b[i + n - 1].x = 1 / (1.0 * (double) i * i);
b[n - 1 - i].x = -b[i + n - 1].x;
}
for(m = 1; m <= 3 * n - 2; m <<= 1, cnt++);
for(int i = 0; i < m; i++) rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (cnt - 1));
FFT(a, 1); FFT(b, 1);
for(int i = 0; i <= m; i++) a[i] = a[i] * b[i];
FFT(a, -1);
for(int i = n - 1; i <= 2 * n - 2; i++) printf("%.3lf\n", a[i].x / m);
return 0;
}
这可能是我写得最认真的一篇题解吧, 希望看这篇题解的泥萌可以康懂
标签:include,frac,题解,sum,complex,ZJOI2014,2b,aligned 来源: https://www.cnblogs.com/ztlztl/p/11104862.html