题解 | #动物牛的探险之旅#

动物牛的探险之旅

https://www.nowcoder.com/practice/14553bb44c854257ac3a86da2b35dfdd

核心思想就是:三类边一定比一类和二类更好。此外,如果有三类边形成的环,则打破这个环。

#include <queue>
#include <unordered_map>
#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param n int整型 
     * @param edges int整型vector<vector<>> 
     * @return int整型
     */
    
    int ret = 0;
    void dfs(vector<vector<bool>>& A, vector<vector<bool>>& A_both, unordered_set<int> &vis, int cur, int last){
        if(vis.count(cur)) return;
        vis.insert(cur);
        if(last!=-1){
            for(auto u: vis){
                if(u==last) continue;
                if(A[u][cur]){
                    ret++;
                    A[u][cur] = false;
                    A[cur][u] = false;
                }
            }
        }
        for(int i=0;i<A.size();i++){
            if(A_both[cur][i]) dfs(A, A_both, vis, i, cur);
        }
        for(int i=0;i<A.size();i++){
            if(A[cur][i]) dfs(A, A_both, vis, i, cur);
        }
    }

    void dfs_both(vector<vector<bool>>& A, unordered_set<int> &vis, int cur, int last){
        if(vis.count(cur)) return;
        vis.insert(cur);
        if(last!=-1){
            for(auto u: vis){
                if(u==last) continue;
                if(A[u][cur]){
                    ret++;
                    A[u][cur] = false;
                    A[cur][u] = false;
                }
            }
        }
        for(int i=0;i<A.size();i++){
            if(A[cur][i]) dfs(A, A, vis, i, cur);
        }
    }
    
    int maxRemovableEdges(int n, vector<vector<int> >& edges) {
        unordered_set<int> vis_a;
        unordered_set<int> vis_b;
        unordered_set<int> vis_ab;
        vector<vector<bool>> A_a(n, vector<bool>(n, false));
        vector<vector<bool>> A_b(n, vector<bool>(n, false));
        vector<vector<bool>> A_ab(n, vector<bool>(n, false));
        for(auto edge : edges){
            int type = edge[0], u = edge[1]-1, v = edge[2]-1;
            if(A_ab[u][v]){
                ret++;
                continue;
            }
            if(type==3){
                if(A_a[u][v]){
                    A_a[u][v] = false;
                    A_a[v][u] = false;
                    ret++;
                }
                if(A_b[u][v]){
                    A_b[u][v] = false;
                    A_b[v][u] = false;
                    ret++;
                }
                A_ab[u][v] = true;
                A_ab[v][u] = true;
            }
            else if(type==1){
                if(A_a[u][v]){
                    ret++;
                }
                else {
                    A_a[u][v] = true;
                    A_a[v][u] = true;
                }
            }
            else if(type==2){
                if(A_b[u][v]){
                    ret++;
                }
                else {
                    A_b[u][v] = true;
                    A_b[v][u] = true;
                }
            }
        }
        dfs(A_a, A_ab, vis_a, 0, -1);
        if(vis_a.size()!=n) return -1;
        dfs(A_b, A_ab, vis_b, 0, -1);
        if(vis_b.size()!=n) return -1;
        for(int i=0;i<n;i++){
            if(!vis_ab.count(i)) dfs_both(A_ab, vis_ab, 0, -1);
        }
        return ret;
    }
};

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-08 13:15
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-10 11:55
点赞 评论 收藏
分享
叶扰云倾:进度更新,现在阿里云面完3面了,感觉3面答得还行,基本都答上了,自己熟悉的地方也说的比较细致,但感觉面试官有点心不在焉不知道是不是不想要我了,求阿里收留,我直接秒到岗当阿里孝子,学校那边的房子都退租了,下学期都不回学校,全职猛猛实习半年。这种条件还不诱人吗难道 然后现在约到了字节的一面和淘天的复活赛,外加猿辅导。华为笔试完没动静。 美团那边之前投了个base广州的,把我流程卡麻了,应该是不怎么招人,我直接简历挂了,现在进了一个正常的后端流程,还在筛选,不知道还有没有hc。
点赞 评论 收藏
分享
头顶尖尖的程序员:我也是面了三四次才放平心态的。准备好自我介绍,不一定要背熟,可以记事本写下来读。全程控制语速,所有问题都先思考几秒,不要急着答,不要打断面试官说话。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务