题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=4034

题意:中文题面

思路:树链剖分入门题。 剖分后就是一个简单的区间更新和区间求和问题。用线段树去维护一下。 由于有一个操作是关于子树的,可以用DFS序来求,但是由于剖分后的序列都是连续的,所以只需要记录下返回当前根时前一个点的位置即可进行子树操作。

#define _CRT_SECURE_NO_DEPRECATE
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<stdio.h>
#include<queue>
#include<vector>
#include<stack>
#include<map>
#include<set>
#include<time.h>
#include<cmath>
#include<sstream>
#include<assert.h>
using namespace std;
#define L(x) x<<1
#define R(x) x<<1|1
typedef long long int LL;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3fLL;
const int MAXN = + ;
int val[MAXN], head[MAXN], tot, cnt;
struct Edge{
int to,next;
Edge(int _to = , int _next = ) :to(_to), next(_next){};
}Edges[MAXN * ];
void add(int u, int v){
Edges[tot].to = v;
Edges[tot].next = head[u];
head[u] = tot++;
}
int id[MAXN], endid[MAXN], son[MAXN], deep[MAXN], size[MAXN], fa[MAXN], reid[MAXN], top[MAXN];
void Init(){
tot = ; cnt = ;
memset(head, -, sizeof(head));
memset(son, -, sizeof(son));
}
void DFS1(int u, int p,int dep){
fa[u] = p; size[u] = ; deep[u] = dep;
for (int i = head[u]; i != -; i = Edges[i].next){
if (Edges[i].to != p){
DFS1(Edges[i].to, u,dep+);
size[u] += size[Edges[i].to];
if (son[u] == - || size[Edges[i].to] > size[son[u]]){
son[u] = Edges[i].to;
}
}
}
}
void DFS2(int u, int tp){
id[u] = ++cnt; reid[id[u]] = u; top[u] = tp;
if (son[u] == -){
endid[u] = cnt;
return;
}
DFS2(son[u], tp);
for (int i = head[u]; i != -; i = Edges[i].next){
if (son[u] != Edges[i].to&&Edges[i].to != fa[u]){
DFS2(Edges[i].to, Edges[i].to);
}
}
endid[u] = cnt;
}
struct Node{
int st, ed;
LL sum, lazy;
}Seg[MAXN * ];
void Build(int l, int r, int k){
Seg[k].st = l; Seg[k].ed = r; Seg[k].lazy = ;
if (l == r){
Seg[k].sum = val[reid[l]];
return;
}
int mid = (l + r) / ;
Build(l, mid, L(k)); Build(mid + , r, R(k));
Seg[k].sum = Seg[L(k)].sum + Seg[R(k)].sum;
}
void pushUp(int k){
Seg[k].sum = Seg[L(k)].sum + Seg[R(k)].sum;
}
void pushDown(int k){
if (Seg[k].lazy){
Seg[L(k)].sum += 1LL*Seg[k].lazy*(Seg[L(k)].ed - Seg[L(k)].st + );
Seg[L(k)].lazy += Seg[k].lazy;
Seg[R(k)].sum += 1LL*Seg[k].lazy*(Seg[R(k)].ed - Seg[R(k)].st + );
Seg[R(k)].lazy += Seg[k].lazy;
Seg[k].lazy = ;
}
}
void Add(int l, int r, int k,int val){
if (Seg[k].st == l&&Seg[k].ed == r){
Seg[k].lazy += val;
Seg[k].sum += 1LL * val * (r - l + );
return;
}
pushDown(k);
if (r <= Seg[L(k)].ed){
Add(l, r, L(k),val);
}
else if (l >= Seg[R(k)].st){
Add(l, r, R(k),val);
}
else{
Add(l, Seg[L(k)].ed, L(k), val);
Add(Seg[R(k)].st, r, R(k), val);
}
pushUp(k);
}
LL Query(int l, int r, int k){
if (Seg[k].st == l&&Seg[k].ed == r){
return Seg[k].sum;
}
pushDown(k);
LL sum = ;
if (r <= Seg[L(k)].ed){
sum=Query(l, r, L(k));
}
else if (l >= Seg[R(k)].st){
sum=Query(l, r, R(k));
}
else{
sum=Query(l, Seg[L(k)].ed, L(k)) + Query(Seg[R(k)].st, r, R(k));
}
pushUp(k);
return sum;
}
LL Query(int x){
LL ans = ;
while (top[x]!=){
ans += Query(id[top[x]], id[x],);
x = fa[top[x]];
}
ans += Query(,id[x], );
return ans;
}
int main(){
//#ifdef kirito
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
//#endif
// int start = clock();
int n, m;
while (~scanf("%d%d",&n,&m)){
Init();
for (int i = ; i <= n; i++){
scanf("%d", &val[i]);
}
for (int i = ; i < n; i++){
int u, v;
scanf("%d%d", &u, &v);
add(u, v); add(v, u);
}
DFS1(, , ); DFS2(, );
Build(, n, );
while (m--){
int ope, x, a;
scanf("%d", &ope);
switch (ope)
{
case :scanf("%d%d", &x, &a); Add(id[x],id[x] , , a); break;
case :scanf("%d%d", &x, &a); Add(id[x], endid[x], , a); break;
default: scanf("%d", &x); printf("%lld\n", Query(x)); break;
}
}
}
//#ifdef LOCAL_TIME
// cout << "[Finished in " << clock() - start << " ms]" << endl;
//#endif
return ;
}

最新文章

  1. 火狐浏览器与Chorme的兼容性小问题
  2. Linux互斥和同步应用程序(四):posix互斥信号和同步
  3. oracle动态视图v$,v_$,gv$,gv_$与x$之间的关系
  4. USACO-palsquare 遇到的一个坑
  5. js功能代码大全
  6. SpringMVC的映射器、适配器、解析器
  7. [Ccodeforces 736C] Ostap and Tree - 树形DP
  8. xls到xml
  9. (predicted == labels).sum().item()作用
  10. 利用Navicate把SQLServer转MYSQL的方法(连数据)
  11. (后端)excel设置日期格式的步骤
  12. CF1114D Flood Fill(DP)
  13. 上传文件服务器与web内容服务分离
  14. 大数据学习笔记02-HDFS-常用命令
  15. ArrayList详细
  16. bzoj5044: 岛屿生成
  17. Linux-文件查找命令find
  18. jzoj3519
  19. ElasticSearch的概念解析
  20. Linux input子系统学习总结(三)----Input设备驱动

热门文章

  1. 【leetcode】816. Ambiguous Coordinates
  2. Word文档粘贴到DEDECMS
  3. 小陈现有2个任务A,B要完成,每个任务分别有若干步骤如下 一道网上没啥题解的难题(至少我是这么觉得的)
  4. 基于ELK Stack7.1.0构建多用户安全认证日志系统
  5. CodeForces - Path Queries (并查集+离线查询)
  6. Netty精进01
  7. 死锁(Deadlock)
  8. Jenkins使用三:管理slave节点(配置SSH公钥和私钥)
  9. Python变量和字符串详解
  10. RSA - 原理、特点(加解密及签名验签)及公钥和私钥的生成