924尽量减少恶意软件的传播

最新推荐文章于 2024-07-23 22:06:37 发布

akoasm..

最新推荐文章于 2024-07-23 22:06:37 发布

阅读量200

点赞数 3

文章标签：深度优先算法

本文链接：https://blog.csdn.net/2302_78952982/article/details/137807389

版权

本题采用广度优先搜索的方法

先回顾一下广度优先搜索下面的是二叉树的广度优先搜索而题目类似于图论的广度优先搜索

// 102.二叉树的层序遍历
class Solution {
    public List<List<Integer>> resList = new ArrayList<List<Integer>>();

    public List<List<Integer>> levelOrder(TreeNode root) {
        //checkFun01(root,0);
        checkFun02(root);

        return resList;
    }

    //DFS--递归方式
    public void checkFun01(TreeNode node, Integer deep) {
        if (node == null) return;
        deep++;

        if (resList.size() < deep) {
            //当层级增加时，list的Item也增加，利用list的索引值进行层级界定
            List<Integer> item = new ArrayList<Integer>();
            resList.add(item);
        }
        resList.get(deep - 1).add(node.val);

        checkFun01(node.left, deep);
        checkFun01(node.right, deep);
    }

    //BFS--迭代方式--借助队列
    public void checkFun02(TreeNode node) {
        if (node == null) return;
        Queue<TreeNode> que = new LinkedList<TreeNode>();
        que.offer(node);

        while (!que.isEmpty()) {
            List<Integer> itemList = new ArrayList<Integer>();
            int len = que.size();

            while (len > 0) {
                TreeNode tmpNode = que.poll();
                itemList.add(tmpNode.val);

                if (tmpNode.left != null) que.offer(tmpNode.left);
                if (tmpNode.right != null) que.offer(tmpNode.right);
                len--;
            }

            resList.add(itemList);
        }

    }
}

class Solution {

    public int minMalwareSpread(int[][] graph, int[] initial) {

        int n = graph.length;

        int[] ids = new int[n];

        Map<Integer, Integer> idToSize = new HashMap<Integer, Integer>();

        int id = 0;

        for (int i = 0; i < n; ++i) {

            if (ids[i] == 0) {

                id++;

                int size = 1;

                Queue<Integer> queue = new LinkedList<Integer>();

                queue.offer(i);

                ids[i] = id;

                while (!queue.isEmpty()) {

                    int u = queue.poll();

                    for (int v = 0; v < n; ++v) {

                        if (ids[v] == 0 && graph[u][v] == 1) {

                            size++;

                            queue.offer(v);

                            ids[v] = id;

                        }

                    }

                }

                idToSize.put(id, size);

            }

        }

        Map<Integer, Integer> idToInitials = new HashMap<Integer, Integer>();

        for (int u : initial) {

            idToInitials.put(ids[u], idToInitials.getOrDefault(ids[u], 0) + 1);

        }

        int ans = n + 1, ansRemoved = 0;

        for (int u : initial) {

            int removed = (idToInitials.get(ids[u]) == 1 ? idToSize.get(ids[u]) : 0);

            if (removed > ansRemoved || (removed == ansRemoved && u < ans)) {

                ans = u;

                ansRemoved = removed;

            }

        }

        return ans;

    }

}