bzoj4827 [Hnoi2017]礼物（FFT）

最新推荐文章于 2019-02-26 18:50:00 发布

Coco_T_

最新推荐文章于 2019-02-26 18:50:00 发布

阅读量323

点赞数

分类专栏： FFT 文章标签： FFT

本文链接：https://blog.csdn.net/wu_tongtong/article/details/79565671

版权

FFT 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

题目链接

分析：
求 $\sum_{i=1}^{n}(x_i-y_i)^2$

假设我们旋转了 $j$ 位，每一位上加 $c$ ，则有式子

$\sum_{i=1}^{n}(x_{i+j}+c-y_i)^2$

$=\sum_{i=1}^{n}x_{i+j}^2+y_i^2+c^2+2x_{i+j}c-2y_ic-2x_{i+j}y_i$

$=\sum x_i^2+\sum y_i^2+nc^2+2c \sum (x_i-y_i)-2\sum x_{i+j}y_i$

上式中只有 $x_{i+j}y_i$ 需要计算

受这道题的启发
我们可以把 $y$ 串翻转，变成一个卷积的形式
又因为有一个循环的形式，我们把x串加倍即可

最后枚举c分别计算上式即可

Q.

为什么卷积可以完成这个任务呢

A.

我们看一个例子：
$x:a,b,c$
$y:d,e,f$
如分析，首先我们把x加倍，变成环的形式： $x:a,b,c,a,b,c$
把y翻转，方便卷积： $y:f,e,d$

x：              a   b   c   a   b   c
y：                          f   e   d
//------------------------------------
                da  db  dc  da  db  dc
            ea  eb  ec  ea  eb  ec
        fa  fb  fc  fa  fb  fc
                 *   *   *   *

看一下标出来的几位，就是题目所求啦

GXZlegend大大表示， $c$ 不用枚举!
$f=\sum_{i=1}^{n}(x_i-y_i)^2$
$f'=\sum_{i=1}^{n}(x_i+c-y_i)^2$
$f'-f=\sum_{i=1}^{n}(2x_ic-2y_ic+c^2)=c*\sum 2x_i-2y_i+c$

$c={\sum y_i-\sum x_i \over n}$

也就是说无论如何旋转，c的最优值总是固定的 ${\sum y_i-\sum x_i \over n}$ （四舍五入到整数）

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#define ll long long

using namespace std;

const int N=200010;
const int INF=1e9;
const double Pi=acos(-1.0);
struct node{
    double x,y;
    node (double xx=0,double yy=0) {
        x=xx;y=yy;
    }
};
node a[N],b[N],o[N],_o[N];

node operator +(const node &a,const node &b) {return node(a.x+b.x,a.y+b.y);}
node operator -(const node &a,const node &b) {return node(a.x-b.x,a.y-b.y);}
node operator *(const node &a,const node &b) {return node(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}

int Sx2=0,Sy2=0,S=0,x[N],y[N],c,f[N];
int n,m,fn;

void init(int n) {
    for (int i=0;i<=n;i++) {
        o[i]=node(cos(2.0*Pi*i/n),sin(2.0*Pi*i/n));
        _o[i]=node(cos(2.0*Pi*i/n),-sin(2.0*Pi*i/n));
    }
}

void FFT(int n,node *a,node *w) {
    int i,j=0,k;
    for (i=0;i<n;i++) {
        if (i>j) swap(a[i],a[j]);
        for (int l=n>>1;(j^=l)<l;l>>=1);
    }
    for (i=2;i<=n;i<<=1) {
        int m=i>>1;
        for (j=0;j<n;j+=i)
            for (k=0;k<m;k++) {
                node z=a[j+m+k]*w[n/i*k];
                a[j+k+m]=a[j+k]-z;
                a[j+k]=a[j+k]+z;
            }
    }
}

int main()
{
    c=0;
    scanf("%d%d",&n,&m);
    for (int i=0;i<n;i++) {
        scanf("%d",&x[i]);
        Sx2+=x[i]*x[i];          //sum x^2
        S+=2*x[i];               //sum 2(x-y)
    }
    for (int i=0;i<n;i++) {
        scanf("%d",&y[i]);
        Sy2+=y[i]*y[i];          //sum y^2
        S-=2*y[i];
    }

    fn=1;
    while (fn<=n+n) fn<<=1;
    init(fn);
    for (int i=0;i<n*2;i++) a[i]=node(x[i%n],0);     //加倍环变序列 
    for (int i=0;i<n;i++) b[i]=node(y[n-i-1],0);     //翻转 
    FFT(fn,a,o); FFT(fn,b,o);
    for (int i=0;i<=fn;i++) a[i]=a[i]*b[i];
    FFT(fn,a,_o);
    for (int i=0;i<=fn;i++) f[i]=(int)(a[i].x/(double)fn+0.5);

    int ans=INF;
    for (c=-m;c<=m;c++)
        for (int i=n;i<2*n;i++) 
            ans=min(ans,Sx2+Sy2+c*S+n*c*c-2*f[i]);
    printf("%d",ans);
    return 0;
}