Infected Tree(递归/树形)

Infected Tree

传送门 Problem - C - Codeforces

思路

一开始第一眼以为是树形dp，就先把每个节点下面的节点个数(不包括自己)全求出来了，然后后面发现没用…

然后又想到这个会不会是从下至上的dp。然后一个节点会被分为两种情况？被毒和不被毒。然后发现这个不太好想。思考过程中发现，一个节点中毒就跟一开始的根节点一样，这个过程和结构是类似的。而一个毒只会向下传递一层，而一层的另一个节点会被砍掉，最终的结果就是得到一个连续的且是一个节点一个节点串联起来的链子。说白了就是一个路径。此问题就转化为求出从根节点到度数为1或2的最短路径。

然后想到一个点的下方的节点有两种情况：
下面有两个节点，那么要么选择一个然后让另一个中毒，这一层的两个点对答案都没有贡献(所以层数*2)。其中，被砍掉的节点的下面的节点都对答案产生贡献，中毒的节点此时就像是一开始的根节点一样，继续处理，一直处理到节点度数为2或者为1。
下面只有一个节点，那么显然这个节点可以被砍掉然后后面的节点都被拯救，那么只有这个节点不会对答案有贡献(层数+1)。

最终的答案就是除了根节点以外所有点减去路径涉及到的节点个数。

#include <bits/stdc++.h>
typedef long double ld;
typedef long long ll;
#define pb push_back
#define mk make_pair
#define mt make_tuple
#define eb emplace_back
#define pob pop_back
#define rz resize
#define mem(a,b) memset(a,b,sizeof(a))
#define all(a) (a).begin(),(a).end()
#define rall(a) (a).rbegin(),(a).rend()
#define debug(a) cout<<#a<<"="<<a<<endl;
#define qwe(i,a,b) for(int i=a;i<=b;i++)
#define ewq(i,a,b) for(int i=a;i>=b;i--)
inline ll rr(){ll f=1,x=0;char ch;do{ch=getchar();if(ch=='-')f=-1;}while(ch<'0'||ch>'9');do{x=x*10+ch-'0';ch=getchar();}while(ch>='0'&&ch<='9');return f*x;}
using namespace std;
const ll INF=0x3f3f3f3f,inf=0x3f3f3f3f3f3f3f;
const ll mod[2]={int(1e9 + 7), 10007};
const int base[2]={29,31};
const int maxn=6e5+6;
ll qpow(ll x,ll n){ll ans=1;while(n>0){if(n&1)ans=ans*x%mod[1];x=x*x%mod[1];n>>=1;}return ans;}

int n,m;
int to[maxn],head[maxn],nxt[maxn],cnt; // 图
int f[maxn],degree[maxn]; // 树形dp值
void add_edge(int u,int v) {
    nxt[++cnt]=head[u];
    head[u]=cnt;
    to[cnt]=v;
}
void init(/* arguments */) {
    cnt=0;
    n=rr();
    for(int i=0;i<=n;i++) {
        head[i]=f[i]=degree[i]=0;
    }

    for(int i=1;i<n;i++) {
        int s=rr(),t=rr();
        add_edge(s,t);
        add_edge(t,s);
        degree[s]++,degree[t]++;
    }

}
int find_shortest_path(int u,int fa,int h) {
    int p=INF;
    if(degree[u]==1 && u==1) return h+1;
    if(u!=1 && degree[u]==1) {return h<<1;}
    if(u!=1 && degree[u]==2) {return (h<<1)+1;}
    for(int i=head[u];i;i=nxt[i]) {
        int v=to[i];
        if(v==fa) continue;
        p=min(find_shortest_path(v,u,h+1),p);
    }
    return p;
}
int main(int argc, char const *argv[]) {
    int t=rr();
    while (t--) {
        init();

        int p=find_shortest_path(1,0,0);
        std::cout << n-p-1 << '\n';
    }
    return 0;

}