#include <ctime>
#include <cstdlib>
#include <memory>
#define frand() ((double)rand()/(double)RAND_MAX)
#define MAXLONG (2147483647)
double * pso_optimization(int s, int p, int d, int T, double c1, double c2, double *vmax, double wmax, double wmin, double *domax, double *domin, CAL_FIT cal_fit, void *adds)
{
srand(time(NULL));
double *swm_v = (double*)malloc(s*p*d*sizeof(double));
double *swm_d = (double*)malloc(s*p*d*sizeof(double));
double *sbest = (double*)malloc(s*d*sizeof(double));
double *gbest = (double*)malloc(d*sizeof(double));
double *fswm_v = swm_v, *fswm_d = swm_d;
memset(swm_v, 0, sizeof(swm_v));
double *fdomax = domax, *fdomin = domin, *fvmax = vmax;
for (int i = 0; i < s; i++)
for (int j = 0; j < p; j++)
{
fdomax = domax; fdomin = domin;
for (int k = 0; k < d; k++, fdomax++, fdomin++, fswm_d++)
*fswm_d = (frand()*(*fdomax-*fdomin))+*fdomin;
}
double *fsbest = sbest, fitness = 0;
double nowbest = 0, *nowidx = NULL;
double nowsbest = 0, *nowsidx = NULL;
double *ffsbest = NULL, *fgbest = NULL;
double w = 0;
for (int t = 0; t < T; t++)
{
w = wmax-(double)t*(wmax-wmin)/(double)T;
fswm_d = swm_d; fsbest = sbest;
nowidx = NULL; nowbest = -MAXLONG;
for (int i = 0; i < s; i++, fsbest+=d)
{
nowsbest = -MAXLONG; nowsidx = NULL;
for (int j = 0; j < p; j++, fswm_d+=d)
{
fitness = cal_fit(fswm_d, d, adds);
if (fitness >= nowsbest)
{
nowsbest = fitness;
nowsidx = fswm_d;
}
}
memcpy(fsbest, nowsidx, d*sizeof(double));
if (nowsbest >= nowbest)
{
nowbest = nowsbest;
nowidx = fsbest;
}
}
memcpy(gbest, nowidx, d*sizeof(double));
if (t >= T) break;
fsbest = sbest; fswm_v = swm_v; fswm_d = swm_d;
for (int i = 0; i < s; i++)
{
for (int j = 0; j < p; j++)
{
fgbest = gbest; ffsbest = fsbest;
fdomax = domax; fdomin = domin; fvmax = vmax;
for (int k = 0; k < d; k++, fgbest++, ffsbest++, fswm_v++, fswm_d++, fdomax++, fdomin++, fvmax++)
{
*fswm_v = max(-*fvmax, min(*fswm_v*w+c1*frand()*(*ffsbest-*fswm_d)+c2*frand()*(*fgbest-*fswm_d), *fvmax));
*fswm_d = max(*fdomin, min(*fswm_d+*fswm_v, *fdomax));
}
}
fsbest = ffsbest;
}
}
free(swm_v); free(swm_d); free(sbest);
return gbest;
}
C++中蚁群优化算法的实现
最新推荐文章于 2022-03-23 21:38:40 发布