hdu4085Peach Blossom Spring

链接：http://acm.hdu.edu.cn/showproblem.php?pid=4085

题意：有n个点m条无向边有边权，给定k，2*k<=n表示前k个点和最后k个点一一对应。求所有点和它对应的那个点连通的最小边权花费。

分析：斯塔纳树的题。因为看到k比较小很容易想到用状态压缩表示哪些点已经连通了，那么我们设状态dp[i][j]表示以i为根的树并且已经包含了j状态的点的最小花费，那么转移有：dp[i][j]=min(dp[i][j']+dp[i][j-j'])其中j'是j的子集，这里我们可以用3^K转移，还有dp[i][j]=min(dp[i'][j]+w[i][i'])表示我们可以通过一条和之前根相连的边更新一个新根的状态。我们会发现第二种转移很想spfa中的松弛操作，那么我们可以从小集合更新到大集合。然后对于这题来说它只要求两两对应的相连即可，那么可能是一棵森林我们需要用另一个check函数来表示合并两颗合法树的最小花费。详见代码。O(n*3*K)

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=1050;
const int mod=1000000007;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int INF=1000000010;
const ll MAX=1ll<<55;
const double eps=1e-5;
const double inf=~0u>>1;
const double pi=acos(-1.0);
typedef double db;
typedef long double ldb;
typedef unsigned int uint;
typedef unsigned long long ull;
queue<int>Q;
int n,m,k,K,T;
int e[15],q[510000],ans[N],dp[55][N];
int tot,u[55],v[2*N],w[2*N],pre[2*N];
inline int getnew(int a) {
    if (a<=k) return a;
    if (a>n-k) return a-n+2*k;
    return a+k;
}
inline void add(int a,int b,int c) {
    v[tot]=b;w[tot]=c;pre[tot]=u[a];u[a]=tot++;
    v[tot]=a;w[tot]=c;pre[tot]=u[b];u[b]=tot++;
}
inline void init() {
    K=2*k;T=1<<K;
    memset(q,0,sizeof(q));
    tot=0;memset(u,-1,sizeof(u));
    for (int i=1;i<=n;i++)
        for (int j=0;j<T;j++) dp[i][j]=INF;
    for (int i=1;i<=K;i++) dp[i][e[i-1]]=0;
}
inline void spfa() {
    int i,x,y;
    while (!Q.empty()) {
        x=Q.front();Q.pop();
        q[x]=0;y=x%10000;x/=10000;
        for (i=u[x];i!=-1;i=pre[i])
        if (dp[v[i]][y]>dp[x][y]+w[i]) {
            dp[v[i]][y]=dp[x][y]+w[i];
            if (!q[v[i]*10000+y]) q[v[i]*10000+y]=1,Q.push(v[i]*10000+y);
        }
    }
}
inline void Steiner_Tree() {
    for (int i=0;i<T;i++) {
        for (int j=i;j;j=(j-1)&i)
            for (int k=1;k<=n;k++) dp[k][i]=min(dp[k][i],dp[k][j]+dp[k][i-j]);
        for (int j=1;j<=n;j++)
        if (dp[j][i]!=INF) q[j*10000+i]=1,Q.push(j*10000+i);
        spfa();
    }
}
inline bool check(int a) {
    int i,ret=0;
    for (i=0;i<k;i++) {
        if (a&e[i]) ret++;
        if (a&e[i+k]) ret--;
    }
    return !ret;
}
int main()
{
    int i,j,t,a,b,c;
    scanf("%d", &t);
    for (e[0]=1,i=1;i<15;i++) e[i]=e[i-1]*2;
    while (t--) {
        scanf("%d%d%d", &n, &m, &k);
        init();
        for (i=1;i<=m;i++) {
            scanf("%d%d%d", &a, &b, &c);
            a=getnew(a);b=getnew(b);add(a,b,c);
        }
        Steiner_Tree();
        for (i=0;i<T;i++) {
            q[i]=check(i);ans[i]=INF;
            for (j=1;j<=n;j++) ans[i]=min(ans[i],dp[j][i]);
        }
        for (i=0;i<T;i++)
        if (q[i])
            for (j=i;j;j=(j-1)&i)
            if (q[j]) ans[i]=min(ans[i],ans[j]+ans[i-j]);
        if (ans[T-1]==INF) printf("No solution\n");
        else printf("%d\n", ans[T-1]);
    }
    return 0;
}