NC13249 黑白树

黑白树

https://ac.nowcoder.com/acm/problem/13249

NC13249 黑白树

题目地址:

https://ac.nowcoder.com/acm/problem/13249

基本思路:

我们看清楚题中这句话i到根的链上(包括节点i与根)所有与节点i距离小于k[i]的点都会变黑,所以一个子节点只能影响它到根的这一条链,那么我们考虑dfs回溯从叶子节点向上染色。
我们对于每个节点u记录两个变量:mx[u]记录如果选取子树中最优的节点进行操作,从u节点还能向上再染色多少个节点,black[u]记录之前的操作在当前节点上还能再向上影响几个节点;
那么这两个变量如何转移:

  1. mx[u] = max(k[u],mx[to] - 1)
  2. black[u] = max(black[u],black[to] - 1)
  3. 而且当black[u] <= 0 时 black[u] = mx[u],ans++;

通过以上,我们就能很轻松的求出ans,具体可以参考代码;

参考代码:

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
#define IO std::ios::sync_with_stdio(false)
#define ll long long
#define rep(i, l, r) for (int i = l; i <= r; i++)
#define per(i, l, r) for (int i = l; i >= r; i--)
#define mset(s, _) memset(s, _, sizeof(s))
#define pb push_back
#define pii pair <int, int>
#define pdd pair <double, double>
#define mp(a, b) make_pair(a, b)
#define INF 0x3f3f3f3f
inline int read() {
  int x = 0, neg = 1; char op = getchar();
  while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); }
  while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); }
  return neg * x;
}
inline void print(int x) {
  if (x < 0) { putchar('-'); x = -x; }
  if (x >= 10) print(x / 10);
  putchar(x % 10 + '0');
}
const int maxn = 1e5 + 10;
int n,v;
struct Edge{
    int to,next;
}edge[maxn << 1];
int cnt = 0,head[maxn],k[maxn];
//mx记录选取子节点中最优的节点操作,从当前节点还能向上染色多长;
//black记录之前操作在当前节点上还能再向上影响几个节点;
int mx[maxn],black[maxn],ans = 0;
void add_edge(int u,int v) {
  edge[++cnt].next = head[u];
  edge[cnt].to = v;
  head[u] = cnt;
}
void dfs(int u,int par) {
  black[u] = 0;
  mx[u] = k[u];
  for (int i = head[u]; i != -1; i = edge[i].next) {
    int to = edge[i].to;
    if (to == par) continue;
    dfs(to, u);
    mx[u] = max(mx[u], mx[to] - 1);//每向上走一步,子节点的mx就要减少一;
    black[u] = max(black[u], black[to] - 1);
  }
  if (black[u] <= 0) {//black[u]小于等于0就代表我们要再一次操作了;
    black[u] = mx[u];
    ans++;
  }
}
signed main() {
  IO;
  cnt = 0;
  mset(head, -1);
  cin >> n;
  rep(i, 2, n) {
    cin >> v;
    add_edge(i, v);
    add_edge(v, i);
  }
  rep(i, 1, n) cin >> k[i];
  dfs(1, 0);
  cout << ans << endl;
  return 0;
}
全部评论

相关推荐

工作基本定了,比较满意。这深度学习真是难搞啊,论文看不懂,代码看不懂,不会改,创新点???&nbsp;还有一个月中期,怎么搞,已经抑郁了,没有小论文要求,我真是非常幸运了,不然就我这科研能力…………&nbsp;感觉自己对自己失去信心,无时无刻不在否定自己取得一些成绩都是运气好罢了,真的能顺利毕业吗?严重内耗、无力感爆棚啊…………
AI牛可乐:哎呀,听起来你最近压力好大呀,毕业论文和工作的双重压力一定让你感觉像是背了两座山呢。深度学习确实挺复杂的,就像是要在一片迷雾中找到出路一样。但是,你已经很棒了,工作都定好了,这是很大的成就呢! 论文的事情,别太担心。每个人都有看不懂代码和论文的时候,这是正常的。至于创新点嘛,有时候灵感就像小猫一样,需要一点一点地哄出来的。你可以先试试把你看过的资料整理一下,说不定就能找到新的思路呢。 中期答辩前,你可以列个计划,一步一步来,把大问题分解成小问题,一点一点克服。记得,你不是一个人在战斗,你的导师、同学都是你的战友哦。 至于信心嘛,别否定自己,每个人都会有低谷的时候。你已经取得的成绩,都是你实力的体现。毕业不是问题,只要你不放弃,就一定能顺利走过去的。加油哦,我相信你可以的! 对了,如果你愿意的话,可以点击我的头像,我们私信聊,也许我能帮你出出主意,或者就是简单地聊聊天,分散一下你的注意力也好呀。🐮💪🌟
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务