关于《算法概论》中,使用dfs对图进行标记的操作的实践

其实本身是在做leetcode 685. Redundant Connection II
题目说一定以根节点开头,但实际好像并非如此…感觉被坑了。
原先的思路是利用书中所说的dfs,选好根节点,进行pre和post标记,根据每个节点的标记判断前向边,横跨边,回边。
不过测试会出现第一个边是需要删除的边的情况。。。

就权当对bfs的边判断的练手了

class Solution {
public:
    bool isBack(int preA, int postA, int preB, int postB) {
        if (preB < preA && preA < postA && postA < postB) return true;
        return false;
    }
    bool isCross(int preA, int postA, int preB, int postB) {
        if (preB < postB && postB < preA && preA < postA) return true;
        return false;
    }
    bool isForward(int preA, int postA, int preB, int postB) {
        if (preA < preB && preB < postB && postB < postA) return true;
        return false;
    }
    vector<int> findRedundantDirectedConnection(vector<vector<int>>& edges) {
        vector<vector<int>> graph;
        vector<bool> visit;
        int N = edges.size();
        graph.resize(N+1);
        visit.resize(N+1);
        for (int i = 0; i < N+1; i++) visit[i] = false;
        for (auto x: edges) {
            graph[x[0]].push_back(x[1]);
        }
        //dfs
        stack<int> s;
        int tag = 0;
        vector<int> pre;
        vector<int> post;
        pre.resize(N+1);
        post.resize(N+1);
        s.push(edges[0][0]);
        visit[edges[0][0]] = true;
        pre[edges[0][0]] = tag++;
        while (!s.empty()) {
            int cur = s.top();
            //cout << cur << endl;
            bool find = false;
            for (int i = 0; i < graph[cur].size(); i++) {
                if (!visit[graph[cur][i]]) {
                    s.push(graph[cur][i]);
                    visit[graph[cur][i]] = true;
                    pre[graph[cur][i]] = tag++;
                    find = true;
                    break;
                }
            }
            if (!find) {
                post[cur] = tag++;
                s.pop();
            }
        }
        vector<int> result;
        for (int i = edges.size()-1; i >= 0; i--) {
            if (isBack(pre[edges[i][0]], post[edges[i][0]], pre[edges[i][1]], post[edges[i][1]])||
               isCross(pre[edges[i][0]], post[edges[i][0]], pre[edges[i][1]], post[edges[i][1]])) {
                result.push_back(edges[i][0]);
                result.push_back(edges[i][1]);
                return result;
            }
        }
        if (result.size() == 0) {
            for (int i = edges.size()-1; i >= 0; i--) {
                if (isForward(pre[edges[i][0]], post[edges[i][0]], pre[edges[i][1]], post[edges[i][1]])) {
                    result.push_back(edges[i][0]);
                    result.push_back(edges[i][1]);
                    return result;
                }
            }
        }
        return result;
    }
};

注:该段代码并不能通过测试(开头不是根的情况无法通过)
但还是有意义的。利用bfs对各个节点定下标记:
通过pre标记和post标记的大小判断边的种类

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值