BZOJ 2194 快速傅立叶之二 快速傅里叶变换

题目大意:给定两个长度为n的序列a和b,求c[k]=Σa[i]*b[i-k]

这东西不是卷积的形式,因此我们将b数组反转,之后就是卷积的形式了

然后就愉♂悦地上FFT吧

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 263000
#define PI 3.1415926535897932384626433832795028841971
using namespace std;
struct Complex{
	double a,b;
	Complex() {}
	Complex(double _,double __):a(_),b(__) {}
	Complex operator + (const Complex &x) const
	{
		return Complex(a+x.a,b+x.b);
	}
	Complex operator - (const Complex &x) const
	{
		return Complex(a-x.a,b-x.b);
	}
	Complex operator * (const Complex &x) const
	{
		return Complex(a*x.a-b*x.b,a*x.b+b*x.a);
	}
	void operator *= (const Complex &x)
	{
		*this=(*this)*x;
	}
}a[M],b[M],p[M];
int n;
void FFT(Complex x[],int n,int type)
{
	static Complex temp[M];
	if(n==1) return ;
	int i;
	for(i=0;i<n;i+=2)
		temp[i>>1]=x[i],temp[i+n>>1]=x[i+1];
	memcpy(x,temp,sizeof(Complex)*n);
	Complex *l=x,*r=x+(n>>1);
	FFT(l,n>>1,type);FFT(r,n>>1,type);
	Complex root(cos(type*2*PI/n),sin(type*2*PI/n)),w(1,0);
	for(i=0;i<n>>1;i++,w*=root)
		temp[i]=l[i]+w*r[i],temp[i+(n>>1)]=l[i]-w*r[i];
	memcpy(x,temp,sizeof(Complex)*n);
}
int main()
{
	int i,digit;
	cin>>n;
	for(i=0;i<n;i++)
		scanf("%lf%lf",&a[i].a,&b[n-i-1].a);
	for(digit=1;digit<=n<<1;digit<<=1);
	FFT(a,digit,1);FFT(b,digit,1);
	for(i=0;i<digit;i++)
		p[i]=a[i]*b[i];
	FFT(p,digit,-1);
	for(i=n-1;i<n+n-1;i++)
		printf("%d\n",int(p[i].a/digit+0.5) );
}
时间: 2024-10-12 19:28:29

BZOJ 2194 快速傅立叶之二 快速傅里叶变换的相关文章

【BZOJ 2194】 快速傅立叶之二

2194: 快速傅立叶之二 Time Limit: 10 Sec Memory Limit: 259 MB Submit: 430 Solved: 240 [Submit][Status][Discuss] Description 请计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ,并且有 n < = 10 ^ 5. a,b中的元素均为小于等于100的非负整数. Input 第一行一个整数N,接下来N行,第i+2..i+N-1行,每行两个数,依次表示a[

BZOJ 2194: 快速傅立叶之二

2194: 快速傅立叶之二 Time Limit: 10 Sec  Memory Limit: 259 MBSubmit: 1203  Solved: 699[Submit][Status][Discuss] Description 请计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ,并且有 n < = 10 ^ 5. a,b中的元素均为小于等于100的非负整数. Input 第一行一个整数N,接下来N行,第i+2..i+N-1行,每行两个数,依次表示a

【BZOJ 2194】2194: 快速傅立叶之二(FFT)

2194: 快速傅立叶之二 Time Limit: 10 Sec  Memory Limit: 259 MBSubmit: 1273  Solved: 745 Description 请计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ,并且有 n < = 10 ^ 5. a,b中的元素均为小于等于100的非负整数. Input 第一行一个整数N,接下来N行,第i+2..i+N-1行,每行两个数,依次表示a[i],b[i] (0 < = i <

bzoj2194 快速傅立叶之二 ntt

bzoj2194 快速傅立叶之二 链接 bzoj 思路 对我这种和式不强的人,直接转二维看. 发现对\(C_k\)贡献的数对(i,j),都是右斜对角线. 既然贡献是对角线,我们可以利用对角线的性质了. 不过右斜角线不太好,我们把每一行都reverse一下,换成左斜角线. 对角线上\(i+j\)相等,可以套上多项式乘法了. 隐藏bug \(a_i,b_i\)均不大于100,而且数字有1e5个 最大值是1e9,而模数是998244353 应该是可以卡掉模数的,但是不故意卡是不可能爆模数的. AC代码

【BZOJ】2194: 快速傅立叶之二

http://www.lydsy.com/JudgeOnline/problem.php?id=2194 题意:求$c[k]=\sum_{k<=i<n} a[i]b[i-k], n<=10^5$ #include <bits/stdc++.h> using namespace std; struct cp { double x, y; cp(double _x=0, double _y=0):x(_x),y(_y) {} cp operator+(const cp &

BZOJ 2194 快速傅立叶之二 ——FFT

[题目分析] 咦,这不是卷积裸题. 敲敲敲,结果样例也没过. 看看看,卧槽i和k怎么反了. 艹艹艹,把B数组取个反. 靠靠靠,怎么全是零. 算算算,最终的取值范围算错了. 交交交,总算是A掉了. [代码] #include <cstdio> #include <cstring> #include <cmath> #include <cstdlib> #include <map> #include <set> #include <

bzoj 2194: 快速傅立叶之二【NTT】

看别的blog好像我用了比较麻烦的方法-- (以下的n都--过 \[ c[i]=\sum_{j=i}^{n}a[i]*b[j-i] \] 设j=i+j \[ c[i]=\sum_{j=0}^{n-i}a[i+j]*b[i+j-i] \] \[ c[i]=\sum_{j=0}^{n-i}a[i+j]*b[j] \] 再设j=n-i-j \[ c[i]=\sum_{n-i-j}^{n-i}a[n-i-j+i]b[n-i-j] \] \[ n-i-j \geq 0 \Rightarrow j \leq

BZOJ.2194.快速傅立叶之二(FFT 卷积)

题目链接 \(Descripiton\) 给定\(A[\ ],B[\ ]\),求\[C[k]=\sum_{i=k}^{n-1}A[i]*B[i-k]\ (0\leq k<n)\] \(Solution\) (先令\(n=n-1\)) 首先往卷积上想.. \(i\)与\(i-k\)的差值是一定的,但是卷积的形式是\[C[k]=\sum_{i=1}^k A[i]*B[k-i]\] 即\(i\)与\(k-i\)的和是一定的. 于是考虑把一个数组反转一下,这里把\(B[\ ]\)反转,那么\[C[k]=

BZOJ_2194_快速傅立叶之二_(FFT+卷积)

描述 http://www.lydsy.com/JudgeOnline/problem.php?id=2194 给出序列\(a[0],a[1],...,a[n-1]\)和\(b[0],b[1],...,b[n-1]\). \(c[k]=\sum_{i=k}^{n-1}a[i]b[i-k]\). 求序列\(c[]\). 分析 这题就是BZOJ_3527_[ZJOI2014]_力_(FFT+卷积)的后半段... 我们来重新分析一下. 首先我们要知道卷积的标准形式: $$c[i]=\sum_{j=0}