代码随想录算法训练营第五十六天 | 图论part06

108. 冗余连接

cpp 复制代码
#include <iostream>
#include <vector>

using namespace std;

void init(vector<int> &father) {
    for (int i = 0; i < father.size(); ++i) {
        father[i] = i;
    }
}

int find(vector<int>& father, int u) {
    if (father[u] == u) return u;
    return father[u] = find(father, father[u]);
}

int isSame(vector<int>& father, int u, int v) {
    u = find(father, u);
    v = find(father, v);
    return u == v;
}

void join(vector<int>& father, int u, int v) {
    u = find(father, u);
    v = find(father, v);
    if (u == v) return;
    father[v] = u;
}
int main()
{
    int n, u, v, resultu, resultv;
    cin >> n;
    vector<int> father(n + 1, 0);
    init(father);
    for (int i = 0; i < n; ++i) {
        cin >> u >> v;
        if (isSame(father, u, v)) {
            resultu = u;
            resultv = v;
        }
        else
            join(father, u, v);
    }
    cout << resultu << " " << resultv << endl;
    return 0;
}

109. 冗余连接II

关键是要能够分出两种情况:

  1. 是不是存在入度为2的点,然后判断哪一条边去掉,还是连通的。在这里也有两种情况,一种是直接删除最后一个边,一种是会形成环。
  2. 是不是存在环
cpp 复制代码
#include <iostream>
#include <vector>

using namespace std;

void init(vector<int>& father) {
    for (int i = 0; i < father.size(); ++i) {
        father[i] = i;
    }
}

int find(vector<int>& father, int u) {
    if (father[u] == u) return u;
    return father[u] = find(father, father[u]);
}

int isSame(vector<int>& father, int u, int v) {
    u = find(father, u);
    v = find(father, v);
    return u == v;
}

void join(vector<int>& father, int u, int v) {
    u = find(father, u);
    v = find(father, v);
    if (u == v) return;
    father[v] = u;
}

bool isTreeAfterRemove(vector<int>& father, const vector<vector<int>>& edges, int deleteEdge) {
    init(father);
    for (int i = 0; i < edges.size(); ++i) {
        if (i == deleteEdge) continue;
        if (isSame(father, edges[i][0], edges[i][1]))
            return false;
        join(father, edges[i][0], edges[i][1]);
    }
    return true;
}

void getRemoveEdge(vector<int>& father, const vector<vector<int>>& edges) {
    init(father);
    for (vector<int> pair : edges) {
        if (isSame(father, pair[0], pair[1])) {
            cout << pair[0] << " " << pair[1] << endl;
            return;
        }
        join(father, pair[0], pair[1]);
    }
}

int main()
{
    int n, u, v;
    cin >> n;
    vector<int> father(n + 1, 0);
    vector<vector<int>> edges;
    vector<int> inDegrees(n + 1, 0);
    for (int i = 0; i < n; ++i) {
        cin >> u >> v;
        inDegrees[v]++;
        edges.emplace_back(vector<int>{u, v});
    }

    vector<int> vec;
    for (int i = n - 1; i >= 0; --i) {
        if (inDegrees[edges[i][1]] == 2)
            vec.push_back(i);
    }
    if (vec.size() > 0) {
        if (isTreeAfterRemove(father, edges, vec[0])) {
            cout << edges[vec[0]][0] << " " << edges[vec[0]][1] << endl;
        }
        else
            cout << edges[vec[1]][0] << " " << edges[vec[1]][1] << endl;
        return 0;
    }
    getRemoveEdge(father, edges);
    return 0;
}
相关推荐
然我17 分钟前
链表指针玩不转?从基础到双指针,JS 实战带你破局
前端·数据结构·算法
EndingCoder25 分钟前
算法与前端的可访问性
前端·算法·递归·树形结构
ysa0510301 小时前
竞赛常用加速技巧#模板
c++·笔记·算法
7 971 小时前
C语言基础知识--文件的顺序读写与随机读写
java·数据结构·算法
2401_841003981 小时前
Kubernetes 资源管理全解析
算法·贪心算法
☆璇2 小时前
【数据结构】排序
c语言·开发语言·数据结构·算法·排序算法
艾莉丝努力练剑5 小时前
【LeetCode&数据结构】单链表的应用——反转链表问题、链表的中间节点问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
_殊途6 小时前
《Java HashMap底层原理全解析(源码+性能+面试)》
java·数据结构·算法
珊瑚里的鱼10 小时前
LeetCode 692题解 | 前K个高频单词
开发语言·c++·算法·leetcode·职场和发展·学习方法
秋说11 小时前
【PTA数据结构 | C语言版】顺序队列的3个操作
c语言·数据结构·算法