斯坦纳树

斯坦纳树:
给n个点m条边的无向图,然后给出k个点的集合,找一个最小的生成树使得包含这k个点(这个生成树必须包含这k个点,其他点可以有也可以没有,但是必须保证最后是个生成树,然后找最小的生成树)。

利用状压加最短路算法实现,目前没有多项式时间复杂度的算法。

dis[x][p]代表以x为根,k个点集合中选择了状态为p的点的最小生成树权值。

两个核心转移方程 dis[x][p]=min(dis[x][p],dis[x][i]+dis[x][p^i]);
和 dis[x][p]=min(dis[x][p],dis[y][p]+cost(x,y));

第一个方程通过枚举p以及p的子集加上枚举n个点实现,复杂度是 n ∗ 3 k n*3^k n3k
第二个方程通过最短路实现即可,最短路转移时,p不改变,因此和正常的最短路算法一样,只不过多了个p而已,复杂度是 2 k ∗ 2^k* 2ko(最短路)。
总的复杂度即为 n ∗ 3 k + 2 k ∗ n*3^k+2^k* n3k+2ko(最短路)。

初始化为把所有的dp[x][0或者1<<x]初始化为0,其余为INF。
例题:魔物消灭计划
这个题相当于模板题。
spfa写法:(如果有负权必须用spfa)

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int MAX_N=110;
const int INF=0x3f3f3f3f;
int a[MAX_N],id[MAX_N];
vector<int>v[MAX_N],w[MAX_N];
queue<int>q;
bool vis[MAX_N];
int dis[MAX_N][1100];
int k;
void spfa(int p){
    int i;
    while(!q.empty()){
        int x=q.front();
        q.pop();
        for(i=0;i<v[x].size();i++){
            int y=v[x][i];
            int cost=w[x][i];
            if(dis[y][p]>dis[x][p]+cost){
                dis[y][p]=dis[x][p]+cost;
                if(!vis[y]){
                    vis[y]=true;
                    q.push(y);
                }
            }
        }
        vis[x]=false;
    }
}
int main(void){
    int n,m,x,y,i,j,xx,yy,zz;
    scanf("%d%d%d%d%d",&n,&m,&k,&x,&y);
    memset(dis,INF,sizeof(dis));
    int st=k+1,ed=k+2;
    id[x]=st,id[y]=ed;
    int cnt=k+2;
    for(i=1;i<=n;i++){
        scanf("%d",&a[i]);
        if(i==x||i==y)
        continue;
        if(a[i]==0)
        id[i]=++cnt;
        else
        id[i]=a[i];
    }
    for(i=1;i<=m;i++){
        scanf("%d%d%d",&xx,&yy,&zz);
        if(xx==yy)
        continue;
        xx=id[xx],yy=id[yy];
        v[xx].push_back(yy);
        w[xx].push_back(zz);
        v[yy].push_back(xx);
        w[yy].push_back(zz);
    }
    int tt=(1<<(k+2)),t;
    for(i=1;i<=k+2;i++)
    dis[i][1<<(i-1)]=0;
    for(i=k+3;i<=cnt;i++)
    dis[i][0]=0;
    for(i=0;i<tt;i++){
        for(j=i;j;j=(j-1)&i){
            for(t=1;t<=cnt;t++){
                dis[t][i]=min(dis[t][i],dis[t][j]+dis[t][i^j]);
            }
        }
        for(j=1;j<=cnt;j++){
            if(dis[j][i]!=INF)
            q.push(j),vis[j]=true;
        }
        spfa(i);
    }
    printf("%d\n",dis[st][tt-1]);
    return 0;
} 

dijkstra写法:

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int MAX_N=110;
const int INF=0x3f3f3f3f;
int a[MAX_N],id[MAX_N];
vector<int>v[MAX_N],w[MAX_N];
bool vis[MAX_N];
int dis[MAX_N][1100];
int k;
struct skt{
	int x,d;
	bool friend operator<(skt a,skt b){
		return a.d>b.d;
	}
};
priority_queue<skt>q;
void dijkstra(int p){
    int i;
    while(!q.empty()){
        skt t=q.top();
        q.pop();
        int x=t.x;
        if(vis[x])
        continue;
        vis[x]=true;
        for(i=0;i<v[x].size();i++){
            int y=v[x][i];
            int cost=w[x][i];
            if(dis[y][p]>dis[x][p]+cost){
                dis[y][p]=dis[x][p]+cost;
                t.x=y,t.d=dis[y][p];
                q.push(t);
            }
        }
    }
}
int main(void){
    int n,m,x,y,i,j,xx,yy,zz;
    scanf("%d%d%d%d%d",&n,&m,&k,&x,&y);
    memset(dis,INF,sizeof(dis));
    int st=k+1,ed=k+2;
    id[x]=st,id[y]=ed;
    int cnt=k+2;
    for(i=1;i<=n;i++){
        scanf("%d",&a[i]);
        if(i==x||i==y)
        continue;
        if(a[i]==0)
        id[i]=++cnt;
        else
        id[i]=a[i];
    }
    for(i=1;i<=m;i++){
        scanf("%d%d%d",&xx,&yy,&zz);
        if(xx==yy)
        continue;
        xx=id[xx],yy=id[yy];
        v[xx].push_back(yy);
        w[xx].push_back(zz);
        v[yy].push_back(xx);
        w[yy].push_back(zz);
    }
    int tt=(1<<(k+2)),t;
    for(i=1;i<=k+2;i++)
    dis[i][1<<(i-1)]=0;
    for(i=k+3;i<=cnt;i++)
    dis[i][0]=0;
    for(i=0;i<tt;i++){
        for(j=i;j;j=(j-1)&i){
            for(t=1;t<=cnt;t++){
                dis[t][i]=min(dis[t][i],dis[t][j]+dis[t][i^j]);
            }
        }
        for(j=1;j<=cnt;j++){
            vis[j]=false;
            if(dis[j][i]!=INF){
                skt t;t.x=j,t.d=dis[j][i];
                q.push(t);
            }
        }
        dijkstra(i);
    }
    printf("%d\n",dis[st][tt-1]);
    return 0;
} 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值