模板 樹鏈剖分

2021-09-28 21:49:50 字數 3142 閱讀 7005

題目描述

如題,已知一棵包含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取模)

輸入輸出樣例

輸入5 5 2 24

7 3 7 8 0

1 21 5

3 14 1

3 4 2

3 2 2

4 51 5 1 3

2 1 3輸出2

21說明/提示

時空限制:1s,128m

( 其實,純隨機生成的樹lca+暴力是能過的,可是,你覺得可能是純隨機的麼233 )

#include

#include

#define si 100010

#define re register int

#define ls(p) (p<<1)

#define rs(p) (p<<1|1)

using

namespace std;

struct edge e[si<<1]

;int n,m,r,p,tot,head[si]

,w[si]

,wt[si]

,a[si<<2]

,tag[si<<2]

;int cnt,son[si]

,id[si]

,fa[si]

,d[si]

,sz[si]

,top[si]

;inline

intread()

while

(ch>=

'0'&&ch<=

'9')

return x*cf;

}inline

void

add(

int x,

int y)

inline

void

dfs1

(int x,

int f,

int deep)

}inline

void

dfs2

(int x,

int topf)

}inline

void

pushup

(int p)

inline

void

pushdown

(int p,

int x)

inline

void

build

(int p,

int l,

int r)

int mid=

(l+r)

>>1;

build(ls

(p),l,mid)

;build(rs

(p),mid+

1,r)

;pushup

(p);

return;}

inline

void

update

(int p,

int l,

int r,

int nl,

int nr,

int k)

pushdown

(p,r-l+1)

;int mid=

(l+r)

>>1;

if(nl<=mid)

update(ls

(p),l,mid,nl,nr,k);if

(nr>mid)

update(rs

(p),mid+

1,r,nl,nr,k)

;pushup

(p);

return;}

inline

intquery

(int p,

int l,

int r,

int nl,

int nr)

inline

intqrange

(int x,

int y)

if(d[x]

>d[y]

)swap

(x,y)

;return

(ans+

query(1

,1,n,id[x]

,id[y]))

%p;}

inline

void

updrange

(int x,

int y,

int k)

if(d[x]

>d[y]

)swap

(x,y)

;update(1

,1,n,id[x]

,id[y]

,k%p);}

intmain()

dfs1

(r,0,1

),dfs2

(r,r)

;build(1

,1,n);

while

(m--

)else

if(q==2)

else

if(q==3)

else

printf

("%d\n"

,query(1

,1,n,id[x]

,id[x]

+sz[x]-1

));}

return0;

}

樹鏈剖分 模板

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...