樹鏈剖分模板

2022-05-21 18:24:10 字數 2984 閱讀 6359

如題,已知一棵包含n個結點的樹(連通且無環),每個節點上包含乙個數值,需要支援以下操作:

操作1: 格式: 1 x y z 表示將樹從x到y結點最短路徑上所有節點的值都加上z

操作2: 格式: 2 x y 表示求樹從x到y結點最短路徑上所有節點的值之和

操作3: 格式: 3 x z 表示將以x為根節點的子樹內所有節點值都加上z

操作4: 格式: 4 x 表示求以x為根節點的子樹內所有節點值之和

輸入格式:

第一行包含4個正整數n、m、r、p,分別表示樹的結點個數、操作個數、根節點序號和取模數(即所有的輸出結果均對此取模)。

接下來一行包含n個非負整數,分別依次表示各個節點上初始的數值。

接下來n-1行每行包含兩個整數x、y,表示點x和點y之間連有一條邊(保證無環且連通)

接下來m行每行包含若干個正整數,每行表示乙個操作,格式如下:

操作1: 1 x y z

操作2: 2 x y

操作3: 3 x z

操作4: 4 x

輸出格式:

輸出包含若干行,分別依次表示每個操作2或操作4所得的結果(對p取模)

**:(感覺自己寫的有點長```>_<```······)

1 #include2

using

namespace

std;

3const

int max = 100005

;4 typedef struct

node

9int mid()

10 node(): rson(null), lson(null), l(0), r(0), val(0), tag(0

){}11 }node, *pointer;

1213

intmod;

1415

class

segment_tree

1625

26void

pushdown(pointer rt)

2736}37

38void build(pointer rt,int l, int

r)39

50 rt->rson = new

node();

51 rt->lson = new

node();

52 build(rt->lson, l, rt->mid());

53 build(rt->rson, rt->mid(), r);

54pushup(rt); 55}

5657

void updata(pointer rt, int l, int r, int

k)58

64else

6571}72

73long

long query(pointer rt, int l, int

r)7484}

85};

8687

class

tree_chain_division

88107

return

;108

}109

110void second_dfs(int u, int

anc)

111121

return

;122

}123

124public

:125

inta[max];

126127

void

init()

128131

132void add_edge(int

from, int

to)133

138139

void chain_division(int root, int

n)140

148149

int get_lca(int x, int

y)150

158return deep[x] < deep[y] ?x : y;

159}

160161

void chai_add(int x, int y, int w) //

x y z 表示將樹從x到y結點最短路徑上所有節點的值都加上w

162171

if(deep[x] 172 tree.updata(tree.rt, tid[y], tid[x] + 1

, w);

173return

;174

}175

176long

long chai_query(int x, int y) //

x y 表示求樹從x到y結點最短路徑上所有節點的值之和

177187

if(deep[x] 188 ans = (ans + tree.query(tree.rt, tid[y], tid[x] + 1)) %mod;

189return

ans;

190}

191192

void tree_add(int x, int w) //

x w 表示將以x為根節點的子樹內所有節點值都加上w

193197

198long

long tree_query(int x) //

x 表示求以x為根節點的子樹內所有節點值之和

199202

};203

204 inline int

qread()

205211

212tree_chain_division division;

213int

main()

214226

division.chain_division(root, n);

227for(int i = 1; i <= m; ++i)

228237

}238

return0;

239 }

樹鏈剖分 模板

class match node a n struct no no aa n 4 void init void addpage int x,int y void dfs int s,int faa,int h 根節點,父節點和深度的 if max 0 son s sign void dfs2 int...

模板 樹鏈剖分

define maxn 50010 define l u u 1 define r u u 1 1 寫在類裡面爆棧 int n,m,q int tim 時間戳 int num maxn 樹上每個節點的初始值 int siz maxn siz u 表示以u為根的子樹的節點數 int top maxn ...

樹鏈剖分模板

點權模板 1 m a b c將節點a到節點b路徑上所有點都染成顏色c 2 q a b詢問節點a到節點b路徑上的顏色段數量 連續相同顏色被認為是同一段 如 112221 由3段組成 11 222 和 1 const int n 100100 struct edge g n 2 int cnt,head...