洛谷 P3915 树的分解

最新推荐文章于 2024-08-18 18:02:49 发布

啥也不会hh

最新推荐文章于 2024-08-18 18:02:49 发布

阅读量150

点赞数

分类专栏：洛谷题单搜索文章标签：算法数据结构 c++ 深度优先

本文链接：https://blog.csdn.net/qq_61935738/article/details/126152439

版权

洛谷题单搜索专栏收录该内容

11 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何解决一道名为'树的分解'的算法问题。通过递归枚举每个连通块中是否存在指定数量的点，若找到则更新计数器，并在满足条件时判断总节点数是否能被目标数整除来确定解的存在性。代码实现使用了邻接表数据结构，遍历树的连通分量并计算大小。

摘要由CSDN通过智能技术生成

题目链接：

P3915 树的分解 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

解题思路：

现递归枚举每个连通块中是否有k个点，若存在k个点则将他置为0，且连通块为k个点的计数器++，最后只需要判断n / k == 计数器即可

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 100010, M = 2 * N;

int cnt;
int n, k;
int st[N];
int h[N], e[M], ne[M], idx;

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx;
    idx ++ ;
}

void dfs(int u, int fa)
{
    st[u] = 1;
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (j != fa)
        {
            dfs(j, u);
            st[u] += st[j];
        }
    }
    
    if (st[u] == k)
    {
        st[u] = 0;
        cnt ++ ;
    }
}

int main()
{
    int T;
    cin >> T;
    while (T -- )
    {
        cin >> n >> k;
        
        idx = 0;
        memset(h, -1, sizeof h);
        for (int i = 0; i < n - 1; i ++ )
        {
            int a, b;
            scanf("%d %d", &a, &b);
            add(a, b), add(b, a);
        }
        
        if (n % k != 0) //说明无解 
        {
            puts("NO");
            continue;
        }
        
        memset(st, 0, sizeof st);
        cnt = 0;
        dfs(1, -1);
        
        if (n / k == cnt) puts("YES");
        else puts("NO");
    }
    
    return 0;
}