题意:有一个经典DP问题是树的最小点集覆盖,那个问题是说:要你选出一个最小的点集,使得树上所有的点要么是点集里的点,要么和点集里的点相连。就是一个01DP.。。
现在这个题的意思是,树上的点要么在点集中,要么离点集的最小距离<=k,k给定。然后你就没办法DP了。
题解:考虑直径,把一个端点选做根,另一个端点就是深度最大的点,考虑这个点,想要把它覆盖了,并且同时覆盖面尽量广,那么就要选择他的k爹((爹的)^k-1爹),然后呢,你就需要把这个k爹周围k层的所有点染色,然后继续选择深度最大的点找爹,如果没有k爹了,就选根了,同时染色也就结束了。仔细想想还是挺对的。
为了表示一定是对的,我的代码中没有选直径端点做根,也没有选1做根,而是随机了一个根,以证明至少结果是完全正确的。
Code:
#include<bits/stdc++.h>
#define random(x,y) x+rand()%(y-x)
using namespace std;
const int maxn = 1e5+100;
int first[maxn],nxt[maxn*2],des[maxn*2],tot;
int fa[maxn],dep[maxn];
bool vis[maxn];
int rk[maxn];
int n,k,type,now;
inline void addEdge(int x,int y){
tot++;
des[tot] =y;
nxt[tot] = first[x];
first[x] =tot;
}
void input(){
scanf("%d%d%d",&n,&k,&type);
for (int i=1;i<n;i++){
rk[i]=i;
int u,v;
scanf("%d%d",&u,&v);
addEdge(u,v);addEdge(v,u);
}
rk[n]=n;
}
void dfs(int node,int father){
fa[node] =father;
dep[node] = dep[father]+1;
for (int t = first[node];t;t=nxt[t]){
int v = des[t];
if (v==father)continue;
dfs(v,node);
}
}
bool cmp(const int &x,const int &y){
return dep[x]>dep[y];
}
void paint(int node,int father,int step){
if (step>k)return ;
vis[node] = true;
for (int t = first[node];t;t=nxt[t]){
int v = des[t];
if (v==father)continue;
paint(v,node,step+1);
}
}
void solve(){
int root =random(1,n);
dfs(root,0);
sort(rk+1,rk+1+n,cmp);
now =1;
int ans =0;
while (now<=n){
// cout<<now<<endl;
int x = rk[now];
if (vis[x]){now++;continue;}
for (int i = 0;i<k&&x!=root;i++){
x = fa[x];
}
ans ++;
paint(x,0,0);
}
cout<<ans<<endl;
}
int main(){
input();
solve();
return 0;
}