[HNOI2002]营业额统计

           初学splay,比着cxlove的抄的。。http://blog.csdn.net/ACM_cxlove?viewmode=contents           by---cxlove 作为模板用。。只有插入,求前驱,后继。剩下的用途慢慢积累吧。。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;

const int N = 100005;
const int INF = 0x3f3f3f3f;
int pre[N], key[N], ch[N][2], root, tot;
//分别表示父结点,键值,左右孩子(0左1右),根结点,结点数量
int n;
//新建一个结点
void NewNode(int &r, int father, int k)
{
    r = ++ tot;
    pre[r] = father;
    key[r] = k;
    ch[r][0] = ch[r][1] = 0;
}
//旋转,kind为1是右旋,为0是左旋
void Rotate(int x, int kind)
{
    int y = pre[x];
    ch[y][!kind] = ch[x][kind];
    pre[ch[x][kind]] = y;
    if(pre[y])
        ch[pre[y]][ch[pre[y]][1] == y] = x;
    pre[x] = pre[y];
    ch[x][kind] = y;
    pre[y] = x;
}
//Splay调整,将结点r调整到goal下面
void Splay(int r, int goal)
{
    while(pre[r] != goal)
    {
        if(pre[pre[r]] == goal) Rotate(r, ch[pre[r]][0] == r);
        else
        {
            int y = pre[r];
            int kind = ch[pre[y]][0] == y;
            if(ch[y][kind] == r)
            {
                Rotate(r, !kind);
                Rotate(r, kind);
            }
            else
            {
                Rotate(y, kind);
                Rotate(r, kind);
            }
        }
    }
    if(goal == 0) root = r;
}

int Insert(int k)
{
    int r = root;
    while(ch[r][key[r] < k])
    {
        //no repeat;
        if(key[r] == k)
        {
            Splay(r, 0);
            return 0;
        }
        r = ch[r][key[r] < k];
    }
    NewNode(ch[r][key[r] < k], r, k);
    Splay(ch[r][key[r] < k], 0);
    return 1;
}
//找前驱
int get_pre(int x)
{
    int tmp = ch[x][0];
    if(tmp == 0) return INF;
    while(ch[tmp][1])
        tmp = ch[tmp][1];
    return key[x] - key[tmp];
}
//找后继
int get_next(int x)
{
    int tmp = ch[x][1];
    if(tmp == 0) return INF;
    while(ch[tmp][0])
        tmp = ch[tmp][0];
    return key[tmp] - key[x];
}

int main()
{
    while(scanf("%d", &n) != EOF)
    {
        root = tot = 0;
        int ans = 0;
        for(int i = 0; i < n; i ++)
        {
            int num;
            if(scanf("%d", &num) == EOF) num = 0;
            if(i == 0)
            {
                ans += num;
                NewNode(root, 0, num);
                continue;
            }
            if(Insert(num) == 0) continue;
            int a = get_pre(root);
            int b = get_next(root);
            ans += min(a, b);
        }
        printf("%d\n", ans);
    }
    return 0;
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值