首页 > 编程语言 >AcWing算法基础课笔记——最小生成树与二分图

AcWing算法基础课笔记——最小生成树与二分图

时间:2024-06-09 22:29:47浏览次数:28  
标签:二分 return int res st 基础课 dist find AcWing

目录

朴素版prim算法 —— 模板题 AcWing 858. Prim算法求最小生成树

时间复杂度是 O(n2+m)O(n2+m), nn 表示点数,mm 表示边数

int n;      // n表示点数
int g[N][N];        // 邻接矩阵,存储所有边
int dist[N];        // 存储其他点到当前最小生成树的距离
bool st[N];     // 存储每个点是否已经在生成树中


// 如果图不连通,则返回INF(值是0x3f3f3f3f), 否则返回最小生成树的树边权重之和
int prim()
{
    memset(dist, 0x3f, sizeof dist);

    int res = 0;
    for (int i = 0; i < n; i ++ )
    {
        int t = -1;
        for (int j = 1; j <= n; j ++ )
            if (!st[j] && (t == -1 || dist[t] > dist[j]))
                t = j;
    
        if (i && dist[t] == INF) return INF;
    
        if (i) res += dist[t];
        st[t] = true;
    
        for (int j = 1; j <= n; j ++ ) dist[j] = min(dist[j], g[t][j]);
    }
    
    return res;

}

题目代码

#include<iostream>
#include<cstring>
using namespace std;
const int N = 510, INF = 0x3f3f3f3f;

int n, m;
int g[N][N];
int dist[N];
bool st[N];

int prim() {
	memset(dist, 0x3f, sizeof dist);
	
	int res = 0;
	for(int i = 0; i < n; i ++ ) {
		//找到距离集合最小的,没有被访问的点 
		int t = -1;
		for(int j = 1; j <= n; j ++ ) {
			if(!st[j] && (t == -1 || dist[j] < dist[t]) )
				t = j;
		}
		
		//如果该点不是0,且距离为INF,说明是不连通的 
		if(i && dist[t] == INF) return INF;
		//如果i!=0,那么更新res。 i=0时,第一个点的dist是INF 
		if(i) res += dist[t];
		
		//更新每个点到集合的距离 
		for(int j = 1; j <= n; j ++ ) dist[j] = min(dist[j], g[t][j]);
		
		//设置t为已访问 
		st[t] = true;
	}
	
	return res;
}

int main() {
	scanf("%d%d", &n, &m);
	
	memset(g, 0x3f, sizeof g);
	
	while(m --) {
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		g[a][b] = g[b][a] = c;
	}
	
	int t = prim();
	
	if(t == INF) puts("impossible");
	else printf("%d\n", t);
	
	return 0; 
} 

Kruskal算法 —— 模板题 AcWing 859. Kruskal算法求最小生成树

时间复杂度是 O(mlogm)O(mlogm), nn 表示点数,mm 表示边数

int n, m;       // n是点数,m是边数
int p[N];       // 并查集的父节点数组

struct Edge     // 存储边
{
    int a, b, w;

    bool operator< (const Edge &W)const
    {
        return w < W.w;
    }

}edges[M];

int find(int x)     // 并查集核心操作
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int kruskal()
{
    sort(edges, edges + m);

    for (int i = 1; i <= n; i ++ ) p[i] = i;    // 初始化并查集
    
    int res = 0, cnt = 0;
    for (int i = 0; i < m; i ++ )
    {
        int a = edges[i].a, b = edges[i].b, w = edges[i].w;
    
        a = find(a), b = find(b);
        if (a != b)     // 如果两个连通块不连通,则将这两个连通块合并
        {
            p[a] = b;
            res += w;
            cnt ++ ;
        }
    }
    
    if (cnt < n - 1) return INF;
    return res;

}

题目代码

#include<iostream>
#include<algorithm> 
using namespace std;

const int N = 200010;

int n, m;
int p[N]; //并查集祖宗节点 

struct Edge {
	int a, b, w;
	
	bool operator< (const Edge &W) const {
		return w < W.w;
	}
}edges[N];

int find(int x) {
	if(p[x] != x) p[x] = find(p[x]);
	return p[x];
}

