从固定点开始遍历图的所有生成树

struct Edge { int u, v, eid; };
struct Graph {
	int  m_vn;
	vector<Edge> m_edges;
	vector<vector<int>> m_adj_list;
	vector<int> m_vert_depth;
	vector<int> m_select_tree_edges;
	vector<int> m_candi_edges;

	Graph() { m_vn = 0; }
	Graph(int _n, vector<pair<int, int>> edges){
		m_vn = _n;
		m_adj_list.resize(_n);
		m_vert_depth.resize(_n, 0);
		for (auto e : edges) 
			AddEdge(e.first, e.second);
	}

	void AddEdge(int u, int v) {
		int eid = m_edges.size();
		m_adj_list[u].push_back(eid);
		m_edges.push_back({ u,v,eid });
	}

	void SelectEdge(int selected_vid, int start) {
		if (m_select_tree_edges.size() == m_vn-1) {
			cout << "tree edges:=================================" << endl;
			for (int eid : m_select_tree_edges)
				cout << m_edges[eid].u << "->" << m_edges[eid].v << endl;
			return;
		}
		//expand the candidate edges from selcted vertex's adj verts
		int from = m_candi_edges.size();
		for (int& eid : m_adj_list[selected_vid])
			if (!m_vert_depth[m_edges[eid].v])
				m_candi_edges.push_back(eid);

		//select an edge from the candidate set
		for (int k = start; k < m_candi_edges.size(); ++k) {
			Edge& e = m_edges[m_candi_edges[k]];
			if (!m_vert_depth[e.v]) {
				m_select_tree_edges.push_back(m_candi_edges[k]);
				m_vert_depth[e.v] = m_vert_depth[e.u] + 1;
				SelectEdge(e.v, k + 1);
				m_vert_depth[e.v] = 0;
				m_select_tree_edges.pop_back();
			}
		}
		//when the edge(x,selected_vid) is removed, then remove its adj edges 
		m_candi_edges.erase(m_candi_edges.begin() + from, m_candi_edges.end());
	}

	void EnumAllSpanningTreeFrom(int root) {
		vector<int> cand_edges;
		m_vert_depth[root] = 1;
		SelectEdge(root,0);
	}
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Researcher-Du

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值