【每日一题】3月9日Tree Decoration
Tree Decoration
https://ac.nowcoder.com/acm/problem/24623
题意
约翰在装饰他家的一颗春分树(不是知道是啥),树上有 个节点,第一个节点是根,其余节点都有唯一的父节点,第
个节点的父节点是
,由于根节点没有父节点,记作-1
约翰可以往每个节点上挂装饰物,但是费用是变化的,在第个节点上挂一个装饰物需要花费
元钱,要求以
为根的子树上必须有
个装饰物,然后问在哪些节点上挂装饰物,才能满足要求,并且使得装饰费用最少
直接从最底下开始dfs就可以了
最底下的一定是要挂的 因为最底下的就没有子树了 因此我们从最底下开始
够了的话就不用选
不够的话就下去更新数量
code
#include<bits/stdc++.h>
using namespace std;
typedef long long ll; typedef unsigned long long ull; typedef long double ld;
const ll MOD = 1e9 + 7;
const int N = 1e6 + 7;
const int INF = 0x3f3f3f3f;
ll c[N] , t[N] , ans , sz[N];
ll n;
vector<int> edge[N];
void dfs(int u){
for(auto& v : edge[u]){
dfs(v);
sz[u] += sz[v];
t[u] = min(t[u] , t[v]);
}
if(sz[u] < c[u]){
ans += (c[u] - sz[u]) * t[u];
sz[u] = c[u];
}
}
void solve(){
scanf("%lld" , &n);
ll fa;
for(int i = 2 ; i <= n ; ++i){
scanf("%lld" , &fa);
scanf("%lld%lld" , &c[i] , &t[i]);
if(fa == -1) continue;
edge[fa].push_back(i);
}
dfs(1);
printf("%lld\n" , ans);
}
int main(void){
solve();
return 0;
} 每日一题 文章被收录于专栏
写每日一题呀
查看6道真题和解析