cf1042F. Leaf Sets

链接

点击跳转

题解

这题的做法是这样的

自低向上合并，对于一个节点，先把其子树中的每个集合最深的的深度列一张表统计下来，然后不断取最小的和次小的合并

可以先这样想：每次合并两棵子树中的集合，如果两个来自不同子树的两个集合可以合并，那么深度和肯定不超过$k$，并且合并之后新的深度就等于原来的$max$，其实就相当于大的那个把小的那个给“吸收”了，合并的结果就是集合少了一个，答案减少了$1$，那么我就吸收的越多越好，每次看看最小的那个能否被“吸收”，显然只需要看看次小的能否把它“吸收”就行了，每个子树都贪心的这样搞一搞，接下去再处理父亲节点…依次类推

这样可以得到一个启发式合并的做法

但是稍微优化一下就可以少一个$log$，就是子树自身合并完了之后，没必要把所有的集合信息都传递给父亲节点，只需要传递那个深度最小的节点，因为别的子树来和它合并的时候，如果能跟深度更深的合并，肯定也能和它合并，所以干脆不要那些深度比较深的了

代码

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define maxe 2000010
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
#define de(x) cerr<<#x<<" = "<<x<<endl
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
int n, k, deg[maxn], dep[maxn], ans;
struct Graph
{
    int etot, head[maxn], to[maxe], next[maxe];
    void clear(int N)
    {
        for(int i=1;i<=N;i++)head[i]=0;
        etot=0;
    }
    void adde(int a, int b){to[++etot]=b;next[etot]=head[a];head[a]=etot;}
    #define forp(_,__) for(auto p=__.head[_];p;p=__.next[p])
}G;
int dfs(int pos, int pre)
{
    dep[pos]=dep[pre]+1;
    if(deg[pos]==1){ans++;return dep[pos];}
    vector<int> v;
    forp(pos,G)if(G.to[p]!=pre)v.emb(dfs(G.to[p],pos));
    sort(v.begin(),v.end(),greater<int>());
    while(v.size()>1 and v.back()+v.at(v.size()-2)-2*dep[pos]<=k)v.pop_back(), ans--;
    return v.back();
}
int main()
{
    int i, u, v, r=-1;
    n=read(), k=read();
    rep(i,n-1)
    {
        u=read(), v=read();
        G.adde(u,v), G.adde(v,u);
        deg[u]++, deg[v]++;
    }
    rep(i,n)if(deg[i]>1)break;
    dfs(i,0);
    cout<<ans;
    return 0;
}