64,654
社区成员
发帖
与我相关
我的任务
分享
#include <iostream>
#include <fstream>
#include <cmath>
const double pi=3.1415926;
using namespace std;
const int p=8;
const int n=1<<p;
const int m=n;
typedef struct//c语言的写法,尽量用堆空间,不要用栈
{
double r; //real
double i; //image
}complex;//定义一个复数类,有两个元素
void ZeroArray(complex c[n][n])//对复数数组进行初始化
{
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
c[i][j].r=0;
c[i][j].i=0;
}
}
}
void Fourier(const double f[n][n],complex g[n][n])//不能把第二维或者更高维的大小省略
{
/*const double c=1/sqrt((double)n);//没有sqrt(int),只有sqrt(float),
//sqrt(double),返回值固定是double
const double d=1/sqrt((double)m);*/
const double c=1/n;
for(int k=0;k<=n;k++)
{
const double qk=2*pi*k/n;
for(int l=0;l<=n;l++)
{
const double ql=2*pi*l/m;
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
g[k][l].r+=cos(qk*i+ql*j)*f[k][l];
g[k][l].i+=-sin(qk*i+ql*j)*f[k][l];
}
}
g[k][l].r*=c;
g[k][l].i*=c;
}
}
}
int main()
{
//将1对应的二进制数左移p位 N=2^p
cout<<"n="<<n<<endl;
//system("pause");
static double f1[n][n];//,f2[n];
for(int i=0;i<=n;i++)
{
const double x=2*pi*i/n;
for(int j=0;i<=n;j++)
{
const double y=2*pi*j/n;
f1[i][j]=sin(5*x+7*y);
}
}//先求f,再求g,。。。。。。。。。。。。。。。
static complex g1[n][n];
ZeroArray(g1);
Fourier(f1,g1);
ofstream out1("data1.dat");
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
out1<<"\t"<<f1[i][j];
}
out1<<endl;
}
out1.close();
/*ofstream out2("data2.dat");
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
out2<<"\t"<<g1[i][j].r;
}
out2<<endl;
}
out2.close();
ofstream out3("data3.dat");
for(int i=0;i<=n;i++)
{
for(int j=0;j<=n;j++)
{
out3<<"\t"<<g1[i][j].i;
}
out3<<endl;
}
out3.close();*/
system("pause");
return 0;
}