P3384 模板 樹鏈剖分

2021-07-26 23:56:14 字數 3354 閱讀 3414

傳送門

題目描述

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

5 5 2 24

7 3 7 8 0

1 2

1 5

3 1

4 1

3 4 2

3 2 2

4 5

1 5 1 3

2 1 3

輸出樣例#1:

2 21

說明時空限制:1s,128m

資料規模:

對於30%的資料:n<=10,m<=10

對於70%的資料:n<=1000,m<=1000

對於100%的資料:n<=100000,m<=100000

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

樣例說明:

樹的結構如下:

各個操作如下:

故輸出應依次為2、21(重要的事情說三遍:記得取模)

看到題目就知道,這題是裸的樹鏈剖分。

樹鏈剖分的大體意思是將一棵樹分成許多重鏈,然後按照各個點在重鏈中的位置進行標號,再用線段樹將這些重鏈hash成線型結構進行儲存和運算,詳細解釋戳這裡

code:

#include

#include

const int maxn=1e5+10;

const int inf=1e9;

struct edge

e[2*maxn];

struct tree

t[1000001];

ints[2

*maxn],head[2

*maxn],deep[2

*maxn];

int top[2

*maxn],son[2

*maxn],size[2

*maxn],f[2

*maxn],num[2

*maxn],pos[2

*maxn];

int n,m,root,p,e_num=-1,w,x,y,z,tot;

inline int max(int a,int b)

inline int min(int a,int b)

inline void swap(int &a,int &b)

inline void add(int from,int to)

void dfs(int now,int depth,int fa)

}void dfs2(int now,int high)

inline void pushdown(int l,int r,int now)

int mid=(l+r)>>1,s1=now<<1,s2=now<<1|1;

t[s1].plus=(t[s1].plus+t[now].plus)%p;

t[s2].plus=(t[s2].plus+t[now].plus)%p;

t[s1].num=(t[s1].num+(mid-l+1)*t[now].plus)%p;

t[s2].num=(t[s2].num+(r-mid)*t[now].plus)%p;

t[now].plus=0;

}inline void update(int now)

void build(int l,int r,int now)

int mid=(l+r)>>1;

build(l,mid,now<<1);

build(mid+1,r,now<<1|1);

update(now);

}void add(int l,int r,int l,int r,int now,int z)

pushdown(l,r,now);

int mid=(l+r)>>1;

if(l<=mid) add(l,r,l,mid,now<<1,z);

if(r>mid) add(l,r,mid+1,r,now<<1|1,z);

update(now);

}int ask(int l,int r,int l,int r,int now)

void addpath(int

x,int

y,int num)//1

if(deep[top[x]]y]]) swap(x,y);

l=min(pos[x],pos[top[x]]);

r=max(pos[x],pos[top[x]]);

add(l,r,1,n,1,z);

addpath(f[top[x]],y,z);

}int askpath(int

x,int

y)//2

if(deep[top[x]]y]]) swap(x,y);

l=min(pos[x],pos[top[x]]);

r=max(pos[x],pos[top[x]]);

int ans=ask(l,r,1,n,1);

return (ans+askpath(f[top[x]],y))%p;

}inline void addnum(int

x,int z)//3

inline int asknum(int

x)//4

int main()

return

0;}

P3384 模板 樹鏈剖分

p3384 模板 樹鏈剖分 樹鏈剖分是把一棵樹劃分成幾條鏈,這幾條鏈又能組成陣列,然後把陣列建成線段樹,繼而相當於在樹樹上操作。include include include include include using namespace std const int maxn 2e5 10 int ...

P3384 模板 樹鏈剖分

題目描述 如題,已知一棵包含n個結點的樹 連通且無環 每個節點上包含乙個數值,需要支援以下操作 操作1 格式 1 x y z 表示將樹從x到y結點最短路徑上所有節點的值都加上z 操作2 格式 2 x y 表示求樹從x到y結點最短路徑上所有節點的值之和 操作3 格式 3 x z 表示將以x為根節點的子...

P3384 模板 樹鏈剖分

題目大意 如題,已知一棵包含n個結點的樹 連通且無環 每個節點上包含乙個數值,需要支援以下操作 操作1 格式 1 x y z 表示將樹從x到y結點最短路徑上所有節點的值都加上z 操作2 格式 2 x y 表示求樹從x到y結點最短路徑上所有節點的值之和 操作3 格式 3 x z 表示將以x為根節點的子...