[洛谷P3401] 洛谷树
洛谷题目连接:洛谷树
题目背景
萌哒的Created equal小仓鼠种了一棵洛谷树！
（题目背景是辣鸡小仓鼠乱写的QAQ）。
题目描述
树是一个无环、联通的无向图，由n个点和n-1条边构成。树上两个点之间的路径被定义为他们之间的唯一一条简单路径——显然这是一条最短路径。
现在引入一个概念——子路径。假设树上两个点p1和pn之间的路径是P=<p1,p2,p3,…,pn>，那么它的子路径被定义为某一条路径P’，满足P’=<pi,pi+1,pi+2,…,pj>，其中1<=i<=j<=n。显然，原路径是一条子路径，任意一个点也可以作为子路径。
我们给每条边赋予一个边权。萌萌哒的Sugar问小仓鼠：对于任意两个点u和v，你能快速求出，u到v的路径上所有子路径经过的边的边权的xor值的和是多少。具体地说就是，你把u到v的路径上所有子路径全部提出来，然后分别把每个子路径上经过的边的边权xor在一起，最后求出得到的所有xor值的和。
什么？你不知道xor？那就去百度啊！
这时候，fjzzq2002大爷冒了粗来：窝还要你滋磁修改某条边边权的操作！
小仓鼠那么辣鸡，当然不会做这道题啦。于是他就来向你求救！
输入输出格式
输入格式：
第一行两个正整数n和q，表示点的个数，查询和询问的总次数。
接下来n-1行，每行两个正整数u、v、w，表示u和v两个点之间有一条边权为w的边。
接下来q行，格式为1 u v或2 u v w。如果为1 u v操作，你需要输出u到v的路径上所有子路径经过的边的边权的xor值的和是多少；如果为2 u v w操作，你需要把u到v这条边的边权改为w，保证这条边存在。
输出格式：
对于每个1操作，输出答案。
输入输出样例
输入样例#1：
5 3
1 2 3
2 3 3
2 4 6
4 5 1
1 3 4
2 2 4 7
1 3 5
输出样例#1：
14
26
说明
本题时限1s，内存限制128M，因新评测机速度较为接近NOIP评测机速度，请注意常数问题带来的影响。
【数据范围】
No n= q= 备注
1 100 5 无
2 100 20 无
3 100 100 无
4 5000 1000 无
5 5000 2000 无
6 5000 3000 无
7 10000 10000 第i条边连接第i个点和第i+1个点，且没有2操作
8 10000 20000 第i条边连接第i个点和第i+1个点，且没有2操作
9 10000 10000 第i条边连接第i个点和第i+1个点
10 10000 20000 第i条边连接第i个点和第i+1个点
11 10000 10000 没有2操作
12 10000 20000 没有2操作
13 20000 20000 没有2操作
14 30000 30000 没有2操作
15 30000 10000 无
16 20000 20000 无
17 20000 20000 无
18 30000 20000 无
19 20000 30000 无
20 30000 30000 无
对于100%的数据，所有边权小于等于1023。
一句话题意: 给出一棵树,要求出一条路径上的所有子段的异或值的和.并且要求支持修改操作.
题解: 要求一条路径上的子段的异或值的和,如果直接用朴素算法的话,需要枚举这个子段的两个端点.那么这样复杂度就是\(O(n^2)\)的了.
我们先来考虑这样一种情况:假设边的权值只有0和1.
因为题目要求的是把所有子段的异或和加起来,所以先将每个点到根的异或和统计起来,用树剖维护到线段树中.因为0与0异或后仍然是0,1与1异或也是,而这样都不能对答案做出贡献,所以就不用考虑加入答案.只有一个0和一个1异或后的值为1,也就是说答案就是0与1可以形成的组合的数量,也就是从根节点开始的异或和为0的个数乘以1的个数.
那么对于那些权值不是0和1的也可以转化成二进制的形式,然后再按位处理,也就是说,对于一条路径的查询可以分成11次查询的和(1023\(\leq2^{10}\)),分别查询第0~第10位的答案之和.
然后对于修改操作,因为我们在线段树中维护的是异或和,所以需要将修改的那条边的子树中的值都要修改(对于二进制每一位的修改),因为只有0和1,所以直接将区间的信息翻转一下就可以了.
可能讲的有点不清楚,可以通过样例模拟一下,然后看下代码理解一下吧.
#include<bits/stdc++.h>
#define ll(x) (x<<1)
#define rr(x) (x<<1|1)
using namespace std;
typedef int _int;
#define int long long
const int N=30000+5;
int n, m, ecnt = 0, last[N], val[N], pre[N];
int size[N], son[N], fa[N], top[N], id[N], idx = 0, dep[N], vis[N];
struct edge{
int to, nex, w;
}e[N*2];
struct segment_tree{
int l, r, size, res1, tag;
segment_tree operator + (const segment_tree a) const{
segment_tree temp;
temp.size = size+a.size;
temp.res1 = res1+a.res1;
return temp;
}
}t[N*4][12];
int gi(){
int ans = 0, f = 1; char i = getchar();
while(i<'0' || i>'9'){ if(i == '-') f = -1; i = getchar(); }
while(i>='0' && i<='9') ans = ans*10+i-'0', i = getchar();
return ans * f;
}
void add(int x, int y, int z){
e[++ecnt].to = y;
e[ecnt].w = z;
e[ecnt].nex = last[x];
last[x] = ecnt;
}
void up(int x, int k){ t[x][k].res1 = t[ll(x)][k].res1+t[rr(x)][k].res1; }
void dfs1(int x, int f, int deep){
size[x] = 1, dep[x] = deep, fa[x] = f;
for(int to, i=last[x];i;i=e[i].nex){
to = e[i].to;
if(to == f) continue;
pre[to] = e[i].w^pre[x], val[to] = e[i].w;
dfs1(to, x, deep+1); size[x] += size[to];
if(size[to] > size[son[x]]) son[x] = to;
}
}
void dfs2(int x, int f, int tp){
top[x] = tp, id[x] = ++idx, vis[idx] = x;
if(son[x]) dfs2(son[x], x, tp);
for(int to, i=last[x];i;i=e[i].nex){
to = e[i].to;
if(to != f && to != son[x]) dfs2(to, x, to);
}
}
void build(int x, int l, int r, int k){
t[x][k].l = l, t[x][k].r = r, t[x][k].size = r-l+1;
if(l == r){
if((pre[vis[l]]>>k) & 1) t[x][k].res1 = 1;
return;
}
int mid = (l+r>>1);
build(ll(x), l, mid, k), build(rr(x), mid+1, r, k); up(x, k);
}
void pushdown(int x, int k){
if(t[x][k].tag == 0) return;
t[ll(x)][k].tag ^= 1, t[ll(x)][k].res1 = t[ll(x)][k].size-t[ll(x)][k].res1;
t[rr(x)][k].tag ^= 1, t[rr(x)][k].res1 = t[rr(x)][k].size-t[rr(x)][k].res1;
t[x][k].tag = 0;
}
void update(int x, int l, int r, int k){
if(l <= t[x][k].l && t[x][k].r <= r){
t[x][k].res1 = t[x][k].size-t[x][k].res1;
t[x][k].tag ^= 1;
return;
}
int mid = (t[x][k].l+t[x][k].r>>1); pushdown(x, k);
if(l <= mid) update(ll(x), l, r, k);
if(mid < r) update(rr(x), l, r, k); up(x, k);
}
segment_tree query(int x, int l, int r, int k){
if(l <= t[x][k].l && t[x][k].r <= r) return t[x][k];
segment_tree temp; temp.size = temp.res1 = 0;
if(r < t[x][k].l || t[x][k].r < l) return temp;
pushdown(x, k); temp.size = t[x][k].size;
return query(ll(x), l, r, k)+query(rr(x), l, r, k);
}
void modify(int x, int y, int v){
if(dep[x] < dep[y]) swap(x, y);
for(int i=0;i<=10;i++)
if(((v^val[x])>>i) & 1) update(1, id[x], id[x]+size[x]-1, i);
val[x] = v;
}
int ask(int xx, int yy){
int x, y, res = 0; segment_tree temp, check;
for(int i=0;i<=10;i++){
x = xx, y = yy; temp.size = temp.res1 = 0;
while(top[x] != top[y]){
if(dep[top[x]] < dep[top[y]]) swap(x, y);
temp = temp+(check=query(1, id[top[x]], id[x], i));
x = fa[top[x]];
}
if(id[x] > id[y]) swap(x, y);
temp = temp+(check=query(1, id[x], id[y], i));
res += (temp.size-temp.res1)*temp.res1*(1<<i);
}
return res;
}
_int main(){
int opt, x, y, z; n = gi(), m = gi();
for(int i=1;i<n;i++) x = gi(), y = gi(), z = gi(), add(x, y, z), add(y, x, z);
dfs1(1, -1, 1); dfs2(1, -1, 1);
for(int i=0;i<=10;i++) build(1, 1, n, i);
for(int i=1;i<=m;i++){
opt = gi(), x = gi(), y = gi();
if(opt == 1) cout << ask(x, y) << endl;
else z = gi(), modify(x, y, z);
}
return 0;
}
