zoukankan      html  css  js  c++  java
  • BZOJ.4827.[HNOI2017]礼物(FFT)

    BZOJ
    洛谷


    首先不妨枚举旋转多少次(k),然后题目就是要求(sum_{i=1}^{n}(x_{i+k}-y_i+c)^2)的最小值,其中(kin[0,n))(cin[-m,m])(显然(c)枚举到(m)就足够了,而另一个环增加可以看成这个环减少),且(x_{i+n}=x_i)(是个环啊)。
    把式子拆开:$$sum_{i=1}^{n}(x_{i+k}-y_i+c)^2=sum_{i=1}^{n}left[x_{i+k}^2+y_i^2-2x_{i+k}y_i+c^2+2c(x_{i+k}-y_i) ight]$$

    再把(sum)拆了:$$sum_{i=1}^nx_{i+k}^2+sum_{i=1}^n y_i^2-2sum_{i=1}^n x_{i+k}y_i+nc^2+2cleft(sum_{i=1}^n x_i-sum_{i=1}^n y_i ight)$$

    我们发现前面两项(sum_{i=1}^nx_{i+k}^2+sum_{i=1}^n y_i^2)是不变的,后面两项(nc^2+2cleft(sum_{i=1}^n x_i-sum_{i=1}^n y_i ight))在序列已经确定的情况下只与(c)有关,枚举(c)求个最小值就可以了。

    然后我们需要求(sum_{i=1}^n x_{i+k}y_i)的最大值,它只与(k)有关。
    这不就和某些FFT的题一样么,反转(x)(y)数组,变为求(sum_{i=1}^n x_{i+k}y_{n-i+1}),就是卷积了。(Ans_k)就是多项式相乘后的第(n+k+1)项。
    枚举(k)取一个(max{Ans_k})就可以了。


    //12380kb	1120ms(为啥我的FFT这么慢啊QAQ)
    #include <cmath>
    #include <cstdio>
    #include <cctype>
    #include <algorithm>
    //#define gc() getchar()
    #define MAXIN 300000
    #define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
    typedef long long LL;
    const int N=(1<<18)+5;//3n!
    const double PI=acos(-1);
    
    int X[N],Y[N],rev[N];
    char IN[MAXIN],*SS=IN,*TT=IN;
    struct Complex
    {
    	double x,y;
    	Complex(double x=0,double y=0):x(x),y(y) {}
    	Complex operator +(const Complex &a) {return Complex(x+a.x, y+a.y);}
    	Complex operator -(const Complex &a) {return Complex(x-a.x, y-a.y);}
    	Complex operator *(const Complex &a) {return Complex(x*a.x-y*a.y, x*a.y+y*a.x);}
    }A[N],B[N];
    
    inline int read()
    {
    	int now=0;register char c=gc();
    	for(;!isdigit(c);c=gc());
    	for(;isdigit(c);now=now*10+c-'0',c=gc());
    	return now;
    }
    void FFT(Complex *a,int lim,int opt)
    {
    	for(int i=1; i<lim; ++i) if(i<rev[i]) std::swap(a[i],a[rev[i]]);
    	for(int i=2; i<=lim; i<<=1)
    	{
    		int mid=i>>1; Complex Wn(cos(PI/mid),opt*sin(PI/mid));
    		for(int j=0; j<lim; j+=i)
    		{
    			Complex w(1,0),t;
    			for(int k=j; k<j+mid; ++k,w=w*Wn)
    				a[k+mid]=a[k]-(t=w*a[k+mid]), a[k]=a[k]+t;
    		}
    //		for(int j=0; j<mid; ++j) W[j]=w, w=w*Wn;//预处理果然并不会快...
    //		for(int j=0; j<lim; j+=i)
    //			for(int k=0; k<mid; ++k)
    //				a[j+k+mid]=a[j+k]-(t=W[k]*a[j+k+mid]), a[j+k]=a[j+k]+t;
    	}
    	if(opt==-1) for(int i=0; i<lim; ++i) a[i].x/=lim;
    }
    int Calc(int n)
    {
    	for(int i=1; i<=n; ++i) A[i+n]=A[i]=Complex(X[i],0);
    	for(int i=1; i<=n; ++i) B[i]=Complex(Y[n-i+1],0);
    	int lim=1,l=-1;
    	while(lim<=3*n) lim<<=1,++l;
    	for(int i=1; i<lim; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<l);
    	FFT(A,lim,1), FFT(B,lim,1);
    	for(int i=0; i<lim; ++i) A[i]=A[i]*B[i];
    	FFT(A,lim,-1);
    	int ans=0;
    	for(int k=0; k<n; ++k) ans=std::max(ans,(int)(A[n+k+1].x+0.5));
    	return ans;
    }
    
    int main()
    {
    	int n=read(),m=read();
    	for(int i=1; i<=n; ++i) X[i]=read();
    	for(int i=1; i<=n; ++i) Y[i]=read();
    	int ans=0,s=0;
    	for(int i=1; i<=n; ++i) ans+=X[i]*X[i]+Y[i]*Y[i], s+=X[i]-Y[i];
    	int mn=2e9;
    	for(int c=-m; c<=m; ++c) mn=std::min(mn,n*c*c+2*c*s);
    	ans+=mn-2*Calc(n);
    	printf("%d
    ",ans);
    
    	return 0;
    }
    
  • 相关阅读:
    robot framework 文本校验,文本与关键字重复的处理
    python pip切本地源
    sqlalchemy 多个数据库地址配置
    python SQLAlchemy的简单配置和查询
    根据列表中字典中的某个value排序
    python SQLAlchemy中子查询subquery的使用
    python SQLAlchemy中query与query()
    weekly paper read
    C++调用windowsAPI
    English 邮件
  • 原文地址:https://www.cnblogs.com/SovietPower/p/10037221.html
Copyright © 2011-2022 走看看