斯坦纳树 学习总结

所谓斯坦纳树,就是一类使关键点联通的最优化问题OwO

考试的时候被考了一发,然后就被炸飞了(然而不知道为什么其他人都会

所以就做了点题目

首先裸的斯坦纳树是这样的

设f(i,S)表示当前在第i个点,关键点联通状态为S的最优代价

转移有两种:

第一种是合并两个联通块,f(i,S)=max{f(i,T)+f(i,S^T)}

在这里注意到T和S^T在i点是联通的,所以可以将其合并

第二种是扩展当前状态,即 f(i,S)=max{f(j,S)+edge(i,j)}

注意到这一步的转移是没有固定顺序的,所以我们要利用SPFA来进行转移

其实也是挺好写的,每次只是先做一遍子集枚举,之后把有用的状态放入队列跑SPFA

 

BZOJ 4006 管道连接

这里要求相同频道的情报站要连接,但并不要求所有关键情报站都连接

所以我们可以利用斯坦纳树求出形成每个可能的联通块的最优代价

最后在做一次DP即可,即g(S)=max(g(T)+g(S^T))

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iostream>
#include<queue>
using namespace std;
 
const int maxn=1010;
int n,m,p,u,v,w,oo;
int h[maxn],cnt=0;
struct edge{
    int to,next,w;
}G[20010];
int g[1050];
int f[maxn][1050];
int c[maxn],d[maxn];
int bin[maxn];
bool vis[maxn];
queue<int>Q;
 
void add(int x,int y,int z){
    ++cnt;G[cnt].to=y;G[cnt].next=h[x];G[cnt].w=z;h[x]=cnt;
}
void read(int &num){
    num=0;char ch=getchar();
    while(ch<'!')ch=getchar();
    while(ch>='0'&&ch<='9')num=num*10+ch-'0',ch=getchar();
}
void SPFA(int st){
    while(!Q.empty()){
        int u=Q.front();Q.pop();
        for(int i=h[u];i;i=G[i].next){
            int v=G[i].to;
            if(f[v][st]>f[u][st]+G[i].w){
                f[v][st]=f[u][st]+G[i].w;
                if(!vis[v])Q.push(v),vis[v]=true;
            }
        }vis[u]=false;
    }return;
}
bool check(int S){
    for(int i=1;i<=p;++i){
        int now=(S&bin[i]);
        if(now==0||now==bin[i])continue;
        return false;
    }return true;
}
 
int main(){
    read(n);read(m);read(p);
    for(int i=1;i<=m;++i){
        read(u);read(v);read(w);
        add(u,v,w);add(v,u,w);
    }
    for(int i=1;i<=p;++i)scanf("%d%d",&c[i],&d[i]);
    for(int i=1;i<=p;++i){
        for(int j=1;j<=p;++j)if(c[j]==c[i])bin[i]|=(1<<(j-1));
    }
    memset(g,0x3f,sizeof(g));memset(f,0x3f,sizeof(f));
    for(int i=1;i<=p;++i)f[d[i]][1<<(i-1)]=0;oo=g[0];
    for(int S=1;S<(1<<p);++S){
        for(int i=1;i<=n;++i){
            for(int T=S-1;T;T=(T-1)&S){
                f[i][S]=min(f[i][S],f[i][T]+f[i][S^T]);
            }
            if(f[i][S]<oo)Q.push(i),vis[i]=true;
        }SPFA(S);
        if(!check(S))continue;
        for(int i=1;i<=n;++i)g[S]=min(g[S],f[i][S]);
        for(int T=S-1;T;T=(T-1)&S)g[S]=min(g[S],g[T]+g[S^T]);
    }printf("%d\n",g[(1<<p)-1]);
    return 0;
}

hdu 4085

跟上一道题大同小异,也是有一个多余的限制

同样做DP就可以了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<queue>
using namespace std;

const int maxn=52;
int T,n,m,d,u,v,w,oo,lim;
int h[maxn],cnt=0;
struct edge{
    int to,next,w;
}G[10010];
bool vis[52];
int g[1050];
int f[52][1050];
bool tmp[1050];
queue<int>Q;

inline void add(int x,int y,int z){
    ++cnt;G[cnt].to=y;G[cnt].next=h[x];G[cnt].w=z;h[x]=cnt;
}
inline void read(int &num){
    num=0;char ch=getchar();
    while(ch<'!')ch=getchar();
    while(ch>='0'&&ch<='9')num=num*10+ch-'0',ch=getchar();
}
inline bool check(int S){
    int cnt=0;
    for(int i=1;i<=d;++i)if(S>>(i-1)&1)cnt++;
    for(int i=d+1;i<=lim;++i)if(S>>(i-1)&1)cnt--;
    return cnt==0;
}
inline void SPFA(int st){
    while(!Q.empty()){
        int u=Q.front();Q.pop();
        for(int i=h[u];i;i=G[i].next){
            int v=G[i].to;
            if(f[v][st]>f[u][st]+G[i].w){
                f[v][st]=f[u][st]+G[i].w;
                if(!vis[v])Q.push(v),vis[v]=true;
            }
        }vis[u]=false;
    }return;
}
int main(){
    read(T);
    while(T--){
        memset(h,0,sizeof(h));cnt=0;
        read(n);read(m);read(d);lim=(d<<1);
        for(int i=1;i<=m;++i){
            read(u);read(v);read(w);
            add(u,v,w);add(v,u,w);
        }
        memset(g,0x3f,sizeof(g));oo=g[0];
        memset(f,0x3f,sizeof(f));
        for(int i=1;i<=d;++i)f[i][1<<(i-1)]=0;
        for(int i=d+1;i<=lim;++i)f[n-(lim-i)][1<<(i-1)]=0;
        for(int S=1;S<(1<<lim);++S)tmp[S]=check(S);
        for(int S=1;S<(1<<lim);++S){
            for(int i=1;i<=n;++i){
                for(int T=S-1;T;T=(T-1)&S){
                    f[i][S]=min(f[i][S],f[i][T]+f[i][S^T]);
                }
                if(f[i][S]<oo)Q.push(i),vis[i]=true;
            }SPFA(S);
            if(!tmp[S])continue;
            for(int i=1;i<=n;++i)g[S]=min(g[S],f[i][S]);
            for(int T=S-1;T;T=(T-1)&S)g[S]=min(g[S],g[T]+g[S^T]);
        }
        if(g[(1<<lim)-1]==oo)printf("No solution\n");
        else printf("%d\n",g[(1<<lim)-1]);
    }return 0;
}

hdu 3331

首先考虑如果所有点都是关键点的话,显然是USACO的灌水OwO

做法就是对于建立虚拟节点,向每个点连边,边权为挖井费用

然后Kruscal就可以了

对应到这道题目上我们建立虚拟节点之后可以看作要使关键点和虚拟节点都联通

就变成斯坦纳树的裸题了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<queue>
using namespace std;

const int maxn=1010;
int n,m,p,u,v,w,oo;
int h[maxn],cnt=0;
struct edge{
    int to,next,w;
}G[50010];
bool vis[maxn];
int f[maxn][72];
queue<int>Q;

void add(int x,int y,int z){
    ++cnt;G[cnt].to=y;G[cnt].next=h[x];G[cnt].w=z;h[x]=cnt;
}
void SPFA(int st){
    while(!Q.empty()){
        int u=Q.front();Q.pop();
        for(int i=h[u];i;i=G[i].next){
            int v=G[i].to;
            if(f[v][st]>f[u][st]+G[i].w){
                f[v][st]=f[u][st]+G[i].w;
                if(!vis[v])Q.push(v),vis[v]=true;
            }
        }vis[u]=false;
    }return;
}

int main(){
    while(scanf("%d%d%d",&n,&m,&p)==3){
        memset(h,0,sizeof(h));cnt=0;
        for(int i=1;i<=n+m;++i){
            scanf("%d",&w);
            add(0,i,w);add(i,0,w);
        }
        for(int i=1;i<=p;++i){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);add(v,u,w);
        }
        memset(f,0x3f,sizeof(f));n++;oo=f[0][0];
        for(int i=0;i<n;++i)f[i][1<<i]=0;
        for(int S=1;S<(1<<n);++S){
            for(int i=0;i<=n+m;++i){
                for(int T=S-1;T;T=(T-1)&S){
                    f[i][S]=min(f[i][S],f[i][T]+f[i][S^T]);
                }
                if(f[i][S]<oo)Q.push(i),vis[i]=true;
            }SPFA(S);
        }
        int ans=oo;
        for(int i=0;i<=n;++i)ans=min(ans,f[i][(1<<n)-1]);
        printf("%d\n",ans); 
    }return 0;
}

