CodeForces 1385F :Removing Leaves 拓扑排序

最新推荐文章于 2021-03-13 11:29:10 发布

匿枫

最新推荐文章于 2021-03-13 11:29:10 发布

阅读量82

点赞数

本文链接：https://blog.csdn.net/tlyzxc/article/details/113947952

版权

传送门

题目描述

给定一棵树，现在定义一次操作为"选定拥有同样父亲的
k个叶节点，并将这个叶节点一起删去"。请问，最多能够进行多少次操作。

分析

我们贪心的去想，如果有叶节点符合条件，是否存在不需要删除的情况，答案是不会，因为删除的叶子结点去其他节点并不会产生影响，所以所有符合条件的叶子结点都必须要删除
所以我们只需要暴力的删除所有符合条件的叶子结点就可以了
叶子结点的度数为1，所以我们可以采用拓扑排序的方法

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 2e5 + 10;
const ll mod= 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a){char c=getchar();T x=0,f=1;while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
while(isdigit(c)){x=(x<<1)+(x<<3)+c-'0';c=getchar();}a=f*x;}
int gcd(int a,int b){return (b>0)?gcd(b,a%b):a;}
int h[N],ne[N * 2],e[N * 2],idx;
int in[N],si[N],n,k;
bool st[N];
int ans;

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

void run(){
    queue<int> Q;
    for(int i = 1;i <= n;i++)
        if(in[i] == 1) Q.push(i);
    while(Q.size()){
        int t = Q.front();
        Q.pop();
        st[t] = true;
        for(int i = h[t];~i;i = ne[i]){
            int j = e[i];
            if(st[j]) continue;
            si[j]++,in[j]--;
            if(si[j] % k == 0){
                ans++;
                if(in[j] == 1) Q.push(j);
            }
        }
    }
}

void init(){
    memset(h,-1,sizeof h);
    memset(in,0,sizeof in);
    memset(si,0,sizeof si);
    memset(st,0,sizeof st);
    ans = idx = 0;
}

int main(){
    int T;
    read(T);
    while(T--){
        init();
        read(n),read(k);
        for(int i = 1;i < n;i++){
            int x,y;
            read(x),read(y);
            add(x,y);
            add(y,x);
            in[x]++,in[y]++;
        }
        run();
        di(ans);
    }
    return 0;
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*/