如題,已知一棵包含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為根節點的子...