洛谷P3915（树的分解）（思路+AC代码）

六年级蒟蒻来题解了（洛谷名：zldx）

题目传送门

题目描述

给出 N个点的树和 K，问能否把树划分成 N/K个连通块，且每个连通块的点数都是 K。

输入格式

第一行，一个整数 T，表示数据组数。接下来 T组数据，对于每组数据：

第一行，两个整数 N, K。

接下来 N - 1行，每行两个整数 Ai, Bi，表示边 (Ai, Bi)。点用 1, 2, ...， N编号。

输出格式

对于每组数据，输出 YES 或 NO。

样例 #1

样例输入 #1

2
4 2
1 2
2 3
3 4
4 2
1 2
1 3
1 4

样例输出 #1

YES
NO

提示

• 对于 60 %的数据,1 <=N, K <=10^3；
• 对于 100 %的数据，1 <= T <= 10，1 <= N ,K <= 10^5。

思路：

为了能满足题目的要求，我们得将一个树去把他分解开来。那我们先来想：我们可以先把这几组数据换成一个树，然后如果我们想从上面开始枚举是不行的，因为这样会考虑不到下面的子树，从而就十分的麻烦。那既然上面不行那么我们就从下面往上面去想。多推几个例子（普通和特殊交替），我们会发现在每一个小树如果他的数量比k大的话那就无法分配，如果等于的话那就刚刚好可以作为一个区块，如果小于的话就把它的个数算在更大一点的树上。这样一直推下去就可以完成这样操作。

如：

 代码：

#include<bits/stdc++.h>
using namespace std;
typedef vector<int> ve;
typedef vector<vector<int>> ve1;
ve1 hhh;
int fl;
int n,k;
vector<bool> b(n,false);
int ditui(int x){
//    cout<<endl;
    if(fl==1){
        return -1;
    }
    if(b[x]==1){
        return 0;
    }
    int sum=1;
    b[x]=true;
    for(auto i:hhh[x]){
        sum+=ditui(i);
        if(fl==1){
            return -1;
        }
    }
    if(sum>k){
        fl=1;
        return -1;
    }else{
        return sum%k;
    }
}
int main() {
    int t;
    cin>>t;
    for(int i=0;i<t;i++){
        cin>>n>>k;
        hhh=ve1(n+1);
        b=vector<bool>(n+1,false);
        for(int j=0;j<n-1;j++){
            int a,b;
            cin>>a>>b;
            hhh[a].push_back(b);
            hhh[b].push_back(a);
        }
        fl=0;
        int num=ditui(1);
        if(num==0){
            cout<<"YES"<<endl;
        }else{
            cout<<"NO"<<endl;
        }
    }
    return 0;
}

如果对您有帮助的话，求求你点赞、关注、收藏！！求求了！！！（磕头） ‘砰砰砰’