对数据的处理
//dfs1 里边用到的数组
int fa[30010],siz[30010],son[30010],h[30010];
对dfs1过程处理之后得到了储存数据的数组
//
// main.cpp
// 树链剖分_bzoj1036
//
// Created by 陈冉飞 on 2019/7/27.
// Copyright © 2019 陈冉飞. All rights reserved.
//
#include <iostream>
#include <vector>
#define maxn 30000+50
using namespace std;
vector<int> g[maxn];
int fa[maxn],h[maxn],siz[maxn],son[maxn];
void dfs1(int u,int f){
//首先赋上父节点的值
fa[u] = f;
//然后再赋上深度
h[u] = h[f] + 1;
//size要赋上值
siz[u] = 1;
//重儿子也要初始化
son[u] = 0;
//定义一个索引,是u这个节点相邻的
int v;
for (int i = 0; i < g[u].size(); i++) {
//拿到与u这个节点相邻的所有节点的索引
v = g[u][i];
//首先要确认这个节点是不是父亲节点,防止刷新了父亲节点的值,不是的话再传入下一层的dfs1中遍历
if (v != f) {
dfs1(v, u);
//然后当从子节点中返回来之后,刷新当前层的值
siz[u] += siz[v];
//判断是否更新当前节点u的重儿子的值
if(siz[son[u]] < siz[v]){ //如果大于的话,就刷新当前的son的值
son[u] = v;
}
}
}
}
int main(int argc, const char * argv[]) {
int n,i,a,b;
scanf("%d",&n);
for (i = 1; i < n ; i++) {
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
}
dfs1(1, 0);
//检验输出
for(i = 0;i < 30000+50;i++){
if (g[i].size() != 0) {
cout<<"*****************"<<endl;
for (int j = 0; j < g[i].size(); j++) {
cout<<g[i][j]<<" ";
}
cout<<endl;
cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<endl;
}
}
return 0;
}
打印结果“:
//dfs2
所要用到的数组int val[maxn],top[maxn],pos[maxn],A[maxn],cnt = 0;
//
// main.cpp
// 树链剖分_bzoj1036
//
// Created by 陈冉飞 on 2019/7/27.
// Copyright © 2019 陈冉飞. All rights reserved.
//
#include <iostream>
#include <vector>
#define maxn 30000+50
using namespace std;
vector<int> g[maxn];
int fa[maxn],h[maxn],siz[maxn],son[maxn];
int val[maxn],top[maxn],pos[maxn],A[maxn],cnt = 0;
void dfs1(int u,int f){
//首先赋上父节点的值
fa[u] = f;
//然后再赋上深度
h[u] = h[f] + 1;
//size要赋上值
siz[u] = 1;
//重儿子也要初始化
son[u] = 0;
//定义一个索引,是u这个节点相邻的
int v;
for (int i = 0; i < g[u].size(); i++) {
//拿到与u这个节点相邻的所有节点的索引
v = g[u][i];
//首先要确认这个节点是不是父亲节点,防止刷新了父亲节点的值,不是的话再传入下一层的dfs1中遍历
if (v != f) {
dfs1(v, u);
//然后当从子节点中返回来之后,刷新当前层的值
siz[u] += siz[v];
//判断是否更新当前节点u的重儿子的值
if(siz[son[u]] < siz[v]){ //如果大于的话,就刷新当前的son的值
son[u] = v;
}
}
}
}
void dfs2(int u,int f,int k){ //k为当前的这个节点的所在链的链首
//更新当前节点的所在链的链首
top[u] = k;
//更新pos
pos[u] = ++cnt;
//将数据映射到A里边
A[cnt] = val[u];
if (son[u] != 0) { //如果重儿子存在,不为零,就延伸当前的链
dfs2(son[u], u, k);
}
//然后开始遍历所有的相邻的情况
int v; //定义一个索引
for (int i = 0; i < g[u].size(); i++) {
v = g[u][i];
//新创一条链就应该让新增节点不等于最大的上边已经创了链的节点
if (v != f && son[u] != v) {
dfs2(v, u, v);
}
}
}
int main(int argc, const char * argv[]) {
int n,i,a,b;
scanf("%d",&n);
for (i = 1; i < n ; i++) {
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
}
dfs1(1, 0);
//检验输出
// for(i = 0;i < 30000+50;i++){
// if (g[i].size() != 0) {
// cout<<"*****************"<<endl;
// for (int j = 0; j < g[i].size(); j++) {
// cout<<g[i][j]<<" ";
// }
// cout<<endl;
// cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<endl;
// }
// }
//首先拿到val的值
for(i = 1;i <= n;i++){
scanf("%d",&val[i]);
}
dfs2(1, 0, 1);
//检验输出
cout<<"一共的链数 "<<cnt<<endl;
for(i = 1;i <= cnt; i ++){
cout<<i<<" "<<A[i]<<endl;
}
for (i = 0; i < 30000+50; i++) {
if (g[i].size() != 0) {
cout<<"*****************"<<endl;
for (int j = 0; j < g[i].size(); j++) {
cout<<g[i][j]<<" ";
}
cout<<endl;
cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<" 所在的链的链首 "<<top[i]<<" 所在的序号数 "<<pos[i]<<endl;
}
}
return 0;
}
运行结果
建树
tree[maxn<<3]
//
// main.cpp
// 树链剖分_bzoj1036
//
// Created by 陈冉飞 on 2019/7/27.
// Copyright © 2019 陈冉飞. All rights reserved.
//
#include <iostream>
#include <vector>
#define maxn 30000+50
using namespace std;
vector<int> g[maxn];
//dfs1
int fa[maxn],h[maxn],siz[maxn],son[maxn];
//dfs2
int val[maxn],top[maxn],pos[maxn],A[maxn],cnt = 0;
//建树update
int tree[maxn<<3],ma[maxn<<3];
void dfs1(int u,int f){
//首先赋上父节点的值
fa[u] = f;
//然后再赋上深度
h[u] = h[f] + 1;
//size要赋上值
siz[u] = 1;
//重儿子也要初始化
son[u] = 0;
//定义一个索引,是u这个节点相邻的
int v;
for (int i = 0; i < g[u].size(); i++) {
//拿到与u这个节点相邻的所有节点的索引
v = g[u][i];
//首先要确认这个节点是不是父亲节点,防止刷新了父亲节点的值,不是的话再传入下一层的dfs1中遍历
if (v != f) {
dfs1(v, u);
//然后当从子节点中返回来之后,刷新当前层的值
siz[u] += siz[v];
//判断是否更新当前节点u的重儿子的值
if(siz[son[u]] < siz[v]){ //如果大于的话,就刷新当前的son的值
son[u] = v;
}
}
}
}
void dfs2(int u,int f,int k){ //k为当前的这个节点的所在链的链首
//更新当前节点的所在链的链首
top[u] = k;
//更新pos
pos[u] = ++cnt;
//将数据映射到A里边
A[cnt] = val[u];
if (son[u] != 0) { //如果重儿子存在,不为零,就延伸当前的链
dfs2(son[u], u, k);
}
//然后开始遍历所有的相邻的情况
int v; //定义一个索引
for (int i = 0; i < g[u].size(); i++) {
v = g[u][i];
//新创一条链就应该让新增节点不等于最大的上边已经创了链的节点
if (v != f && son[u] != v) {
dfs2(v, u, v);
}
}
}
void update(int k,int l,int r,int i,int v){
//k 为线段树中的建树num的依据 i 为往哪个方向建树的,v为值
if(l == r){
tree[k] = v;
return;
}
int mid = (l+r)>>1;
if (i <=mid) {
update(k<<1, l, mid, i, v);
}else if(i >= mid+1){
update(k<<1|1, mid+1, r, i, v);
}
//再pushup值
tree[k] = tree[k<<1]+tree[k<<1|1];
}
int main(int argc, const char * argv[]) {
int n,i,a,b;
scanf("%d",&n);
for (i = 1; i < n ; i++) {
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
}
dfs1(1, 0);
//检验输出
// for(i = 0;i < 30000+50;i++){
// if (g[i].size() != 0) {
// cout<<"*****************"<<endl;
// for (int j = 0; j < g[i].size(); j++) {
// cout<<g[i][j]<<" ";
// }
// cout<<endl;
// cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<endl;
// }
// }
//首先拿到val的值
for(i = 1;i <= n;i++){
scanf("%d",&val[i]);
}
dfs2(1, 0, 1);
//检验输出
cout<<"一共的链数 "<<cnt<<endl;
for(i = 1;i <= cnt; i ++){
cout<<i<<" "<<A[i]<<endl;
}
for (i = 0; i < 30000+50; i++) {
if (g[i].size() != 0) {
cout<<"*****************"<<endl;
for (int j = 0; j < g[i].size(); j++) {
cout<<g[i][j]<<" ";
}
cout<<endl;
cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<" 所在的链的链首 "<<top[i]<<" 所在的序号数 "<<pos[i]<<endl;
}
}
//然后开始建树update
for (i = 1; i <= n; i++) {
update(1, 1, n, i, val[i]);
}
//检验输出
for (i = 1; i <= n<<3 ; i++) {
if(tree[i] != 0){
cout<<i<<" "<<tree[i]<<endl;
}
}
return 0;
}
运行结果
查询树上的两点之间的权值之和、权值的最大值
把不同的链上的转换为一个链上的,再相加
//
// main.cpp
// 树链剖分_bzoj1036
//
// Created by 陈冉飞 on 2019/7/27.
// Copyright © 2019 陈冉飞. All rights reserved.
//
//#include <bits/stdc++.h>
#include <iostream>
#include <cmath>
#include <vector>
#define maxn 30050
using namespace std;
vector<int> g[maxn];
//dfs1
int fa[maxn],h[maxn],siz[maxn],son[maxn];
//dfs2
int val[maxn],top[maxn],pos[maxn],A[maxn],cnt = 0,n;
//建树update
int tree[maxn<<3],mx[maxn<<3]; //mx是用来储存最大值的,而tree是用用来储存和的
void dfs1(int u,int f){
//首先赋上父节点的值
fa[u] = f;
//然后再赋上深度
h[u] = h[f] + 1;
//size要赋上值
siz[u] = 1;
//重儿子也要初始化
son[u] = 0;
//定义一个索引,是u这个节点相邻的
int v;
for (int i = 0; i < g[u].size(); i++) {
//拿到与u这个节点相邻的所有节点的索引
v = g[u][i];
//首先要确认这个节点是不是父亲节点,防止刷新了父亲节点的值,不是的话再传入下一层的dfs1中遍历
if (v != f) {
dfs1(v, u);
//然后当从子节点中返回来之后,刷新当前层的值
siz[u] += siz[v];
//判断是否更新当前节点u的重儿子的值
if(siz[son[u]] < siz[v]){ //如果大于的话,就刷新当前的son的值
son[u] = v;
}
}
}
}
void dfs2(int u,int f,int k){ //k为当前的这个节点的所在链的链首
//更新当前节点的所在链的链首
top[u] = k;
//更新pos
pos[u] = ++cnt;
//将数据映射到A里边
A[cnt] = val[u];
if (son[u] != 0) { //如果重儿子存在,不为零,就延伸当前的链
dfs2(son[u], u, k);
}
//然后开始遍历所有的相邻的情况
int v; //定义一个索引
for (int i = 0; i < g[u].size(); i++) {
v = g[u][i];
//新创一条链就应该让新增节点不等于最大的上边已经创了链的节点
if (v != f && son[u] != v) {
dfs2(v, u, v);
}
}
}
void update(int k,int l,int r,int i,int v){
//k 为线段树中的建树num的依据 i 为往哪个方向建树的,v为值
if(l == r){
tree[k] = mx[k] = v;
return;
}
int mid = (l+r)>>1;
if (i <=mid) {
update(k<<1, l, mid, i, v);
}else if(i >= mid+1){
update(k<<1|1, mid+1, r, i, v);
}
//再pushup值
tree[k] = tree[k<<1]+tree[k<<1|1];
mx[k] = max(mx[k<<1],mx[k<<1|1]);
}
//先写在线段树 树上 的求和、 求最值
int psum(int k,int l,int r,int goal_l,int goal_r){
if (goal_l == l && goal_r == r) {
return tree[k];
}
int mid = (l+r)>>1;
if(goal_r <=mid){
return psum(k<<1, l, mid, goal_l, goal_r);
}else if(goal_l >= mid+1){
return psum(k<<1|1, mid+1, r, goal_l, goal_r);
}else{
// return 10;
return psum(k<<1, l, mid, goal_l, mid)+psum(k<<1|1, mid+1, r, mid+1, goal_r);
}
}
//线段树上的求最值 goal_l到goal_r
int pmax(int k,int l,int r,int goal_l,int goal_r){
if (goal_l == l &&goal_r == r) {
return mx[k];
}
int mid = (l+r)>>1;
if (goal_r<=mid) {
return pmax(k<<1, l, mid, goal_l, goal_r);
}else if(goal_l >= mid+1){
return pmax(k<<1|1, mid+1, r, goal_l, goal_r);
}else {
// return 5;
return max(pmax(k<<1, l, mid, goal_l, mid),pmax(k<<1|1, mid+1, r, mid+1, goal_r));
}
}
//按照题干求两个节点之间的路径的和、路径所经过的所有的节点的最大值
int tsum(int u,int v){
int total_sum = 0;
//u ,v 为始终的两个节点 这两个节点可能不在一条链上
while (top[u] != top[v]) {
//注意在一条链上求和的时候有顺序的要求,因为在将链往线段树上赋值的时候是有顺序的 所以深度小的应该在前面
if(h[top[u]]<h[top[v]]){
swap(u, v);
}
total_sum += psum(1, 1, n, pos[top[u]], pos[u]);
//然后节点往上爬
u = fa[top[u]];
}
//跳出来了之后意味着在一条链上了,所以此时可以进行一条链上的求和
if (h[u]<h[v]) {
swap(u, v);
}
total_sum+= psum(1, 1, n, pos[v], pos[u]);
return total_sum;
}
int tmax(int u,int v){
int total_max = -0x3f3f3f3f;
while (top[u] != top[v]) {
if(h[top[u]]<h[top[v]]){
swap(u, v);
}
total_max = max(total_max,pmax(1, 1, n, pos[top[u]], pos[u]));
//然后节点往上爬
u = fa[top[u]];
}
if(h[u]<h[v]){
swap(u, v);
}
total_max = max(total_max,pmax(1, 1, n, pos[v], pos[u]));
return total_max;
}
int main(int argc, const char * argv[]) {
int i,a,b;
scanf("%d",&n);
for (i = 1; i < n ; i++) {
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
}
//首先拿到val的值
for(i = 1;i <= n;i++){
scanf("%d",&val[i]);
}
dfs1(1, 0);
//检验输出
// for(i = 0;i < 30000+50;i++){
// if (g[i].size() != 0) {
// cout<<"*****************"<<endl;
// for (int j = 0; j < g[i].size(); j++) {
// cout<<g[i][j]<<" ";
// }
// cout<<endl;
// cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<endl;
// }
// }
dfs2(1, 0, 1);
//检验输出
// cout<<"一共的链数 "<<cnt<<endl;
// for(i = 1;i <= cnt; i ++){
// cout<<i<<" "<<A[i]<<endl;
// }
// for (i = 0; i < 30000+50; i++) {
// if (g[i].size() != 0) {
// cout<<"*****************"<<endl;
// for (int j = 0; j < g[i].size(); j++) {
// cout<<g[i][j]<<" ";
// }
// cout<<endl;
// cout<<"父节点 "<<fa[i]<<" 深度 "<<h[i]<<" size "<<siz[i]<<" 重儿子 "<<son[i]<<" 所在的链的链首 "<<top[i]<<" 所在的序号数 "<<pos[i]<<endl;
// }
// }
//然后开始建树update
for (i = 1; i <= n; i++) {
update(1, 1, n, i, A[i]);
}
//检验输出
// for (i = 1; i <= n<<3 ; i++) {
// if(tree[i] != 0){
// cout<<i<<" "<<tree[i]<<endl;
// }
// }
//开始输入所有请求
int qu_total;
scanf("%d",&qu_total);
char s[10];
int c,d;
while (qu_total--) {
scanf("%s",s);
if (s[1] == 'M') { //求max
scanf("%d%d",&c,&d);
printf("%d\n",tmax(c, d));
// cout<<tmax(c,d)<<endl;
}else if(s[1] == 'S'){ //求和
scanf("%d%d",&c,&d);
printf("%d\n",tsum(c, d));
// cout<<tsum(c, d)<<endl;
}else if(s[1] == 'H'){ //改变某个值
//此时要用到建树时的update 来刷新权值,还要利用pos来找到他的位置
scanf("%d%d",&c,&d);
update(1, 1, n, pos[c], d);
}
}
return 0;
}