POJ2352(Treap || 线段树)

来源:互联网 发布:java反转单链表递归 编辑:程序博客网 时间:2024/05/21 15:50

题目链接:http://poj.org/problem?id=2352
题目大意:先按Y的升序然后再按X的升序给出坐标,求出每个坐标左下方有多少个点,最后输出有从0到n-1个点的坐标有多少个。
本题可以说是线段树和树状数组的入门题,不用管Y按顺序依次求出每个X插入之前0到x有多少个坐标即可。这道题还可以用Treap来做,也是模板题,由于是刚学Treap,就用这个来写了。

资料:树堆,在数据结构中也称Treap,是指有一个随机附加域满足堆的性质的二叉搜索树,其结构相当于以随机数据插入的二叉搜索树。其基本操作的期望时间复杂度为O(logn)。相对于其他的平衡二叉搜索树,Treap的特点是实现简单,且能基本实现随机平衡的结构。
右旋

#include <iostream>#include <cstdio>#include <cstring>#include <string>#include <queue>#include <ctime>#include <cstdlib>using namespace std;const int maxn = 20000 + 5;int cnt = 1, rt = 0;//第一个节点从1开始int n, ans[maxn], x, y;struct Tree{    int key, pri, size, son[2];    //插入值,优先级,大小,左右儿子    void set(int x, int y, int z)//初始化    {        key = x;        pri = y;        size = z;        son[0] = son[1] = 0;    }}T[maxn];void push_up(int rt){    T[rt].size = T[T[rt].son[0]].size + 1 + T[T[rt].son[1]].size;}void rotate(int p, int &x)//旋转时左旋右旋放在一起{                       //p=1时左旋,p=0时右旋    int y = T[x].son[!p];    T[x].son[!p] = T[y].son[p];    T[y].son[p] = x;    push_up(x), push_up(y);    x = y;}void Insert(int key, int &x){//每次插入时从根遍历插到叶子节点,然后再回溯旋转    if (x == 0)    {        T[x=cnt++].set(key, rand(), 1);        return;    }    else {        T[x].size++;        int p = key <= T[x].key;        Insert(key, T[x].son[!p]);        if (T[x].pri > T[T[x].son[!p]].pri)            rotate(p, x);    }}int find(int key, int &x){    if (x == 0) return 0;    if (key >= T[x].key)        return T[T[x].son[0]].size + 1 + find(key, T[x].son[1]);    else return find(key, T[x].son[0]);}int main(){    //freopen("input.txt", "r", stdin);    cin >> n;    for (int i = 1; i <= n; i++)    {        scanf("%d%d", &x, &y);        ans[find(x, rt)]++;        Insert(x, rt);    }    for (int i = 0; i < n; i++) printf("%d\n", ans[i]);    return 0;}
0 0
原创粉丝点击