poj3468 线段树 or splay

poj3468  裸线段树。因为在熟悉splay 所以就用splay交了一发。。。开始用的scanf()!==2 居然TLE了。。。然后我就当单组测试数据做的 然后就过了  囧TZ

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
#define rt ch[root][1]
#define lrt ch[rt][0]
#define ls ch[x][0]
#define rs ch[x][1]
#define inf 0x3f3f3f3f
const int maxn=100100;

int root,top;//root为新建节点时的根节点
int ch[maxn][2],f[maxn];
long long sz[maxn],num[maxn],val[maxn],sum[maxn],add[maxn];
int n,m;

//ch[x][0],ch[x][1]:x的左右儿子,f[x]:x的父亲节点,val[x]:x的值,sz[x]以x为根的子树的节点个数
inline void pushup(int x)
{
    sz[x]=sz[ls]+sz[rs]+1;
    sum[x]=sum[ls]+sum[rs]+val[x];
    //维护其他信息
}
inline void pushdown(int x)
{
    if(add[x])
    {
        val[x]+=add[x];
        add[ls]+=add[x];
        add[rs]+=add[x];
        sum[ls]+=(long long)(sz[ls]*add[x]);
        sum[rs]+=(long long)(sz[rs]*add[x]);
        add[x]=0;
    }
    //维护其他信息
}

inline void link(int y,int x,int d)
//把节点y的(左或右)儿子设为x
{
    f[x]=y;
    ch[y][d]=x;
}

inline int is(int x)
//判断x为左儿子0还是右儿子1
{
    return ch[f[x]][1]==x;
}

inline void rotate(int x,int d)
{
    int y=f[x];
    int z=f[y];
    pushdown(y);
    pushdown(x);
    link(y,ch[x][d],!d);
    if(z) link(z,x,is(y));
    f[x]=z;
    link(x,y,d);
    pushup(y);
}
inline void zag(int x)
{
    rotate(x,0);
}
inline void zig(int x)
{
    rotate(x,1);
}
inline void splay(int x,int goal=0)//第二个元素没有传参就默认为0,splay到0即将该节点旋转到根
//将根为r的子树调整到x的父节点为goal的位置
{
    pushdown(x);

    while(f[x]!=goal)
    {
        int y=f[x];
        int z=f[y];
        if(z==goal)
        {
            rotate(x,!is(x));
            break;
        }
        if(ch[z][0]==y)
        {
            if(ch[y][0]==x) zig(y),zig(x);
            else zag(x),zig(x);
        }
        else
        {
            if(ch[y][1]==x) zag(y),zag(x);
            else zig(x),zag(x);
        }
    }
    if(goal==0) root=x;
    pushup(x);
}

//splay操作一般不做改动
inline void newnode(int &x,int father,long long v)
{
    x=++top;
    f[x]=father;
    val[x]=v;
    add[x]=0;
    ls=rs=0;
    sz[x]=0;

    //.....按题意建立信息
}
inline bool insert(int v)
{
    int x=root;
    while(ch[x][val[x]<v])
    {
        if(val[x]==v)
        {
            splay(x);
            return false;
        }
        x=ch[x][val[x]<v];
    }
    newnode(ch[x][val[x]<v],x,v);
    splay(ch[x][val[x]<v]);
    return true;
}

inline void build(int &x,int y,int l,int r)
{
    if(l>r) return ;
    int mid=(l+r)>>1;
    newnode(x,y,num[mid]);
    build(ls,x,l,mid-1);
    build(rs,x,mid+1,r);
    pushup(x);

}

inline void init(int n)
{
    root=top=0;
    ch[0][0]=ch[0][1]=f[0]=val[0]=add[0]=sz[0]=sum[0]=0;
    newnode(root,0,0);
    newnode(rt,root,0);
    sz[root]=2;
    for(int i=0; i<n; i++) scanf("%lld",&num[i]);
    build(lrt,rt,0,n-1);
    pushup(rt);
    pushup(root);
    //for(int x=0;x<=top;x++){
    //    printf("x=%d f[x]=%d ls=%d rs=%d val=%d sum=%d add=%d sz=%d\n",x,f[x],ls,rs,val[x],sum[x],add[x],sz[x]);
    //}
}

inline int pre(int x)//获得前驱,调用时pre(root)即可
{
    x=ls;
    if(x==0) return inf;
    while(rs) x=rs;
    return val[x];
}

inline int next(int x)//调用时next(root)即可,后继
{
    x=rs;
    if(x==0) return inf;
    while(ls) x=ls;
    return val[x];
}

