首页 > 其他 > 详细

B. Alyona and a tree

时间:2018-08-20 00:57:29      阅读:228      评论:0      收藏:0      [点我收藏+]

二分+dfs序+前缀和+瞎几把yy

 

#include "a.h"

int n;
aLL a;
struct edge
{
  int to, nx, w;
} e[N << 1];
arr fi;
int ce = 1;
aLL sum, ans;
aLL vec, up, down, res;
ll tot = 1;
ll rs = 0;

inline void add(int u, int v, int w)
{
  e[++ce] = edge{v, fi[u], w};
  fi[u] = ce;
}

void dfs(int x)
{
  if (fi[x] == -1)
  {
    rs = res[x];
    rs += up[x], rs -= down[x];
    ans[x] = rs;
    return;
  }
  for (int i = fi[x]; i != -1; i = e[i].nx)
  {
    int v = e[i].to, w = e[i].w;
    sum[++tot] = sum[tot - 1] + w;
    int p = lower_bound(sum + 1, sum + 1 + tot, sum[tot] - a[v]) - sum;
    if (tot - 1 >= p)
      up[vec[tot - 1]]++, down[vec[p - 1]]++;
    vec[tot] = v;
    dfs(v);
    tot--;
    res[x] += rs;
    rs = 0;
  }
  rs = res[x];
  rs += up[x], rs -= down[x];
  ans[x] = rs;
}

int main()
{
  // file("test");
  while (~scanf("%d", &n))
  {
    me(fi, -1);
    For(i, 1, n) sdf(a[i]);
    For(i, 2, n)
    {
      int x, y;
      sdf(x), sdf(y);
      add(x, i, y);
    }
    vec[1] = 1;
    sum[1] = 0;
    dfs(1);
    For(i, 1, n)
        printf("%d ", ans[i]);
  }
  return 0;
}

 

B. Alyona and a tree

原文:https://www.cnblogs.com/planche/p/9503352.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!