求树的直径

        for (int mask = 3; mask <= (1 << n) - 1; mask++) { //所有子集
            int vis = 0;
            int res = 0;
            if ((mask & (mask - 1)) == 0) continue;//至少有两个点

            function<int(int)>dfs = [&](int i)->int{ //计算树的直径
                vis |= (1 << i);
                int maxlen = 0, sec = 0;
                for (int j : g[i]) {
                    if ((vis >> j & 1) == 0 && (mask >> j & 1)) {
                        int d = dfs(j);
                        if (d > maxlen) {
                            sec = maxlen;
                            maxlen = d;
                        } else if (d > sec) {
                            sec = d;
                        }
                    }
                }
                res = max(res, ma + sec);
                return ma + 1;
            };

            dfs(__builtin_ctz(mask));

            if (vis == mask)ans[res - 1]++;

        }



        return ans;
    }


};
  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
以下是Python求直径最小生成树的步骤: 1. 首先,需要使用Kruskal算法来生成最小生成树。 2. 然后,需要对最小生成树进行遍历,找到最长的边,即为直径。 3. 最后,输出直径和最小生成树。 下面是Python代码示例: ```python # 定义边的类 class Edge: def __init__(self, u, v, w): self.u = u self.v = v self.w = w # 定义并查集类 class UnionFind: def __init__(self, n): self.parent = list(range(n)) self.rank = [0] * n def find(self, x): if self.parent[x] != x: self.parent[x] = self.find(self.parent[x]) return self.parent[x] def union(self, x, y): px, py = self.find(x), self.find(y) if px == py: return False if self.rank[px] < self.rank[py]: px, py = py, px self.parent[py] = px if self.rank[px] == self.rank[py]: self.rank[px] += 1 return True # 定义Kruskal算法函数 def kruskal(n, edges): uf = UnionFind(n) edges.sort(key=lambda e: e.w) res = [] for e in edges: if uf.union(e.u, e.v): res.append(e) if len(res) == n - 1: break return res # 定义求直径函数 def diameter(n, edges): mst = kruskal(n, edges) dist = [-1] * n dist[mst[0].u] = 0 q = [mst[0].u] while q: u = q.pop(0) for e in mst: if e.u == u and dist[e.v] == -1: dist[e.v] = dist[u] + e.w q.append(e.v) elif e.v == u and dist[e.u] == -1: dist[e.u] = dist[u] + e.w q.append(e.u) u = max(range(n), key=lambda i: dist[i]) dist = [-1] * n dist[u] = 0 q = [u] while q: u = q.pop(0) for e in mst: if e.u == u and dist[e.v] == -1: dist[e.v] = dist[u] + e.w q.append(e.v) elif e.v == u and dist[e.u] == -1: dist[e.u] = dist[u] + e.w q.append(e.u) return max(dist) # 测试 n = 5 edges = [Edge(0, 1, 1), Edge(0, 2, 2), Edge(0, 3, 3), Edge(0, 4, 4), Edge(1, 2, 5), Edge(1, 3, 6), Edge(1, 4, 7), Edge(2, 3, 8), Edge(2, 4, 9), Edge(3, 4, 10)] print("最小生成树:", kruskal(n, edges)) print("直径:", diameter(n, edges)) ```

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值