inline int get_k(int x,int k)//返回的是第k小的点的标号(从小到大),值要用val来获取
{
    pushdown(x);
    int num=sz[ls]+1;
    if(num==k) return x;
    if(k<num) return get_k(ls,k);
    return get_k(rs,k-num);
}

inline void update()
{
    int l,r;
    long long c;
    scanf("%d %d %lld",&l,&r,&c);
    int x=get_k(root,l);
    splay(x);
    int y=get_k(root,r+2);
    splay(y,root);
    add[lrt]+=c;
    sum[lrt]+=sz[lrt]*c;
}
//查询一段区间的值
inline void query()
{
    int l,r;
    scanf("%d%d",&l,&r);
    int x,y;
    x=get_k(root,l);
    y=get_k(root,r+2);
    splay(x);
    splay(y,root);
    printf("%lld\n",sum[lrt]);
}

int main(){
    scanf("%d%d",&n,&m);
    init(n);
    for(int i=1;i<=m;i++){
        int l,r,d;
        char c;
        scanf("%c%c",&c,&c);
        if(c==‘Q‘){
            //cout<<c<<endl;
            query();
        }
        if(c==‘C‘){
            update();

        }
    }
}

poj3468 线段树 or splay

时间: 2024-10-12 17:36:09

poj3468 线段树 or splay的相关文章

刷题总结——二逼平衡树(bzoj3224线段树套splay)

题目: Description 您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:1.查询k在区间内的排名2.查询区间内排名为k的值3.修改某一位值上的数值4.查询k在区间内的前驱(前驱定义为小于x,且最大的数)5.查询k在区间内的后继(后继定义为大于x,且最小的数) Input 第一行两个数 n,m 表示长度为n的有序序列和m个操作第二行有n个数,表示有序序列下面有m行,opt表示操作标号若opt=1 则为操作1,之后有三个数l,r,k 表示查询k在区间[l,r

poj3468 线段树

http://poj.org/problem?id=3468 题目链接, 很经典的线段树的应用, 这里复习一下, 再写一遍, 代码如下: #include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int maxn = 100000 + 10; int N, Q; int a[maxn]; struct Segment{ int l, r; long long

poj3468 线段树+lazy标记

A Simple Problem with Integers Time Limit: 5000MS   Memory Limit: 131072K Total Submissions: 92921   Accepted: 28910 Case Time Limit: 2000MS Description You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of

POJ3468——线段树区间更新——A Simple Problem with Integers

You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval. Input The firs

BZOJ_1901_&amp;_ZJU_2112_Dynamic_Rankings(主席树+树状数组/线段树+(Treap/Splay))

描述 http://www.lydsy.com/JudgeOnline/problem.php?id=1901 给出一个长度为n的数列A,有m次询问,询问分两种:1.修改某一位置的值;2.求区间[l,r]内的第k小的值. 分析 对于只有第一种询问的问题: POJ_2104_Kth(主席树) 现在要求动态.我们思考这样一个问题:把求区间第k小的问题变成求区间和值的问题,这个好解决吧?对于静态的问题,我们使用前缀和即可解决,那么对于动态的呢?使用树状数组维护前缀和.那么现在把问题变回求区间第k小值的

3196. 二逼平衡树【线段树套splay】

Description 您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作: 1.查询k在区间内的排名 2.查询区间内排名为k的值 3.修改某一位值上的数值 4.查询k在区间内的前驱(前驱定义为小于x,且最大的数) 5.查询k在区间内的后继(后继定义为大于x,且最小的数) Input 第一行两个数 n,m 表示长度为n的有序序列和m个操作 第二行有n个数,表示有序序列 下面有m行,opt表示操作标号 若opt=1 则为操作1,之后有三个数l,r,k 表示查询k在区间

POJ-3468(线段树+区间更新+区间查询)

A Simple Problem With Integers POJ-3468 这题是区间更新的模板题,也只是区间更新和区间查询和的简单使用. 代码中需要注意的点我都已经标注出来了,容易搞混的就是update函数里面还需要计算sum数组.因为这里查询的时候是直接用sum查询结点. //区间更新,区间查询 #include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #inc

poj3468(线段树-区间修改)

模板题: #include<cstdio> #include<cstring> #define ll long long const int N = 100000 + 10; ll sum[N << 2]; ll addv[N << 2]; int num[N]; int n,q; void pushUp(int u) { sum[u] = sum[u * 2] + sum[u * 2 + 1]; } void build(int u, int L, int

POJ3468(线段树区间维护)

#include<cstdio> #define lson n<<1,l,mid #define rson (n<<1)|1,mid+1,r #define gmid (a[n].l+a[n].r)>>1 using namespace std; const int MAX_N=100005; typedef long long LL; struct node{ int l,r; LL sum,lazy; }a[MAX_N<<2]; void P