POJ 1785 Binary Search Heap Construction [treap]

题意：给出N个 键值（string类）和优先级，求建好后的treap，按规格要求遍历输出。

范围：N<=5W， 且优先级唯一且也小于5W（题意里没说...）

解法：一开始很傻的直接建treap，然后塞塞塞，最后果断TLE了.. 需要抓住Treap的本质，原本BST中加入heap的原因是为了防止BST退化成一条链，所以heap的权值需要用随机值，这样可以保证随机复杂度是logN，但是此时优先级是已知的，所以题目完全可以构造出一条链，如果按照原来的从根往下加节点，肯定会TLE！我们可以先按键值排序，这样每一个节点都是插在上一个节点的右端，然后根据优先级不断左旋上移即可。（维护heap的复杂度是NlogN，记得数据结构课里有说）

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<stdlib.h>
#include<set>
#include<map>
#include<queue>
#include<vector>
#include<bitset>
#pragma comment(linker, "/STACK:1024000000,1024000000")
template <class T>
bool scanff(T &ret){ //Faster Input
    char c; int sgn; T bit=0.1;
    if(c=getchar(),c==EOF) return 0;
    while(c!='-'&&c!='.'&&(c<'0'||c>'9')) c=getchar();
    sgn=(c=='-')?-1:1;
    ret=(c=='-')?0:(c-'0');
    while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
    if(c==' '||c=='\n'){ ret*=sgn; return 1; }
    while(c=getchar(),c>='0'&&c<='9') ret+=(c-'0')*bit,bit/=10;
    ret*=sgn;
    return 1;
}
#define inf 1073741823
#define llinf 4611686018427387903LL
#define PI acos(-1.0)
#define lth (th<<1)
#define rth (th<<1|1)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define drep(i,a,b) for(int i=a;i>=b;i--)
#define gson(i,root) for(int i=ptx[root];~i;i=ed[i].next)
#define tdata int testnum;scanff(testnum);for(int cas=1;cas<=testnum;cas++)
#define mem(x,val) memset(x,val,sizeof(x))
#define mkp(a,b) make_pair(a,b)
#define findx(x) lower_bound(b+1,b+1+bn,x)-b
#define pb(x) push_back(x)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;


int n;
int root;
struct node{
    int l,r,rnd,fa;
}t[100100];
void insert(int i,int rnd){
    int x=i-1;
    t[x].r=i;
    t[i].fa=x;
    t[i].rnd=rnd;
    t[i].l=t[i].r=0;
    //printf("-->%d %d\n",x,i);
    while(x&&t[t[x].r].rnd>t[x].rnd){

        int tmp=t[x].r;
        //printf("x=%d root=%d! fa=%d tmp=%d ",x,root,t[x].fa,tmp);
        t[x].r=t[tmp].l;
        t[t[tmp].l].fa=x;
        t[tmp].l=x;
        t[tmp].fa=t[x].fa;
        t[x].fa=tmp;

        if(t[t[tmp].fa].l==x)t[t[tmp].fa].l=tmp;
        if(t[t[tmp].fa].r==x)t[t[tmp].fa].r=tmp;
        x=t[tmp].fa;
        if(x==0)root=tmp;
        //printf("-> x=%d root=%d! \n",x,root);
    }
}

struct node2{
    char s[111];
    int rnd;
    bool operator < (const node2 x)const{
        return strcmp(s,x.s)<0;
    }
}a[100100];
int has[100100];

void dfs(int x){
    printf("(");
    if(t[x].l)dfs(t[x].l);
    printf("%s/%d",a[has[t[x].rnd]].s,t[x].rnd);
    if(t[x].r)dfs(t[x].r);
    printf(")");
}

int main(){
    while(scanf("%d",&n)!=EOF,n){
        rep(i,1,n){
            char c;int len=0;
            while(c=getchar(),c<'a'||c>'z');
            a[i].s[len++]=c;
            while(c=getchar(),c>='a'&&c<='z')a[i].s[len++]=c;
            a[i].s[len]=0;
            scanf("%d",&a[i].rnd);
            //printf("-->%s %d\n",a[i].s,a[i].rnd);
        }
        sort(a+1,a+1+n);
        rep(i,1,n)has[a[i].rnd]=i;
        root=1;
        t[1].rnd=a[1].rnd;
        t[1].l=t[1].r=0;
        t[1].fa=0;
        rep(i,2,n)insert(i,a[i].rnd);
        dfs(root);
        printf("\n");
    }
    return 0;
}