POJ 2352 Star Treap||树状数组

                   发现这道题目用Treap居然可以做。。好吧,我从来没有想过,但是稍微思考一下发现这样也确实是有可行性,用Treap写完以后发现,用了500ms+,于是想看看用树状数组写要用多久,树状数组用了接近400ms,其实时间是差不多的,但是代码长度树状数组几乎是Treap的一半。。果然还是树状数组好写呀。

                  Treap的想法是插入横坐标,因为纵坐标单调递增,所以每一次只需要在Treap里面查询那些横坐标小于等于当前坐标的即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;

ll cnt=1,rt;
struct Node{
    ll son[2],pri,val,sz;
    void set_size();
    void newnode(ll v){son[0]=son[1]=0;sz=1,pri=rand(),val=v;}
}tree[100010];

void Node::set_size(){
     sz=1+tree[son[0]].sz+tree[son[1]].sz;
}

void rot(bool p,ll &x){
    ll k=tree[x].son[!p];
    tree[x].son[!p]=tree[k].son[p];
    tree[k].son[p]=x;
    tree[x].set_size();
    tree[k].set_size();
    x=k;
}

void ins(ll key,ll &x){
    if(x){
        bool p=key>=tree[x].val;
        //printf("ins:%d %d %d\n",key,tree[x].val,key>=tree[x].val);
        ins(key,tree[x].son[p]);
        if(tree[tree[x].son[p]].pri>tree[x].pri)
            rot(!p,x);
    }
    else
        tree[x=cnt++].newnode(key);
    tree[x].set_size();
}

void del(ll key,ll &x){
    if(key==tree[x].val){
        if(!tree[x].son[0])
            x=tree[x].son[1];
        else if(!tree[x].son[1])
            x=tree[x].son[0];
        else{
            bool p=tree[tree[x].son[0]].pri>tree[tree[x].son[1]].pri;
            rot(p,x);
            del(key,tree[x].son[p]);
        }
    }
    else
        del(key,tree[x].son[key>=tree[x].val]);
    if(x)
        tree[x].set_size();
}

ll find_num(ll key,ll &x){
    if(!x)
        return 0;
    if(tree[x].val>key)
        return find_num(key,tree[x].son[0]);
    else
        return tree[tree[x].son[0]].sz+1+find_num(key,tree[x].son[1]);
}

ll find_rank(ll level,ll &x){
    if(level==tree[tree[x].son[0]].sz+1)
        return x;
    else if(level>tree[tree[x].son[0]].sz+1)
        return find_rank(level-tree[tree[x].son[0]].sz-1,tree[x].son[1]);
    else
        return find_rank(level,tree[x].son[0]);
}

int n,x,y,level[15010];

int main(){
    while(scanf("%d",&n)!=EOF){
        for(int i=0;i<n;++i)
            scanf("%d%d",&x,&y),level[find_num(x,rt)]++,ins(x,rt);
        for(int i=0;i<n;++i)
            printf("%d\n",level[i]);
        memset(level,0,sizeof(int)*n);
        cnt=1,rt=0;
    }
    return 0;
}


if(!x) tree[x=cnt++]=Tree(key,rand(),1); else{ ++tree[x].sz; bool p=key<tree[x].key; ins(key,tree[x].son[!p]); if(tree[x].pri<tree[tree[x].son[!p]].pri) rot(p,x); }}int find_num(int key,int &x){ if(!x) return 0; else if(tree[x].key<=key) return tree[tree[x].son[0]].sz+1+find_num(key,tree[x].son[1]); else return find_num(key,tree[x].son[0]);}

                    树状数组的想法也是一样的,插入横坐标,查询小于等于当前横坐标的个数即可 

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int n,level[15010],bit[32010],x,y;
void update(int index);
int query(int index);

int main(){
    while(scanf("%d",&n)!=EOF){
        for(int i=1;i<=n;++i){
            scanf("%d%d",&x,&y);
            level[query(x+1)]++;
            update(x+1);
        }
        for(int i=0;i<n;++i)
            printf("%d\n",level[i]);
        memset(level,0,sizeof(int)*n);
        memset(bit,0,sizeof bit);
    }
    return 0;
}

void update(int index){
    while(index<=32001)
        bit[index]++,index+=index&-index;
}

int query(int index){
    int ans=0;
    while(index)
        ans+=bit[index],index-=index&-index;
    return ans;
}

               显然树状数组要短很多呀

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
POJ 2182是一道使用树状数组解决的题目,题目要求对给定的n个数进行排序,并且输出每个数在排序后的相对位置。树状数组是一种用来高效处理前缀和问题的数据结构。 根据引用中的描述,我们可以通过遍历数组a,对于每个元素a[i],可以使用二分查找找到a到a[i-1]中小于a[i]的数的个数。这个个数就是它在排序后的相对位置。 代码中的query函数用来求前缀和,add函数用来更新树状数组。在主函数中,我们从后往前遍历数组a,通过二分查找找到每个元素在排序后的相对位置,并将结果存入ans数组中。 最后,我们按顺序输出ans数组的元素即可得到排序后的相对位置。 参考代码如下: ```C++ #include <iostream> #include <cstdio> using namespace std; int n, a += y; } } int main() { scanf("%d", &n); f = 1; for (int i = 2; i <= n; i++) { scanf("%d", &a[i]); f[i = i & -i; } for (int i = n; i >= 1; i--) { int l = 1, r = n; while (l <= r) { int mid = (l + r) / 2; int k = query(mid - 1); if (a[i > k) { l = mid + 1; } else if (a[i < k) { r = mid - 1; } else { while (b[mid]) { mid++; } ans[i = mid; b[mid = true; add(mid, -1); break; } } } for (int i = 1; i <= n; i++) { printf("%d\n", ans[i]); } return 0; } ``` 这段代码使用了树状数组来完成题目要求的排序功能,其中query函数用来求前缀和,add函数用来更新树状数组。在主函数中,我们从后往前遍历数组a,通过二分查找找到每个元素在排序后的相对位置,并将结果存入ans数组中。最后,我们按顺序输出ans数组的元素即可得到排序后的相对位置。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [poj2182Lost Cows——树状数组快速查找](https://blog.csdn.net/aodan5477/article/details/102045839)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* [poj_2182 线段树/树状数组](https://blog.csdn.net/weixin_34138139/article/details/86389799)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值