Tyvj 1729 文艺平衡树 HYSBZ - 3223 splay翻转区间模板

题目链接:https://vjudge.net/problem/HYSBZ-3223
思路:先把l之前的断开,记为left,然后把r之后的断开,记为left,中间需要翻转的记为mid,打上标记,然后合并。

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;

struct Node
{
    Node *ch[2];
    int s;
    int flip;
    int v;
    int cmp(int k) const
    {
        int d = k - ch[0]->s;
        if(d == 1)
            return -1;
        return d <= 0 ? 0 : 1;
    }
    void maintain()
    {
        s = ch[0]->s + ch[1]->s + 1;
    }
    void pushdown()
    {
        if(flip)
        {
            flip = 0;
            swap(ch[0], ch[1]);
            ch[0]->flip = !ch[0]->flip;
            ch[1]->flip = !ch[1]->flip;
        }
    }
};

Node *null = new Node();

void rotate(Node* &o, int d)
{
    Node* k = o->ch[d^1];
    o->ch[d^1] = k->ch[d];
    k->ch[d] = o;
    o->maintain();
    k->maintain();
    o = k;
}

void splay(Node* &o, int k)
{
    o->pushdown();
    int d = o->cmp(k);
    if(d == 1)
        k -= o->ch[0]->s + 1;
    if(d != -1)
    {
        Node* p = o->ch[d];
        p->pushdown();
        int d2 = p->cmp(k);
        int k2 = (d2 == 0 ? k : k - p->ch[0]->s - 1);
        if(d2 != -1)
        {
            splay(p->ch[d2], k2);
            if(d == d2)
                rotate(o, d^1);
            else
                rotate(o->ch[d], d);
        }
        rotate(o, d^1);
    }
}

// 合并left和right。假定left的所有元素比right小。注意right可以是null,但left不可以
Node* merge(Node* left, Node* right)
{
    splay(left, left->s);
    left->ch[1] = right;
    left->maintain();
    return left;
}

// 把o的前k小结点放在left里,其他的放在right里。1<=k<=o->s。当k=o->s时,right=null
void split(Node* o, int k, Node* &left, Node* &right)
{
    splay(o, k);
    left = o;
    right = o->ch[1];
    o->ch[1] = null;
    left->maintain();
}

const int maxn = 100000 + 10;
struct SplaySequence
{
    int n;
    Node seq[maxn];
    Node *root;

    Node* build(int sz)
    {
        if(!sz)
            return null;
        Node* L = build(sz/2);
        Node* o = &seq[++n];
        o->v = n; // 节点编号
        o->ch[0] = L;
        o->ch[1] = build(sz - sz/2 - 1);
        o->flip = o->s = 0;
        o->maintain();
        return o;
    }

    void init(int sz)
    {
        n = 0;
        null->s = 0;
        root = build(sz);
    }
};

vector<int> ans;
void print(Node* o)
{
    if(o != null)
    {
        o->pushdown();
        print(o->ch[0]);
        ans.push_back(o->v);
        print(o->ch[1]);
    }
}

void debug(Node* o)
{
    if(o != null)
    {
        o->pushdown();
        debug(o->ch[0]);
        printf("%d ", o->v-1);
        debug(o->ch[1]);
    }
}

SplaySequence ss;
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    ss.init(n+1); // 最前面有一个虚拟结点

    while (m--)
    {
        int a, b;
        scanf("%d%d", &a, &b);
        Node *left, *mid, *right, *o;
        split(ss.root, a, left, o); // 如果没有虚拟结点,a将改成a-1,违反split的限制
        split(o, b-a+1, mid, right);
        mid->flip ^= 1;
        ss.root = merge(merge(left, mid), right);
    }
    print(ss.root);
    for(int i = 1; i < ans.size(); i++)
        printf("%d ", ans[i]-1); // 节点编号减1才是本题的元素值

    return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值