AH2017/HNOI2017 礼物

P3723 [AH2017/HNOI2017]礼物

题目大意

给两个数列$a$和$b$，$a$和$b$中的数都小于等于$m$，$b$可以首尾相接地旋转，要选择一个整数$c$，使得$\sum _{i=1}^n(a_i+c-b_i{)}^2$最小。

$1\le n\le 50000,1\le a_i\le m\le 100$

题解

前置知识：快速数论变换（$NTT$）

先不考虑旋转，则差异值为

$$\sum _{i=1}^n(a_i+c-b_i{)}^2=(\sum _{i=1}^n{a}_i^2+b_i^2)-2(\sum _{i=1}^n{a}_i{b}_i)+c(\sum _{i=1}^n{a}_i-b_i)+n\cdot c^2$$

从$-m$到$m$枚举$c$，$(\sum _{i=1}^n{a}_i^2+b_i^2)$，$(\sum _{i=1}^n{a}_i-b_i)$可以提前求出。那么现在的问题就是求旋转后的$(\sum _{i=1}^n{a}_i{b}_i)$了。

我们可以将$b_i$翻转，那么要求的式子就变为

$$\sum _{i=1}^n{a}_i{b}_{n-i+1}$$

我们发现，式子变成了卷积的形式。

如果我们将$b$倍长，那么旋转$t$个位置后对应的式子就是

$$\sum _{i=1}^n{a}_i{b}_{n-i+1+t}=\sum _{i=1}^{n+t}{a}_i{b}_{n-i+1+t}$$

因为$1\le a_i,b_i\le m\le 100$，所以$\sum _{i=1}^n{a}_i{b}_{n-i+1}$的值最多为$5\times 10^8$，取一个较大的模数（比如$1004535809$）即可用$NTT$来解决。当然，用$FFT$也可以解决。

用$NTT$将$a$与倍长后的$b$求卷积，然后取第$n+1$项到第$2n$项的最大值，这就是原来的$\sum _{i=1}^n{a}_i{b}_i$。

那么，求出了$\sum _{i=1}^n{a}_i{b}_i$,，即可求出最小的$\sum _{i=1}^n(a_i+c-b_i{)}^2$。

时间复杂度为$O(n\log n)$。

code

#include<bits/stdc++.h>
using namespace std;
long long ans=1e18,pt=-1e18,f1=0,f2=0,g1=0,g2=0,w,wn,f[500005],g[500005];
const long long G=3,mod=1004535809;
long long mi(long long t,long long v){
	if(!v) return 1;
	long long re=mi(t,v/2);
	re=re*re%mod;
	if(v&1) re=re*t%mod;
	return re;
}
void ch(long long *a,int l){
	for(int i=1,j=l/2;i<l-1;i++){
		if(i<j) swap(a[i],a[j]);
		int k=l/2;
		while(j>=k){
			j-=k;k>>=1;
		}
		j+=k;
	}
}
void ntt(long long *a,int l,int fl){
	for(int i=2;i<=l;i<<=1){
		if(fl==1) wn=mi(G,(mod-1)/i);
		else wn=mi(G,mod-1-(mod-1)/i);
		for(int j=0;j<l;j+=i){
			w=1;
			for(int k=j;k<j+i/2;k++,w=w*wn%mod){
				long long t=a[k],u=w*a[k+i/2]%mod;
				a[k]=(t+u)%mod;
				a[k+i/2]=(t-u+mod)%mod;
			}
		}
	}
	if(fl==-1){
		long long ny=mi(l,mod-2);
		for(int i=0;i<l;i++) a[i]=a[i]*ny%mod;
	}
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%lld",&f[i]);
		f1+=f[i]*f[i];
		f2+=f[i];
	}
	for(int i=n;i>=1;i--){
		scanf("%lld",&g[i]);
		g1+=g[i]*g[i];
		g2+=g[i];
		g[n+i]=g[i];
	}
	int len=1;
	while(len<4*n+4) len<<=1;
	ch(f,len);ch(g,len);
	ntt(f,len,1);ntt(g,len,1);
	for(int i=0;i<len;i++) f[i]=f[i]*g[i]%mod;
	ch(f,len);
	ntt(f,len,-1);
	for(int i=-m;i<=m;i++){
		ans=min(ans,f1+g1+2ll*i*(f2-g2)+1ll*n*i*i);
	}
	for(int i=n+1;i<=2*n;i++){
		pt=max(pt,f[i]);
	}
	ans-=2*pt;
	printf("%lld",ans);
	return 0;
}