int main() {
	scanf("%d%d", &n, &m);
	
	for(int i = 0; i < m; i ++ ) {
		int a, b, w;
		scanf("%d%d%d", &a, &b, &w);
		edges[i] = {a, b, w};
	}
	
	//按权重排序 
	sort(edges, edges + m);
	
	//初始化并查集 
	for(int i = 1; i <= n; i ++ ) p[i] = i;
	
	int res = 0, cnt = 0;
	//从小到大列举每条边,若a,b没有在一个集合中,就将他们之间的边加入 
	for(int i = 0; i <= m; i ++ ) {
		int a = edges[i].a, b = edges[i].b, w = edges[i].w;
		
		a = find(a), b = find(b);
		if(a != b) {
			p[a] = b;
			res += w;
			cnt ++;
		}
	}
	
	if(cnt < n - 1) puts("impossible");
	else printf("%d", res);
	
	return 0;
}

染色法判别二分图 —— 模板题 AcWing 860. 染色法判定二分图

时间复杂度是 O(n+m)O(n+m), nn 表示点数,mm 表示边数

int n;      // n表示点数
int h[N], e[M], ne[M], idx;     // 邻接表存储图
int color[N];       // 表示每个点的颜色,-1表示为染色,0表示白色,1表示黑色

// 参数:u表示当前节点,c表示当前点的颜色
bool dfs(int u, int c)
{
    color[u] = c;
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (color[j] == -1)
        {
            if (!dfs(j, !c)) return false;
        }
        else if (color[j] == c) return false;
    }

    return true;

}

bool check()
{
    memset(color, -1, sizeof color);
    bool flag = true;
    for (int i = 1; i <= n; i ++ )
        if (color[i] == -1)
            if (!dfs(i, 0))
            {
                flag = false;
                break;
            }
    return flag;
}

题目代码

#include<iostream>
#include<cstring> 
using namespace std;
const int N = 100010, M = 2 * N;

int n, m;
int h[N], e[M], ne[M], idx;
int color[N];

void add(int a, int b) {
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx ++;
}

//用c染色u 
bool dfs(int u, int c) {
	color[u] = c;
	
	//循环u的相邻的边进行染色,最后u所在连通块都被染色 
	for(int i = h[u]; i != -1; i = ne[i]) {
		int j = e[i];
		//如果j没有染色,则用3 - c染色j 
		if(!color[j]) {
			if(!dfs(j, 3 - c)) return false; 
		}
		// 如果j染色是错的,则返回false
		else if(color[j] == c) return false; 
	}
	
	//没有问题返回true,此时 u所在连通块都被染色 
	return true;
}

int main() {
	scanf("%d%d", &n, &m);
	
	memset(h, -1, sizeof h);
	
	while(m -- ) {
		int a, b;
		scanf("%d%d", &a, &b);
		add(a, b), add(b, a);
	}
	
	bool flag = true;
	for(int i = 1; i <= n; i ++ ) {
		if(!color[i]) {  //若i没有染色,则用1染色染色
			if(!dfs(i, 1)) {
				// 染色失败
				flag = false;
				break; 
			}
		}
	}
	
	if(flag) puts("Yes");
	else puts("No");
	
	return 0;
}

匈牙利算法 —— 模板题 AcWing 861. 二分图的最大匹配

时间复杂度是 O(nm)O(nm), nn 表示点数,mm 表示边数

int n1, n2;     // n1表示第一个集合中的点数,n2表示第二个集合中的点数
int h[N], e[M], ne[M], idx;     // 邻接表存储所有边,匈牙利算法中只会用到从第二个集合指向第一个集合的边,所以这里只用存一个方向的边
int match[N];       // 存储第二个集合中的每个点当前匹配的第一个集合中的点是哪个
bool st[N];     // 表示第二个集合中的每个点是否已经被遍历过

bool find(int x)
{
    for (int i = h[x]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!st[j])
        {
            st[j] = true;
            if (match[j] == 0 || find(match[j]))
            {
                match[j] = x;
                return true;
            }
        }
    }

    return false;

}

// 求最大匹配数,依次枚举第一个集合中的每个点能否匹配第二个集合中的点
int res = 0;
for (int i = 1; i <= n1; i ++ )
{
    memset(st, false, sizeof st);
    if (find(i)) res ++ ;
}

题目代码

#include<iostream>
#include<cstring>

const int N = 510, M = 100010;

int n1, n2, m;
int h[N], e[M], ne[M], idx;
int match[N];
bool st[N];

void add(int a, int b) {
	e[idx] = b;
	ne[idx] = h[a];
	h[a] = idx ++;
}

bool find(int x) {
	for(int i = h[x]; i != -1; i = ne[i]) {
		int j = e[i];
		if(!st[j]) {
			st[j] = true;
			if(match[j] == 0 || find(match[j])) {
				match[j] = x;
				return true;
			}
		}
	}
	
	return false;
}

