题意:
1378 夹克老爷的愤怒
1.0 秒 131,072.0 KB 80 分 5级题
夹克老爷逢三抽一之后,由于采用了新师爷的策略,乡民们叫苦不堪,开始组织起来暴力抗租。
夹克老爷很愤怒,他决定派家丁常驻村中进行镇压。
诺德县 有N个村庄,编号0 至 N-1,这些村庄之间用N - 1条道路连接起来。
家丁都是经过系统训练的暴力机器,每名家丁可以被派驻在一个村庄,并镇压当前村庄以及距离该村庄不超过K段道路的村庄。
夹克老爷一贯奉行最小成本最大利润的原则,请问要实现对全部村庄的武力控制,夹克老爷需要派出最少多少名家丁?
输入
第1行:2个数N, K中间用空格分隔(1<= N <= 100000, 0 <= K <= N)。
之后N-1行:每行2个数S, E中间用空格分隔,表示编号为S的村庄同编号为E的村庄之间有道路相连。(0 <= S, E < N)。
输出
输出一个数说明要实现对全部村庄的武力控制,夹克老爷需要派出最少多少名家丁?
输入样例
4 1
0 1
0 2
0 3
输出样例
1
思路:
贪心思想+树形dp:
从叶子节点开始,首先肯定在从叶子节点到根的第k个点的位置放置一个家丁。
而且采取能不放就不放的原则。
每到一个点就检查当前这个点能不能覆盖以当前节点为根的所有子树的点
如果可以覆盖,那么这个位置就可以不放。
否则,在当前节点放置一个家丁,返回的就是-k,说明还能控制上面k个节点
mn和mx变量含义:
mx变量表示的是子树中离当前的点最远的距离
mn变量表示的是子树中离当前的点最近的距离,正值表示的是向下延伸的距离,负值表示的是向上延伸的距离。
代码实现:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e6 + 5;
const int mod = 32767;
int n,k;
vector<int>G[maxn];
int ans = 0;
int dfs(int u,int fa){
if(G[u].size() == 1&&G[u][0] == fa) return 1;
int mx = - k - 1; int mn = k + 1;
for(int j = 0;j < G[u].size();j++){
if(G[u][j] != fa){
int d = dfs(G[u][j],u);
mn = min(d,mn);
mx = max(d,mx);
}
}
if(mx >= k){
ans++;
return -k;
}
else{
return (mn + mx < 0? mn : mx) + 1;
}
}
int main(){
scanf("%d%d",&n,&k);
int u,v;
for(int i = 1;i < n;i++){
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
if(k == 0){
printf("%d\n",n);
}
else{
if(dfs(0,-1) > 0){
ans++;
}
printf("%d\n",ans);
}
return 0;
}