Tree Xor（2021牛客暑期多校训练营4）【线段树】

Tree Xor（2021牛客暑期多校训练营4）【线段树】

传送门

题意

给定一颗树，每个点的权值 $a_i$ 在 $[l_i,r_i]$ 之间，已知每条边 $u,v$ 的边权 $w_{u,v}=a_{u}xor{a}_v$ ，

思路

参考：2021牛客多校4_bok_choy_的博客 队友的博客

队友写的好好，我摸了（）

具体实现见代码注释

代码

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int N = 1e5 + 10;

vector<P> seg;
int tr[N * 4]; //不能用结构体+build，点的范围太大了，会RE

//trL和trR是原本结构体中的tr[u].l, tr[u].r
void modify(int u, int trL, int trR, int l, int r, int dep, int Xor)
{
    if(dep < 0)
        return ;
    if(trL >= l && trR <= r) // 已经完全包含
    {
        int preL = trL & (((1 << 30) - 1) ^ ((1 << dep) - 1)); //原区间的相同前缀
        int preXor = Xor & (((1 << 30) - 1) ^ ((1 << dep) - 1));
        seg.push_back(P(preXor ^ preL, (preXor ^ preL) + (1 << dep) - 1));
        // preXor^preL: 新区间的相同前缀
        // (preXor ^ preL) + (1 << dep) - 1): 即后缀全1
        return ;
    }
    int mid = (trL + trR) >> 1;
    if(l <= mid)
        modify(u << 1, trL, mid, l, r, dep - 1, Xor);
    if(r > mid)
        modify(u << 1 | 1, mid + 1, trR, l, r, dep - 1, Xor);
}

int solve(int n)
{
    vector<P> vec;
    for(auto it : seg)
    {
        vec.push_back(P(it.first, -1)); //这里反过来，因为pair的优先级是先first后second，从小到大
        vec.push_back(P(it.second, 1));
    }
    int res = 0;
    sort(vec.begin(), vec.end());
    int len = vec.size();
    int cnt = 0;
    for(int i = 0; i < len; i++)
    {
        cnt -= vec[i].second;
        if(cnt == n) // 被覆盖n次，说明这个区间内的值都可以作为答案
            if(i + 1 < len)
                res += vec[i + 1].first - vec[i].first + 1;
    }
    return res;
}

int head[N], es[N * 2], nxt[N * 2], idx, cost[N * 2];
int a[N];

void addEdge(int u, int v, ll w)
{
    es[++idx] = v;
    nxt[idx] = head[u];
    cost[idx] = w;
    head[u] = idx;
}

// dfs求一组特解
void dfs(int u, int fa, ll w)
{
    a[u] = w;
    for(int i = head[u]; ~i; i = nxt[i])
    {
        int v = es[i];
        if(v == fa)
            continue;
        dfs(v, u, w ^ cost[i]);
    }
}

int l[N], r[N];

int main()
{
    int n;
    scanf("%d", &n);
    memset(head, -1, sizeof head);
    for(int i = 0; i < n; i++)
    {
        scanf("%d%d", l + i, r + i);
    }
    for(int i = 0, u, v; i < n - 1; i++)
    {
        ll w;
        scanf("%d%d%lld", &u, &v, &w);
        u--;
        v--;
        addEdge(u, v, w);
        addEdge(v, u, w);
    }
    dfs(0, -1, 0);
    for(int i = 0; i < n; i++)
    {
        modify(1, 0, (1 << 30) - 1, l[i], r[i], 30, a[i]);
    }
    printf("%d\n", solve(n));
    return 0;
}