int main() {
	scanf("%d%d%d", &n1, &n2, &m);
	
	memset(h, -1, sizeof h);
	
	while(m -- ) {
		int a, b;
		scanf("%d%d", &a, &b);
		add(a, b);
	}
	
	int res = 0;
	for(int i = 1; i <= n1; i ++ ) {
		memset(st, false, sizeof st); // st表示的是当前男生是否访问过该女生
		
		if(find(i))  res ++;
	}
	
	printf("%d\n", res);
	
	return 0;
}

标签:二分,return,int,res,st,基础课,dist,find,AcWing
From: https://blog.csdn.net/Sophia2021XJTU/article/details/139567265

相关文章

  • AcWing算法基础课笔记——求最短路算法
    目录朴素dijkstra算法——模板题AcWing849.Dijkstra求最短路I题目代码堆优化版dijkstra——模板题AcWing850.Dijkstra求最短路II题目代码Bellman-Ford算法——模板题AcWing853.有边数限制的最短路题目代码spfa算法(队列优化的Bellman-Ford算法)——......
  • AcWing 33:链表中倒数第k个节点 ← 尾插法
    【题目来源】https://www.acwing.com/problem/content/32/【题目描述】输入一个链表,输出该链表中倒数第k个结点。注意:  ●k>=1;  ●如果k大于链表长度,则返回NULL;【数据范围】链表长度[0,30]。【输入样例】输入:链表:1->2->3->4->5,k=2【输出样例】输出:4......
  • 8621 二分查找
    Description 编写Search_Bin函数,实现在一个递增有序数组ST中采用折半查找法确定元素位置的算法.输入格式第一行:元素个数n第二行:依次输入n个元素的值(有序)第三行:输入要查找的关键字key的值输出格式输出分两种情形:1.如果key值存在,则输出其在表中的位置x(表位置从0开......
  • C++ 6.8笔记:①判断质数②二分基础算法
    质数试除法判定质数boolprimes(intx){  if(x<2)returnfalse;  for(inti=2;i<=x/i;i++){    if(x%i==0)returnfalse;  }  returntrue;}埃筛1intp[N],k,n;boolf[N];voidprimes(intn){//埃筛,思想:质数的倍数是合数for(inti......
  • 二分查找相关题目(c++)
    1.元素编号输入 n个单调不减的(就是后面的数字不小于前面的数字)非负整数a1​,a2​,…,an​​,然后进行 m次询问。对于每次询问,给出一个整数 q,要求输出数列中第一个大于等于q的数字的编号。(若未找到则默认输出-1)输入共3行:第1行,2个整数n和m,表示数字个数和询问次数;......
  • AcWing 1211:蚂蚁感冒 ← 模拟题
    【题目来源】https://www.acwing.com/problem/content/1213/【题目描述】长100厘米的细长直杆子上有n只蚂蚁。它们的头有的朝左,有的朝右。每只蚂蚁都只能沿着杆子向前爬,速度是1厘米/秒。当两只蚂蚁碰面时,它们会同时掉头往相反的方向爬行。这些蚂蚁中,有1只蚂蚁感冒......
  • PyTorch实现二分类任务
    importtorchimporttorch.nnasnnimporttorch.optimasoptim'''定义模拟数据集'''inputs=torch.randn(5,10)#定义5个样本,每个样本的特征数位10labels=torch.tensor([0,1,0,1,1])#每个样本的标签,二分类,所以为0或1'''定义模型''......
  • 禹晶、肖创柏、廖庆敏《数字图像处理(面向新工科的电工电子信息基础课程系列教材)》Chap
    禹晶、肖创柏、廖庆敏《数字图像处理(面向新工科的电工电子信息基础课程系列教材)》Chapter7插图......
  • Acwing 786.第K个数
    Acwing786.第K个数题目描述786.第k个数-AcWing题库运行代码#include<iostream>#include<algorithm>usingnamespacestd;constintN=100010;intq[N];intmain(){intn,k;scanf("%d%d",&n,&k);for(inti=0;i<n......
  • 代码随想录算法训练营第一天 | 704. 二分查找 27. 移除元素
    704.二分查找题目:给定一个n个元素有序的(升序)整型数组和一个目标值target,写一个函数搜索nums中的target,如果目标值存在返回下标,否则返回-1。提示:1.你可以假设nums中的所有元素是不重复的。2.n将在[1,10000]之间。3.nums的每个元素都将在[-9999,9999]之间。解题:思路:二......