SteinerTree模板

#define N 55//所有点的个数
#define K 10//SteinerTree 最大顶点数,必须精确
#define INF 10000000

//SteinerTree 邻接矩阵模板。（稠密图）时间复杂度 O(N*2^K*(2^K+N))
int dp[(<<K)+][N];
int STV[N];

int SteinerTreeDP(int mat[N][N],int maxid,int *sameset,int size)
{
    //mat为表示距离的邻接矩阵
    //所有的标点从1到maxid
    //SteinerTree 所必须的点集为 sameset[0] 到 sameset[size-1]
    //函数放回最小Steiner Tree的值

    for(int i=;i<(<<size);i++)
        for(int j=;j<=maxid;j++)
            dp[i][j]=INF;

    for (int i=; i<size; i++) {
        dp[(<<i)][sameset[i]]=;
    }
    for (int i=;i<(<<size);i++)
    {
        //step 1
        for(int kk=;kk<=maxid;kk++)
        {
            STV[kk]=;
            for(int j = (i-)&i ; j ;j = (j-)&i)
            {
                dp[i][kk] = min(dp[i][kk],dp[j][kk]+dp[(~j)&i][kk]);
            }
        }
        //step 2
        int kk,stmin=INF,stminid=;
        for (int j = ; stmin = INF, j < maxid; j++)
        {
            for (kk = ; kk <= maxid; kk++)
                if (dp[i][kk] <= stmin && !STV[kk])
                    stmin = dp[i][stminid = kk];

            for (STV[stminid]=,kk = ; kk <= maxid; kk++)
                if(STV[kk]==) dp[i][kk] = min(dp[i][kk], dp[i][stminid] + mat[stminid][kk]);
        }
    }

    int tmin=INF;
    for(int j=;j<=maxid;j++)
        tmin=min(tmin,dp[(<<size)-][j]);
    return tmin;
}