[卷积系列] P3723 [AH2017/HNOI2017]礼物

最新推荐文章于 2024-10-18 23:42:37 发布

wly_sh

最新推荐文章于 2024-10-18 23:42:37 发布

阅读量213

点赞数

分类专栏：卷积 FFT、快速傅里叶变换文章标签：算法 fft

本文链接：https://blog.csdn.net/weixin_45012616/article/details/106683672

版权

卷积同时被 2 个专栏收录

2 篇文章 1 订阅

订阅专栏

FFT、快速傅里叶变换

2 篇文章 0 订阅

订阅专栏

P3723 [AH2017/HNOI2017]礼物

性质：给一个手环加相当于给另一个手环减
由于 $m < = 100$ 所以令增加量为 $c$ (可正可负)， $c \in [- m ， m]$

设一数列为 $a$ ，另一列旋转以后的数列为 $b$ , 那么费用为：
$\sum_{i=1}^n\left(a_i+c-b_i\right)^2$

我们把第i项拿出来拆开，得到：
$\left(a_i+c-b_i\right)^2=a_i^2+b_i^2+c^2+2a_ic-2a_ib_i-2b_ic$
$\text{}$
$\text{}$
那么原式变成了
$\sum_{i=1}^na_i^2+\sum_{i=1}^nb_i^2+nc^2+2c\left(\sum_{i=1}^na_i-\sum_{i=1}^nb_i\right)-2\sum_{i=1}^na_ib_i$

性质：无论怎么旋转，和不变
$c$ 可以暴枚，也可以算二次函数最值
$\text{}$
所以只有最后一项会改变的

问题转化为求 $\sum_{i=1}^na_ib_i$ 的最大值

把数列 $a$ 反过来，变成 $\sum_{i=1}^na_{n-i+1}b_i$
$\text{}$
$a_{n-i+1}相当于多项式A \ 的 n-i+1次项系数$ ， $b_i相当于多项式B 的\ i \ 次项系数$
$\sum_{i=1}^na_{n-i+1}b_i 就是多项式卷积之后 n+1次项的系数$

$\text{}$

将数列 $b$ (或者反过来后的 $a$ ) 倍长，用 $b_{i+k}$ 表示平移了 $k$ 项

在这里插入图片描述
那就是对于 $k \in [0 ， n - 1]$ 求 $\sum_{i=1}^na_{n-i+1}b_{i+k})$
就是求对于 $k \in [0 ， n - 1]$ 求 $m i n (卷积后第 n + k + 1 次项的系数)$
$\text{}$
$\text{}$

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const double pi=acos(-1.0);
const int N=401000;
//int a[N],b[N],c[N<<1],rev[N<<3],limit;
int a[N],b[N],c[N],rev[N],limit;
struct comp{
	double x,y;
	comp(double xx=0,double yy=0) { x=xx;y=yy; }
	friend comp operator + (const comp &a,const comp &b) { return comp(a.x+b.x , a.y+b.y); }
	friend comp operator - (const comp &a,const comp &b) { return comp(a.x-b.x , a.y-b.y); }
	friend comp operator * (const comp &a,const comp &b) { return comp(a.x*b.x-a.y*b.y , a.y*b.x + a.x*b.y); }
}A[N],B[N];//A[N<<3],B[N<<3];//limit最大6*N 

int minn(int x,int y) { return x<y?x:y; }
void fft(comp *a,double f)
{
	for(int i=0;i<limit;i++) if(i<rev[i]) swap(a[i],a[rev[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		comp wn=comp(cos(pi/mid),f*sin(pi/mid));
		for(int L=0,R=(mid<<1);L<limit;L+=R){
			comp w=comp(1.0,0.0);
			for(int k=0;k<mid;k++,w=w*wn)
			{
				comp A1=a[L+k],A2=w*a[L+k+mid];
				a[L+k]=A1+A2;
				a[L+k+mid]=A1-A2;
			}
		}
	}
}
int main()
{
	int n,m,a1=0,a2=0,b1=0,b2=0;scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]),a1+=a[i]*a[i],a2+=a[i];
	for(int i=1;i<=n;i++) scanf("%d",&b[i]),b1+=b[i]*b[i],b2+=b[i];
	for(int i=1;i<=n;i++) c[i]=a[n-i+1],b[i+n]=b[i];
	
	int nn=3*n,l=0;//c:2n b:n -- limit:3n
	limit=1;
	while(limit<nn) limit<<=1,l++;
	for(int i=1;i<limit;i++) rev[i]=(rev[i>>1]>>1)|(i&1)<<(l-1);
	for(int i=0;i<limit;i++) A[i]=comp(1.0*c[i],0.0);
	for(int i=0;i<limit;i++) B[i]=comp(1.0*b[i],0.0);
	 
	fft(A,1);
	fft(B,1);
	for(int i=0;i<limit;i++) A[i]=A[i]*B[i];
	fft(A,-1);

	int ans=2147483647;
	for(int i=1;i<=n;i++)
		for(int c=-m;c<=m;c++)
			ans=minn(ans,a1+b1+n*c*c + 2*c*(a2-b2) -2*(int)(A[i+n].x/limit+0.5));
	printf("%d",ans);
	return 0;
}