LeetCode：2603. 收集树中金币详解（2023.9.21 C++）

最新推荐文章于 2024-11-02 14:05:28 发布

Cosmoshhhyyy

最新推荐文章于 2024-11-02 14:05:28 发布

阅读量329

点赞数 2

分类专栏： LeetCode 文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/Cosmoshhhyyy/article/details/133153957

版权

LeetCode 专栏收录该内容

272 篇文章 5 订阅

订阅专栏

2603. 收集树中金币

题目描述：

给你一个 n 个节点的无向无根树，节点编号从 0 到 n - 1 。给你整数 n 和一个长度为 n - 1 的二维整数数组 edges ，其中 edges[i] = [ai, bi] 表示树中节点 ai 和 bi 之间有一条边。再给你一个长度为 n 的数组 coins ，其中 coins[i] 可能为 0 也可能为 1 ，1 表示节点 i 处有一个金币。

一开始，你需要选择树中任意一个节点出发。你可以执行下述操作任意次：

收集距离当前节点距离为 2 以内的所有金币，或者
移动到树中一个相邻节点。

你需要收集树中所有的金币，并且回到出发节点，请你返回最少经过的边数。

如果你多次经过一条边，每一次经过都会给答案加一。

示例 1：

输入：coins = [1,0,0,0,0,1], edges = [[0,1],[1,2],[2,3],[3,4],[4,5]]
输出：2
解释：从节点 2 出发，收集节点 0 处的金币，移动到节点 3 ，收集节点 5 处的金币，然后移动回节点 2 。

示例 2：

输入：coins = [0,0,0,1,1,0,0,1], edges = [[0,1],[0,2],[1,3],[1,4],[2,5],[5,6],[5,7]]
输出：2
解释：从节点 0 出发，收集节点 4 和 3 处的金币，移动到节点 2 处，收集节点 7 处的金币，移动回节点 0 。

提示：

n == coins.length
1 <= n <= 3 * 104
0 <= coins[i] <= 1
edges.length == n - 1
edges[i].length == 2
0 <= ai, bi < n
ai != bi
edges 表示一棵合法的树。

实现代码与解析：

拓扑 + bfs

class Solution {
public:
    vector<int> h = vector<int>(30010, -1), e = vector<int>(60010, 0), ne = vector<int>(60100, 0), d = vector<int>(30010, 0);
    int idx = 0;
    void add(int a, int b)
    {
        e[idx] = b; ne[idx] = h[a]; h[a] = idx++;
    } 

    int collectTheCoins(vector<int>& coins, vector<vector<int>>& edges) {
        

        int n = coins.size(); // 结点个数
        int rest = n; // 剩余的结点 rest翻译剩余

        for (auto t: edges)
        {
            int a = t[0];
            int b = t[1];
            add(a, b), add(b, a); // 加无向边
            d[a]++, d[b]++; // 两个结点 入度++
        }

        queue<int> q;
        // 删除树中 无金币 的叶子结点
        for (int i = 0; i < n; i++) 
            if (d[i] == 1 && !coins[i]) q.push(i); // 无向边，入度为1的是叶子节点，要和有向的区分开
        while (q.size())
        {
            int t = q.front();
            d[t]--; // d[t]-- 此结点从1变为0 相当于删除此节点和边
            rest--; // 剩余结点--
            q.pop();
            for (int i = h[t]; ~i; i = ne[i])
            {
                int j = e[i];
                d[j]--; // 相邻结点 入度--
                if (d[j] == 1 && !coins[j]) q.push(j); // 入度为1的继续入队，因为有可能有新0金币结点 因为删除 变为叶子结点
            }
        }

        // 拓扑（拓扑排序中的两个去除环节 执行两次） 去除叶子结点 两次
        for (int i = 0; i < 2; i++)
        {
            queue<int> qq;

            for (int i = 0; i < n; i++)
                if (d[i] == 1) qq.push(i); // 叶子结点入队
            
            while(qq.size())
            {
                int t = qq.front();
                d[t]--;
                qq.pop();
                rest--;
                for (int i = h[t]; ~i; i = ne[i])
                {
                    int j = e[i];
                    d[j]--;
                } 
            }
        }

        return rest == 0 ? 0 : 2 * (rest - 1);
    }
};