美文网首页
斯坦纳树

斯坦纳树

作者: 云中翻月 | 来源:发表于2020-11-24 21:39 被阅读0次

适用题目特征

给定含有K(K \leq 20)个关键点的图中,求可选择额外点使得所有关键点连通的最小代价。

原理

最终连通子图必然为树,于是可以将所有关键点是否连通到树中作为状态进行状态压缩。通过枚举根节点是否已经进入集合,来进行DP转移。转移时对于同一个连通集合,可以利用dijkstra算法在遍历图的过程中顺带完成。

例题

Luogu P6192
代码如下

/*
Luogu P6192 
*/
#define method_1
#ifdef method_1
/*

*/
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iomanip>
#include<ctime>
#include<string>
#include<bitset>
#define D(x) cout<<#x<<" = "<<x<<"  "
#define E cout<<endl
#define rep(i,s,t) for(int i=(s);i<=(t);i++)
#define rep0(i,s,t) for(int i=(s);i<(t);i++)
#define rrep(i,t,s) for(int i=(t);i>=(s);i--)
using namespace std;
typedef long long ll;
typedef pair<int,int>pii;
const int maxn=100+5;
const int maxm=510+5; 
const int maxk=10+2;
const int INF=0x3f3f3f3f;
int n,m,k;
struct node{
    int from,to,w;
}edge[maxm<<1];
int head[maxn],tot=1;
int tr[maxm<<1];
void add(int from,int to,int w){
    edge[++tot].from=head[from],head[from]=tot,edge[tot].to=to,edge[tot].w=w;
}
int f[maxn][1<<maxk];
int key[maxk];
int v[maxn];
void init(){
    memset(f,INF,sizeof(f));
}
priority_queue<pii>q;
void dijkstra(int s){
    memset(v,0,sizeof(v));
    while(q.size()){
        int x=q.top().second;q.pop();
        if(v[x]) continue;
        v[x]=1;
        for(int i=head[x];i;i=edge[i].from){
            int y=edge[i].to,w=edge[i].w;
            if(f[tr[i]][s]>f[x][s]+w){
                f[tr[i]][s]=f[x][s]+w;
                q.push(make_pair(-f[tr[i]][s],tr[i]));
            }
        }
    }
}
void solve(){
    rep0(s,1,1<<k){
        rep(i,1,n){
            for(int subs=s&(s-1);subs;subs=s&(subs-1)) f[i][s]=min(f[i][s],f[i][subs]+f[i][s^subs]);
            if(f[i][s]!=INF) q.push(make_pair(-f[i][s],i));
        }
        dijkstra(s);
    }
    cout<<f[key[1]][(1<<k)-1];
}
int main() {
//  ios::sync_with_stdio(false);
//  freopen("最小斯坦纳树.in","r",stdin);
    cin>>n>>m>>k;
    init();
    int from,to,w;
    rep(i,1,m){
        cin>>from>>to>>w;
        add(from,to,w);tr[tot]=to;
        add(to,from,w);tr[tot]=from;
    }
    rep(i,1,k) cin>>key[i],f[key[i]][1<<i-1]=0;
    solve();
    return 0;
}
#endif
#ifdef method_2
/*

*/

#endif
#ifdef method_3
/*

*/

#endif

相关文章

网友评论

      本文标题:斯坦纳树

      本文链接:https://www.haomeiwen.com/subject/hmvwiktx.html