P3384 樹鏈剖分模板 點權

2021-10-01 07:43:18 字數 2790 閱讀 5782

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

大概就是讓你維護一棵樹,要支援修改兩個點之間的值同時+val,支援修改以乙個點為根的子樹的所有點同時+val,支援查詢一條鏈的權值和,和乙個子樹的權值和。

用線段樹來維護樹剖之後的鏈。

主要記錄一下自己這題為啥wa了10+發

1.沒有取模

2.線段樹建樹的時候使用的陣列應該是fp陣列,即記錄這個位置是哪個數字的陣列。

3.在修改、查詢一條鏈和的時候,應該比較他們的爸爸的深度,而不是比較他們自己的深度,因為兩個點不在一條鏈上,若比較自己深度的話,可能會修改了不應該修改的權值。

#include #define ll long long

#define sc scanf

#define pr printf

using namespace std;

const int maxn = 1e5 + 5;

int n, m, root; ll mod;

struct edge

e[maxn * 2];

int head[maxn], tot;

int fa[maxn], son[maxn], deep[maxn], sz[maxn];

int p[maxn], fp[maxn], top[maxn], pos;

void init()

void add(int in, int to)

; head[in] = tot++;

}void dfs1(int u, int f, int dep)

}void dfs2(int u, int sp)

}//線段樹

#define lson left,mid,k<<1

#define rson mid+1,right,k<<1|1

#define imid int mid=(left+right)/2;

struct node

que[maxn * 4];

ll aa[maxn];

void up(int k)

void down(int k)

}void build(int left, int right, int k)

imid;

build(lson);

build(rson);

up(k);

};void update(int left, int right, int k, int ql, int qr, ll val)

down(k);

imid;

update(lson, ql, qr, val);

update(rson, ql, qr, val);

up(k);

}ll query(int left, int right, int k, int ql, int qr)

//樹剖

void change(int u, int v, int val)

update(1, n, 1, p[f1], p[u], val);

u = fa[f1];

f1 = top[u];

} if (deep[u] > deep[v])

swap(u, v);

update(1, n, 1, p[u], p[v], val);

}ll changes(int u, int v)

ans = (ans + query(1, n, 1, p[f1], p[u])) % mod;

u = fa[f1];

f1 = top[u];

} if (deep[u] > deep[v])

swap(u, v);

ans = (ans + query(1, n, 1, p[u], p[v])) % mod;

return ans;

}int main()

int a, b;

init();

for (int i = 1; i < n; i++)

dfs1(root, 0, 1);

dfs2(root, root);

build(1, n, 1);

int op; ll val;

while (m--) }

}/*5 5 1 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

*/

P3384 模板 樹鏈剖分

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

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為根節點的子...