题目链接 Alyona and a tree

比较考验我思维的一道好题。

首先,做一遍DFS预处理出$t[i][j]$和$d[i][j]$。$t[i][j]$表示从第$i$个节点到离他第$2^{j}$近的祖先,$d[i][j]$表示从$i$开始到$t[i][j]$的路径上的路径权值总和。

在第一次DFS的同时,对节点$x$进行定位(结果为$dist(x, y)<=a(y)$)的离$x$最远的$x$的某个祖先,然后进行$O(1)$的差分。

第一次DFS完成后,做第二次DFS统计答案(统计差分后的结果)

时间复杂度$O(NlogN)$

#include <bits/stdc++.h>

using namespace std;

#define REP(i, n)		for(int i(0); i <  (n); ++i)
#define rep(i, a, b) for(int i(a); i <= (b); ++i)
#define dec(i, a, b) for(int i(a); i >= (b); --i)
#define LL long long
#define sz(x) (int)x.size() const int N = 200000 + 10;
const int A = 30 + 1; vector <int> v[N], c[N];
LL a[N], deep[N];
LL x, y;
int n, cnt;
LL t[N][A], d[N][A];
LL g[N], value[N];
LL s[N];
LL ans[N]; void dfs(int x, int fa){ if (g[x]){
t[x][0] = g[x];
d[x][0] = value[x];
for (int i = 0; t[t[x][i]][i]; ++i){
t[x][i + 1] = t[t[x][i]][i];
d[x][i + 1] = d[t[x][i]][i] + d[x][i];
}
int now = x, noww = 0;
bool flag = false;
dec(i, 20, 0){
if (t[now][i] && d[now][i] + noww <= a[x]){
noww += d[now][i];
now = t[now][i];
flag = true;
}
}
if (flag){
--s[g[now]]; ++s[g[x]];
}
} REP(i, sz(v[x])){
int u = v[x][i];
deep[u] = deep[x] + 1;
dfs(u, x);
}
} void dfs2(int x){
ans[x] += s[x];
REP(i, sz(v[x])){
dfs2(v[x][i]);
ans[x] += ans[v[x][i]];
}
} int main(){ scanf("%d", &n);
rep(i, 1, n) scanf("%lld", a + i);
rep(i, 2, n){
scanf("%lld%lld", &x, &y);
g[i] = x; value[i] = y;
v[x].push_back(i), c[x].push_back(y);
} memset(s, 0, sizeof s);
cnt = 0;
deep[1] = 0;
dfs(1, 0);
memset(ans, 0, sizeof ans);
dfs2(1);
rep(i, 1, n - 1) printf("%lld ", ans[i]);
printf("%lld\n", ans[n]);
return 0; }
												

最新文章

  1. PHPStorm 调式JS /同时调式PHP和jS
  2. C/C++中调用python文件
  3. Bluetooth数据包捕获
  4. Java基础——序列化
  5. 服务器编程入门(1)TCP/IP协议族
  6. PHP关于数组缓存JSON、serialize、var_export的说明
  7. 写一个程序,统计自己C语言共写了多少行代码,Github基本操作
  8. [Web开发(1)] MyEclipse/Eclipse 使用Tomcat部署Web/Maven项目经典错误
  9. PHP网站常见安全漏洞,及相应防范措施总结
  10. R语言︱H2o深度学习的一些R语言实践——H2o包
  11. openstack中的环境准备
  12. 四则运算----C++版
  13. vue_VueRouter 路由_路由器管理n个路由_并向路由组件传递数据_新标签路由_编程式路由导航
  14. subversion实用命令整理
  15. CentOS7(64)环境使用rpm命令安装gcc
  16. swoole深入学习 3. upd Server和udp Client
  17. 【java】类的初识
  18. WCF中的AddressHeader作用
  19. 接口文档神器Swagger(上篇)
  20. jenkins启动脚本

热门文章

  1. L1-039 古风排版 (20 分)
  2. 洛谷P4231 三步必杀
  3. Python数据结构之列表、元组及字典
  4. WINDOWS批量替换不同文件夹下的相同文件
  5. Composer 下载安装类库
  6. Excel的数据批量替换
  7. 错误:支持“EFDbContext”上下文的模型已在数据库创建后发生更改。请考虑使用 Code First 迁移更新数据库(http://go.microsoft.com/fwlink/?LinkId
  8. AngularJs 特性 之 MVC
  9. css对html中表格单元格td文本过长的处理
  10. bzoj4455【ZJOI2016】小星星