C++中蚁群优化算法的实现

#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;
}