HDU 4085 Peach Blossom Spring

斯坦纳树。

最后可以是森林，在计算出每个联通状态的最小费用后，还需要进行一次$dp$。

#include<bits/stdc++.h>
using namespace std;

const int INF=0x7FFFFFFF;
int T,n,m,k;
vector<int>g[];
int val[][],id[],d[][],dp[];
int f[*];
queue<int>Q;

void spfa()
{
    while(!Q.empty())
    {
        int h = Q.front(); Q.pop(); f[h]=;
        int x=h/,y=h%;
        for(int i=;i<g[x].size();i++)
        {
            int to = g[x][i];

            if(to<*k)
            {
                if(((<<to)&y)==)
                {
                    if(d[x][y]+val[x][to]<d[to][y|(<<to)])
                        d[to][y|(<<to)]=d[x][y]+val[x][to];
                }
            }

            else
            {
                if(d[x][y]+val[x][to]<d[to][y])
                {
                    d[to][y] = d[x][y]+val[x][to];
                    if(f[to*+y]==)
                    {
                        f[to*+y]=;
                        Q.push(to*+y);
                    }
                }
            }
        }
    }
}

bool check(int x)
{
    int sum1=,sum2=;
    for(int i=;i<=k-;i++) if(x&(<<i)) sum1++;
    for(int i=k;i<=*k-;i++) if(x&(<<i)) sum2++;
    if(sum1==sum2) return ;
    return ;
}

int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&k);

        for(int i=;i<n;i++)
        {
            id[i]=i;
            g[i].clear();
        }

        id[k]=n-k; for(int i=k+;i<=*k-;i++) id[i]=id[i-]+;
        id[n-k]=k; for(int i=n-k+;i<=n-;i++) id[i]=id[i-]+;

        for(int i=;i<n;i++)
            for(int j=;j<n;j++) val[i][j]=INF;
        memset(f,,sizeof f);

        for(int i=;i<=m;i++)
        {
            int a,b,c; scanf("%d%d%d",&a,&b,&c);
            a--; b--; a=id[a]; b=id[b];
            val[a][b]=min(val[a][b],c);
            val[b][a]=val[a][b];
            g[a].push_back(b);
            g[b].push_back(a);
        }

        int st = <<(*k);

        for(int j=;j<st;j++)
            for(int i=;i<n;i++) d[i][j]=INF;

        for(int i=;i<n;i++)
        {
            if(i<*k) d[i][<<i]=;
            else d[i][]=;
        }

        for(int j=;j<st;j++)
        {
            for(int i=;i<n;i++)
            {
                if(i<*k)
                {
                    if(((<<i)&j)==) continue;
                    for (int x = j; x; x = (x-)&j)
                    {
                        int A=x ,B=j-A;
                        if(d[i][A|(<<i)]!=INF&&d[i][B|(<<i)]!=INF)
                            d[i][j] = min(d[i][j], d[i][A|(<<i)]+d[i][B|(<<i)]);
                    }
                }
                else
                {
                    for (int x = j; x; x = (x-)&j)
                    {
                        int A=x ,B=j-A;
                        if(d[i][A]!=INF&&d[i][B]!=INF)
                            d[i][j] = min(d[i][j], d[i][A]+d[i][B]);
                    }
                }

                if(d[i][j]!=INF) Q.push(i*+j);
            }
            spfa();
        }

        for(int j=;j<st;j++)
        {
            dp[j]=INF;
            for(int i=;i<n;i++) dp[j]=min(dp[j],d[i][j]);
        }

        for(int j=;j<st;j++)
        {
            for (int x = j; x; x = (x-)&j)
            {
                int A=x ,B=j-A;
                if(check(A)&&check(B)&&dp[A]!=INF&&dp[B]!=INF)
                    dp[j]=min(dp[j],dp[A]+dp[B]);
            }
        }

        if(dp[st-]!=INF) printf("%d\n",dp[st-]);
        else printf("No solution\n");
    }
    return ;
}