2581. 统计可能的树根数目

最新推荐文章于 2024-07-04 16:16:29 发布

咔咔咔的

最新推荐文章于 2024-07-04 16:16:29 发布

阅读量411

点赞数 13

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

本文链接：https://blog.csdn.net/weixin_45256307/article/details/136411257

版权

leetcode 专栏收录该内容

475 篇文章 2 订阅

订阅专栏

2581. 统计可能的树根数目

题目链接：2581. 统计可能的树根数目

代码如下：

//树形动态规划
//参考：leetcode官方题解
class Solution {
public:
    int rootCount(vector<vector<int>>& edges, vector<vector<int>>& guesses, int k) {
        vector<vector<int>> g(edges.size()+1);
        unordered_set<long long> st;
        for(auto &v:edges)
        {
            g[v[0]].push_back(v[1]);
            g[v[1]].push_back(v[0]);
        }

        auto h=[&](int x,int y)->long long
        {
            return (long long)x<<20|y;
        };

        for(auto& v:guesses)
        {
            st.insert(h(v[0],v[1]));
        }

        int count=0,res=0;
        function<void(int,int)> dfs=[&](int x,int fat)->void
        {
            for(auto& y:g[x])
            {
                if(y==fat)
                    continue;
                count+=st.count(h(x,y));
                dfs(y,x);
            }
        };
        dfs(0,-1);

        function<void(int,int,int)> redfs=[&](int x,int fat,int count)
        {
            if(count>=k)
                res++;
            for(auto &y:g[x])
            {
                if(y==fat)
                    continue;
                redfs(y,x,count-st.count(h(x,y))+st.count(h(y,x)));
            }
        };
        redfs(0,-1,count);
        return res;
    }
};