AT3912 Antennas on Tree

AT3912 Antennas on Tree

%%zzt

只能考虑性质了。

把最后选择的k个点的连通块求出来，连通块内部的点表示都是互异的

连通块外部的点只能形成若干条链，并且这k个点的每一个最多与一个外部点相连。

直接从每个叶子往上跳，当fa[x]度数>2的时停下

如果fa[x]之前没有链，那么可以先省下一次，否则就要把x变成k个点之一

#include<bits/stdc++.h>

#define reg register int

#define il inline

#define fi first

#define se second

#define mk(a,b) make_pair(a,b)

#define numb (ch^'0')

#define pb push_back

#define solid const auto &

#define enter cout<<endl

#define pii pair<int,int>

using namespace std;

typedef long long ll;

template<class T>il void rd(T &x){

    char ch;x=;bool fl=false;while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);

    for(x=numb;isdigit(ch=getchar());x=x*+numb);(fl==true)&&(x=-x);}

template<class T>il void output(T x){if(x/)output(x/);putchar(x%+'');}

template<class T>il void ot(T x){if(x<) putchar('-'),x=-x;output(x);putchar(' ');}

template<class T>il void prt(T a[],int st,int nd){for(reg i=st;i<=nd;++i) ot(a[i]);putchar('\n');}

namespace Modulo{

const int mod=;

int ad(int x,int y){return (x+y)>=mod?x+y-mod:x+y;}

void inc(int &x,int y){x=ad(x,y);}

int mul(int x,int y){return (ll)x*y%mod;}

void inc2(int &x,int y){x=mul(x,y);}

int qm(int x,int y=mod-){int ret=;while(y){if(y&) ret=mul(x,ret);x=mul(x,x);y>>=;}return ret;}

}

//using namespace Modulo;

namespace Miracle{

const int N=1e5+;

int du[N],n;

vector<int>to[N];

int fa[N],ans;

bool vis[N],has[N];

void dfs(int x){

    for(solid y:to[x]){

        if(y==fa[x]) continue;

        fa[y]=x;

        dfs(y);

    }

}

int main(){

    rd(n);int x,y;

    int rt=;

    for(reg i=;i<n;++i){

        rd(x);rd(y);++x;++y;

        to[x].pb(y);to[y].pb(x);

        ++du[x];++du[y];

        if(du[x]>) rt=x;

        if(du[y]>) rt=y;

    }

    if(!rt) printf("");

    else{

        dfs(rt);

        for(reg i=;i<=n;++i){

            if(du[i]==){

                // cout<<" ii "<<i<<endl;

                int x=i;

                while(du[fa[x]]<=) x=fa[x];

                // cout<<" xx "<<x<<endl;

                if(!has[fa[x]]){

                    has[fa[x]]=;

                }else{

                    if(!vis[x]){

                        vis[x]=;

                        ++ans;

                    }

                }

            }

        }

        ot(ans);

    }

    return ;

}

}

signed main(){

    Miracle::main();

    return ;

}

/*

   Author: *Miracle*

*/