貌似这类题目很容易看出来?

因为基本上给定关键点的不是虚树就是斯坦纳树OwO

转载于:https://www.cnblogs.com/joyouth/p/5645626.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: Python实现最小斯坦纳树的代码可以使用Prim算法来解决。具体实现如下: ```python import sys # 为了方便表示的邻接矩阵,使用无穷大代表不可达 inf = sys.maxsize def prim(graph): num_vertices = len(graph) key = [inf] * num_vertices # 记录顶点到最小生成树的最小权值边 parent = [None] * num_vertices # 记录最小生成树中顶点的父节点 visited = [False] * num_vertices # 记录顶点是否已访问 # 将第一个顶点设为起始顶点 key[0] = 0 for _ in range(num_vertices): # 找到未访问的顶点中键值最小的顶点 min_key = inf min_vertex = None for v in range(num_vertices): if not visited[v] and key[v] < min_key: min_key = key[v] min_vertex = v # 将找到的顶点标记为已访问 visited[min_vertex] = True # 更新顶点的最小权值边和父节点 for v in range(num_vertices): if not visited[v] and graph[min_vertex][v] < key[v]: key[v] = graph[min_vertex][v] parent[v] = min_vertex return parent def min_steiner_tree(graph, terminals): num_terminals = len(terminals) # 构建终端间的最短路径 shortest_paths = [[inf] * num_terminals for _ in range(num_terminals)] for i in range(num_terminals): for j in range(num_terminals): shortest_paths[i][j] = dijkstra(graph, terminals[i], terminals[j]) # 在最短路径上生成最小斯坦纳树 steiner_tree = [[inf] * num_terminals for _ in range(num_terminals)] for i in range(num_terminals): for j in range(num_terminals): if i == j: steiner_tree[i][j] = 0 else: for k in range(num_terminals): steiner_tree[i][j] = min(steiner_tree[i][j], shortest_paths[i][k] + shortest_paths[k][j]) # 使用Prim算法生成最小生成树 parent = prim(steiner_tree) return parent # 测试代码 graph = [[0, 7, 9, inf, inf, 14], [7, 0, 10, 15, inf, inf], [9, 10, 0, 11, inf, 2], [inf, 15, 11, 0, 6, inf], [inf, inf, inf, 6, 0, 9], [14, inf, 2, inf, 9, 0]] terminals = [0, 2, 4] parent = min_steiner_tree(graph, terminals) print(parent) ``` 此代码是使用Prim算法在最短路径上生成最小斯坦纳树。输入的是一个邻接矩阵,其中inf表示顶点之间不可达。terminals是终端节点的列表。输出是一个列表,表示每个顶点在生成的最小斯坦纳树中的父节点。 ### 回答2: Python实现最小斯坦纳树的代码可以使用最小生成树算法和动态规划的思想。 首先,我们可以使用Prim算法或Kruskal算法找到最小生成树,即连接所有顶点的最小权重的子。 接下来,对于每一条边,我们通过遍历所有顶点集合的子集来找到最小斯坦纳树。子集的大小从1开始递增,直到包含所有顶点为止。 对于每个子集,我们通过动态规划的方法来找到连接子集中所有顶点的最小权重的边。 具体的实现步骤如下: 1. 使用Prim算法或Kruskal算法找到最小生成树,并保存最小生成树的边集合。 2. 对于每条边e in 最小生成树的边集合: 2.1 对于每个顶点集合V'(从1个元素开始递增到总顶点数): 2.1.1 如果V'包含边e的两个顶点,则忽略该顶点集合。 2.1.2 否则,遍历V'的所有子集V'': 2.1.2.1 如果V''中不包含边e的两个顶点,则忽略该子集。 2.1.2.2 否则,计算通过V''中的顶点连接边e的权重和,并更新最小权重值和对应的边。 3. 最后得到的最小权重值和对应的边即为最小斯坦纳树的结果。 以下是一个简单的Python代码示例: ```python import math def minimum_steiner_tree(graph): n = len(graph) inf = float('inf') dp = [[inf] * n for _ in range(1 << n)] for v in range(n): dp[1 << v][v] = 0 for S in range(1 << n): for v in range(n): for u in range(n): dp[S | (1 << u)][u] = min(dp[S | (1 << u)][u], dp[S][v] + graph[v][u]) return min(dp[-1]) # 测试代码 graph = [[0, 2, 3, math.inf], [2, 0, 1, 3], [3, 1, 0, 2], [math.inf, 3, 2, 0]] result = minimum_steiner_tree(graph) print("最小斯坦纳树的权重为:", result) ``` 权重矩阵graph表示的是无向的邻接矩阵,math.inf表示无穷大,表示两个顶点之间没有边。代码中的结果为最小斯坦纳树的权重。 ### 回答3: Python最小斯坦纳树的代码可以通过使用Dijkstra算法和回溯法来实现。以下是一个可能的实现: ```python import sys def dijkstra(graph, src): n = len(graph) dist = [sys.maxsize] * n dist[src] = 0 visited = [False] * n for _ in range(n): u = min_distance(dist, visited) visited[u] = True for v in range(n): if graph[u][v] > 0 and not visited[v] and dist[v] > dist[u] + graph[u][v]: dist[v] = dist[u] + graph[u][v] return dist def min_distance(dist, visited): min_dist = sys.maxsize min_index = -1 for v in range(len(dist)): if not visited[v] and dist[v] < min_dist: min_dist = dist[v] min_index = v return min_index def tsp_solver(graph, start): n = len(graph) tsp_path = None tsp_cost = sys.maxsize def tsp_recursion(curr_node, visited, current_path, current_cost): nonlocal tsp_path, tsp_cost if len(visited) == n: if graph[curr_node][start] > 0: current_cost += graph[curr_node][start] current_path.append(start) if current_cost < tsp_cost: tsp_cost = current_cost tsp_path = current_path.copy() current_path.pop() current_cost -= graph[curr_node][start] return for next_node in range(n): if next_node not in visited: new_path = current_path.copy() new_path.append(next_node) tsp_recursion(next_node, visited + [next_node], new_path, current_cost + graph[curr_node][next_node]) tsp_recursion(start, [start], [start], 0) return tsp_path, tsp_cost def min_steiner_tree(graph, terminals): n = len(graph) t = len(terminals) dp = [[sys.maxsize] * t for _ in range(1 << t)] # 动态规划表格 path = [[None] * t for _ in range(1 << t)] # 记录路径 for i in range(t): dist = dijkstra(graph, terminals[i]) for j in range(t): dp[1 << i][j] = dist[terminals[j]] for i in range(1 << t): for j in range(t): if dp[i][j] == sys.maxsize: continue for k in range(t): if (i >> k) & 1 == 0 and dp[i][j] + dp[1 << k | i][k] < dp[1 << k | i][k]: dp[1 << k | i][k] = dp[i][j] + dp[1 << k | i][k] path[1 << k | i][k] = j min_cost = sys.maxsize min_path = None for i in range(t): if dp[(1 << t) - 1][i] < min_cost: min_cost = dp[(1 << t) - 1][i] min_path = [i] while len(min_path) < t: last_node = min_path[-1] min_path.append(path[(1 << t) - 1][last_node]) min_path = [terminals[i] for i in min_path] tsp_path, tsp_cost = tsp_solver(graph, terminals[0]) min_cost += tsp_cost min_path += tsp_path[1:] return min_path, min_cost # 测试例子 graph = [ [0, 2, 3, 0, 0], [2, 0, 0, 4, 0], [3, 0, 0, 1, 3], [0, 4, 1, 0, 2], [0, 0, 3, 2, 0] ] terminals = [1, 2, 3] path, cost = min_steiner_tree(graph, terminals) print("最小斯坦纳树路径:", path) print("最小斯坦纳树总成本:", cost) ``` 这段代码通过调用`min_steiner_tree`函数来计算给定和终端点集合的最小斯坦纳树的路径和成本。`graph`代表的邻接矩阵,`terminals`代表终端点的列表。最后将得到的最小斯坦纳树路径和成本打印出来。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值