Codeforces Round #801 (Div. 2) and EPIC Institute of Technology Round D. Tree Queries(树形dp)

题目

t(t<=1e4)组样例，每次给出一棵n(n<=2e5)个点的无根树，

出题人心中想一个树上点，记为点x，

求最小点数个数k，使得无论x是哪个点，

你询问这k个点与x的距离，出题人一一回答后，

你都能找出来出题人想的这个x点是什么，

你并不需要实际给出点号，只需要输出k的最小值

思路来源

dls代码、jiangly代码

题解

赛中写挫了，赛后也确实找到自己写的代码的反例，可能如果赛中写个对拍会好很多吧

赛中先特判掉了n=1和n=2，然后手玩了一下发现，

如果固定树上的两个点u、v，则相当于标记出树上u到v这条链，这条链上的点都可唯一确定，

到这条链上的点，相同深度的如果只有一个，也可以唯一标识，大于一个则不能标识，

没有显然的贪心策略，所以考虑树形dp，dp[i]表示只考虑i这棵子树的时候最少标记几个点，

如果标记的u、v都不是叶子，实际是不如标记的u、v都是叶子优的，因为可以向两端延长

然后发现，子树中如果有链，是有一条链可以不用标记点的，有一次免除机会

比如树(1-2,2-3,2-4)，如果标记了1号点和3号点，实际4号点是不用标记的，因为可以唯一确定

于是赛中就从某个叶子开始dfs，然后对子树dp，

最后特判了一下根是不是需要被标记，然后就挂了，还是有两个tips没有想出来

把叶子当根dfs的问题在于，当这个根直连了一个度=2的点的时候，

得记录一下这个度=2的点的子树里，这个免除机会有没有被用过，

如果度=2的点又连了一个度=2的点，就得再递归考虑，

所以，实际上2度点是要被缩掉的，这决定了这道题的写法：

1. 一条链的情况，实际只用链端一个点就可以确定，特判掉

2. 从某个度>=3的点开始dfs(如果从叶子/度=2的点开始dfs会有上述的递归考虑的问题,会比较难写)

心得

Codeforces Global Round 19 F.Towers(树形dp) 和这个树形dp可能类似

这个题当时也是没想到要把最大的h的点当根做树形dp，wa了若干发然后搞过去了

感觉难度可能在2100-2400之间，dp还是需要多手玩多找找性质，以及练练对拍找反例

代码1

代码1是根据赛中wa了的代码改的，实际可以再写的简洁点，如代码2所示

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long ll;
const int N=2e5+10,INF=0x3f3f3f3f;
int t,n,m,x,y,dp[N],col[N],rt,mx;
bool link[N];
vector<int>e[N];
void dfs(int u,int fa){
    dp[u]=0;
    link[u]=1;
    int son=0;
    bool haslinkson=0;
    for(auto &v:e[u]){
        if(v==fa)continue;
        dfs(v,u);
        dp[u]+=dp[v];
        son++;
        link[u]&=link[v];
        haslinkson|=link[v];
    }
    if(son>1)link[u]=0;
    if(!son)dp[u]=1;
    if(son>1 && haslinkson)dp[u]--;
}
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(int i=1;i<=n;++i){
            e[i].clear();
        }
        mx=0;
        for(int i=1;i<n;++i){
            scanf("%d%d",&x,&y);
            e[x].push_back(y);
            e[y].push_back(x);
            mx=max(mx,(int)e[x].size());
            mx=max(mx,(int)e[y].size());
        }
        if(n==1 || n==2){
            printf("%d\n",n-1);
            continue;
        }
        if(mx<=2){
            puts("1");
            continue;
        }
        int ans=n;
        for(int i=1;i<=n;++i){
            if(((int)e[i].size())==3){
                rt=i;
                dfs(rt,-1);
                ans=min(ans,dp[rt]);
                break;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}
/*
888
8
1 2
2 3
2 4
3 5
3 6
4 7
4 8
*/

代码2

#include <bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long ll;
const int N=2e5+10,INF=0x3f3f3f3f;
int t,n,m,x,y,dp[N],mx;
vector<int>e[N];
void dfs(int u,int fa){
    dp[u]=0;
    int link=0;
    for(auto &v:e[u]){
        if(v==fa)continue;
        dfs(v,u);
        dp[u]+=dp[v];
        if(!dp[v])link++;
    }
    dp[u]+=max(link-1,0);
}
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(int i=1;i<=n;++i){
            e[i].clear();
        }
        mx=0;
        for(int i=1;i<n;++i){
            scanf("%d%d",&x,&y);
            e[x].push_back(y);
            e[y].push_back(x);
            mx=max(mx,(int)e[x].size());
            mx=max(mx,(int)e[y].size());
        }
        if(n==1 || n==2){
            printf("%d\n",n-1);
            continue;
        }
        if(mx<=2){
            puts("1");
            continue;
        }
        for(int i=1;i<=n;++i){
            if(((int)e[i].size())>=3){
                dfs(i,-1);
                printf("%d\n",dp[i]);
                break;
            }
        }
    }
    return 0;
}
/*
888
8
1 2
2 3
2 4
3 5
3 6
4 7
4 8
*/