abc325_E(最小生成树,建虚点)

E - Clique Connect (atcoder.jp)

第一种方法建虚点,1,2,3,4,5,...n点相互连接,相当于全部连接一个点m,这样计算最小生成树权值时会多算一条边,所以把所有边权值减去;

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
typedef pair<int,int>PII;
typedef priority_queue<int> upq;
typedef priority_queue<int,vector<int>,greater<int>> dpq;
const int M=998244353;
const int N=1e6+20;

struct edge{
	int u,v,val;
};
//=========================================================
bool FileIfstream(std::string name){
    std::ifstream f(name.c_str());
    return f.good();
}
//=========================================================
bool cmp(edge a,edge b)
{
    return a.val<b.val;
}
//=========================================================
int p[N];
int find(int a){
    if(p[a] != a) p[a] = find(p[a]);
    return p[a];
}
//=========================================================

vector<edge>e;
void solved()
{
	int n,m; cin>>n>>m;
	for(int i=1;i<=n+m;i++) p[i]=i;
	ll ans=0;
	for(int i=1;i<=m;i++)
	{
		int k,v;
		cin>>k>>v;
		for(int j=1;j<=k;j++){
			int x; cin>>x;
			e.push_back({i+n,x,v}); //建虚点
		}
		ans-=v;//算权值和时会多加一次,减去
	}
	sort(e.begin(),e.end(),cmp);
	int cnt=0;
	for(auto ed:e)
	{
		int pa=find(ed.u),pb=find(ed.v);
		if(pa!=pb) 
		{
			p[pa]=pb;
			ans+=ed.val;
			cnt++;
		}
	}
	if(cnt!=n+m-1) cout<<-1<<'\n';
	else cout<<ans<<'\n';
}
int main()
{
#ifndef ONLINE_JUDGE
    if(!FileIfstream("IO.in")){
        freopen("IO.in","w",stdout);
        return 0;
    }
    freopen("IO.in","r",stdin);
    freopen("IO.out","w",stdout);
#endif
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t=1; //cin>>t;
    while(t--)
    solved();
}

第二种方法,将输入的每一个连通块看成一个团,连通块的边权按从小到大排序,如果两个点不在一个并集里面就合并,权值加上团的边权,n--,如果n!=1的话说明不能构成最小生成树;

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
typedef pair<int,int>PII;
typedef priority_queue<int> upq;
typedef priority_queue<int,vector<int>,greater<int>> dpq;
const int M=998244353;
const int N=1e6+20;


//=========================================================
bool FileIfstream(std::string name){
    std::ifstream f(name.c_str());
    return f.good();
}
//=========================================================

//=========================================================
int p[N];
int find(int a){
    if(p[a] != a) p[a] = find(p[a]);
    return p[a];
}
//=========================================================

vector<PII>v[N];
void solved()
{
	int n,m; cin>>n>>m;
	for(int i=1;i<=n;i++) p[i]=i;
	for(int i=1;i<=m;i++)
	{
		int k,c; cin>>k>>c;
		for(int j=1;j<=k;j++)
		{
			int a; cin>>a;
			v[i].push_back({c,a});
		}
	}
	sort(v+1,v+1+m); ll ans=0;

	for(int i=1;i<=m;i++)
	{
		for(int j=0;j<v[i].size()-1;j++)
		{
			if(find(p[v[i][j].second])!=find(p[v[i][j+1].second])) 
			{
				p[find(v[i][j+1].second)]=find(p[v[i][j].second]);
				ans+=v[i][j].first;
				n--;
			}
		}
	}
	if(n!=1) cout<<-1;
	else cout<<ans;
}
int main()
{
#ifndef ONLINE_JUDGE
    if(!FileIfstream("IO.in")){
        freopen("IO.in","w",stdout);
        return 0;
    }
    freopen("IO.in","r",stdin);
    freopen("IO.out","w",stdout);
#endif
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t=1; //cin>>t;
    while(t--)
    solved();
}

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个使用Builder设计模式实现最小生成搜索器的Python代码示例: ```python from abc import ABC, abstractmethod from typing import List class Graph(ABC): @abstractmethod def add_edge(self, v1: int, v2: int, weight: int): pass @abstractmethod def minimum_spanning_tree(self): pass class Edge: def __init__(self, v1: int, v2: int, weight: int): self.v1 = v1 self.v2 = v2 self.weight = weight class MinimumSpanningTreeBuilder: def __init__(self, graph: Graph): self.graph = graph def build_minimum_spanning_tree(self) -> List[Edge]: self.graph.minimum_spanning_tree() return self.graph.edges class AdjacencyMatrixGraph(Graph): def __init__(self, num_vertices: int): self.num_vertices = num_vertices self.matrix = [[-1] * num_vertices for _ in range(num_vertices)] self.edges = [] def add_edge(self, v1: int, v2: int, weight: int): if v1 == v2: raise ValueError("Cannot add edge to itself") if weight < 1: raise ValueError("Weight must be a positive integer") self.matrix[v1][v2] = weight self.matrix[v2][v1] = weight self.edges.append(Edge(v1, v2, weight)) def minimum_spanning_tree(self): visited_vertices = [0] num_vertices = len(self.matrix[0]) edges = [] while len(visited_vertices) != num_vertices: shortest_edge = None for i in visited_vertices: for j in range(num_vertices): if self.matrix[i][j] != -1: edge = Edge(i, j, self.matrix[i][j]) if shortest_edge is None or edge.weight < shortest_edge.weight: shortest_edge = edge visited_vertices.append(shortest_edge.v2) edges.append(shortest_edge) self.edges = edges if __name__ == '__main__': graph = AdjacencyMatrixGraph(5) graph.add_edge(0, 1, 2) graph.add_edge(0, 3, 6) graph.add_edge(1, 2, 3) graph.add_edge(1,3, 8) graph.add_edge(1, 4, 5) graph.add_edge(2, 4, 7) graph.add_edge(3, 4, 9) builder = MinimumSpanningTreeBuilder(graph) minimum_spanning_tree = builder.build_minimum_spanning_tree() for edge in minimum_spanning_tree: print(f"({edge.v1}, {edge.v2}, {edge.weight})") ```

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值