Jiu Yuan Wants to Eat【2018焦作网络赛】【树链剖分】

106 篇文章 0 订阅
24 篇文章 0 订阅

题目链接

树链剖分学习笔记,可以看这里


  这道题还真挺好的,以前不会做,现在想了发现,学过树链剖分之后,剩下的部分就是处理去反那块比较的不容易些了,但是想了一下午,现在还是给我敲出来了,我们主要难处理的就是关于求反,那么怎么处理求反?

  一开始读题的时候,我还在想,是不是求反是与此数的最大2次幂系数开始,后来推了测试样例,发现,是直接对64为比特位都求反,那么我们似乎可以推一个公式了,因为2^64-1就是满unsigned long long,而且之前用过无符号类型就是在哈希里,它会在满了之后自动向前重新开始,所以相当于是在自动取模,那么,怎么处理去反呢?

  去反可以看作111111......11111-X,那么,换句话说,不就是相当于想乘以“-1”在加上unsigned long long的满值吗?那么建树不就是完成了吗?


#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 1e5+5;
const ull INF=(ull)0-(ull)1;
int N, Q, cnt, head[maxN], root[maxN], depth[maxN], size[maxN], W_son[maxN], top[maxN], id[maxN], num;
ull tree[maxN<<2], lazy[maxN<<2], mult[maxN<<2];
struct Eddge
{
    int nex, to;
    Eddge(int a=-1, int b=0):nex(a), to(b) {}
}edge[maxN<<1];
void addEddge(int u, int v)
{
    edge[cnt] = Eddge(head[u], v);
    head[u] = cnt++;
}
void dfs1(int u, int fa, int deep)
{
    root[u] = fa;
    depth[u] = deep;
    size[u] = 1;
    int maxxSon = -1;
    for(int i=head[u]; i!=-1; i=edge[i].nex)
    {
        int v = edge[i].to;
        if(v == fa) continue;
        dfs1(v, u, deep+1);
        size[u] += size[v];
        if(maxxSon < size[v])
        {
            maxxSon = size[v];
            W_son[u] = v;
        }
    }
}
void dfs2(int x, int topf)
{
    top[x] = topf;
    id[x] = ++num;
    if(!W_son[x]) return;
    dfs2(W_son[x], topf);
    for(int i=head[x]; i!=-1; i=edge[i].nex)
    {
        int y = edge[i].to;
        if(y==W_son[x] || y==root[x]) continue;
        dfs2(y, y);
    }
}
void buildTree(int rt, int l, int r)
{
    lazy[rt] = 0;   mult[rt] = 1;
    if(l == r)
    {
        tree[rt] = 0;
        return;
    }
    int mid = (l + r)>>1;
    buildTree(rt<<1, l, mid);
    buildTree(rt<<1|1, mid+1, r);
    tree[rt] = 0;
}
void pushup(int rt)
{
    tree[rt] = tree[rt<<1] + tree[rt<<1|1];
}
void pushdown(int rt, int l, int r)
{
    if(mult[rt]!=1)
    {
        mult[rt<<1]*=mult[rt];  lazy[rt<<1]*=mult[rt];
        mult[rt<<1|1]*=mult[rt];    lazy[rt<<1|1]*=mult[rt];
        tree[rt<<1]*=mult[rt];
        tree[rt<<1|1]*=mult[rt];
        mult[rt] = 1;
    }
    if(lazy[rt])
    {
        lazy[rt<<1] += lazy[rt];
        lazy[rt<<1|1] += lazy[rt];
        int mid = (l + r)>>1;
        tree[rt<<1] += lazy[rt]*(mid - l + 1);
        tree[rt<<1|1] += lazy[rt]*(r - mid);
        lazy[rt] = 0;
    }
}
void update_add(int rt, int l, int r, int ql, int qr, ull val)
{
    if(ql<=l && qr>=r)
    {
        lazy[rt] += val;
        tree[rt] += val*(r - l + 1);
        return;
    }
    int mid = (l + r)>>1;
    pushdown(rt, l, r);
    if(ql>mid) update_add(rt<<1|1, mid+1, r, ql, qr, val);
    else if(qr<=mid) update_add(rt<<1, l, mid, ql, qr, val);
    else
    {
        update_add(rt<<1, l, mid, ql, qr, val);
        update_add(rt<<1|1, mid+1, r, ql, qr, val);
    }
    pushup(rt);
}
void update_multi(int rt, int l, int r, int ql, int qr, ull val)
{
    if(ql<=l && qr>=r)
    {
        lazy[rt] *= val;
        mult[rt] *= val;
        tree[rt] *= val;
        return;
    }
    pushdown(rt, l, r);
    int mid = (l + r)>>1;
    if(ql>mid) update_multi(rt<<1|1, mid+1, r, ql, qr, val);
    else if(qr<=mid) update_multi(rt<<1, l, mid, ql, qr, val);
    else
    {
        update_multi(rt<<1, l, mid, ql, qr, val);
        update_multi(rt<<1|1, mid+1, r, ql, qr, val);
    }
    pushup(rt);
}
void update_Range_add(int x, int y, ull val)
{
    while(top[x] != top[y])
    {
        if(depth[top[x]] < depth[top[y]]) swap(x, y);
        update_add(1, 1, N, id[top[x]], id[x], val);
        x = root[top[x]];
    }
    if(depth[x] > depth[y]) swap(x, y);
    update_add(1, 1, N, id[x], id[y], val);
}
void update_Range_Multi(int x, int y, ull val)
{
    while(top[x] != top[y])
    {
        if(depth[top[x]] < depth[top[y]]) swap(x, y);
        update_multi(1, 1, N, id[top[x]], id[x], val);
        x = root[top[x]];
    }
    if(depth[x] > depth[y]) swap(x, y);
    update_multi(1, 1, N, id[x], id[y], val);
}
ull query(int rt, int l, int r, int ql, int qr)
{
    if(ql<=l && qr>=r) return tree[rt];
    pushdown(rt, l, r);
    int mid = (l + r)>>1;
    if(ql>mid) return query(rt<<1|1, mid+1, r, ql, qr);
    else if(qr<=mid) return query(rt<<1, l, mid, ql, qr);
    else
    {
        ull ans = query(rt<<1, l, mid, ql, qr);
        ans += query(rt<<1|1, mid+1, r, ql, qr);
        return ans;
    }
}
ull query_Range(int x, int y)
{
    ull ans = 0;
    while(top[x] != top[y])
    {
        if(depth[top[x]] < depth[top[y]]) swap(x, y);
        ans += query(1, 1, N, id[top[x]], id[x]);
        x = root[top[x]];
    }
    if(depth[x] > depth[y]) swap(x, y);
    ans += query(1, 1, N, id[x], id[y]);
    return ans;
}
void init()
{
    cnt = num = 0;
    memset(head, -1, sizeof(head));
    memset(W_son, 0, sizeof(W_son));
}
int main()
{
    while(scanf("%d", &N)!=EOF)
    {
        init();
        for(int i=2; i<=N; i++)
        {
            int e1; scanf("%d", &e1);
            addEddge(i, e1);
            addEddge(e1, i);
        }
        dfs1(1, 1, 0);
        dfs2(1, 1);
        buildTree(1, 1, N);
        scanf("%d", &Q);
        while(Q--)
        {
            int op, u, v;
            ull x;
            scanf("%d", &op);
            if(op == 1)
            {
                scanf("%d%d%llu", &u, &v, &x);
                update_Range_Multi(u, v, x);
            }
            else if(op == 2)
            {
                scanf("%d%d%llu", &u, &v, &x);
                update_Range_add(u, v, x);
            }
            else if(op == 3)
            {
                scanf("%d%d", &u, &v);
                update_Range_Multi(u, v, -1);
                update_Range_add(u, v, INF);
            }
            else
            {
                scanf("%d%d", &u, &v);
                printf("%llu\n", query_Range(u, v));
            }
        }
    }
    return 0;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Wuliwuliii

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值