[234树]2019ICPC银川网络赛 E 2-3-4 tree

234树：有三种结点，分别是2、3、4结点，表示该结点将区间划分为几块。并且某个结点的孩子要么全都有，要么全没有。

这种树是这么构成的：

当要插入x这个数时，从根开始对结点重复这么个过程：

1.如果当前结点是4结点（就是当前结点有3个数x,y,z  x<y<z）：

-当前结点为根，则新建一个结点，包含一个数y，这个结点是新根，并且是一个2结点，左孩子是包含x的结点，右孩子是包含z的结点。操作结束后，当前结点变为新根，继续执行接下来的步骤

-当前结点不为根，y放到当前结点的父结点中去，再把去掉y的当前结点拆成两个2结点。操作结束后当前结点变为当前结点的父结点，继续执行接下来的步骤

2.如果当前结点是叶子结点，插入x

3.否则，当前结点变成一个包含x的区间的子结点

 

做的过程中多次读错题意，1过程中结点变为新根或父亲结点之后是要接着执行接下来的步骤的，而不是从1重新开始= =

我第一版是用数组模拟的，结果表示当前结点是其父亲结点的左儿子还是右儿子的变量dir维护出现错误，导致一直RE（计蒜客上返回T）。所以dir最后其实是并没有用上的。

vector写起来更简单清晰一点。

vecotr版本：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
double ans1,ans2;
int n,m,t,ca,cnt;
struct node{
    int fa;
    vector<int>num,ch;
}tr[maxn];
void getorder(node &rt){
    sort(rt.num.begin(),rt.num.end());
}
int newnode(int x,int fa){
    tr[++cnt].fa=fa;
    tr[cnt].num.clear();
    tr[cnt].ch.clear();
    tr[cnt].num.push_back(x);
    return cnt;
}
void dfs(int now){
    for(int i=0;i<tr[now].num.size();i++) printf("%d%c",tr[now].num[i],i==tr[now].num.size()-1?'\n':' ');
    for(int i=0;i<tr[now].ch.size();i++){
        if(tr[now].ch[i]) dfs(tr[now].ch[i]);
    }
}
void add(int x){
    int now=1,tmp1,tmp2,fl=1;
    while(1){
        int pre=tr[now].fa;
        if((int)tr[now].num.size()==3){
            tmp1=newnode(tr[now].num[0],pre);
            tmp2=newnode(tr[now].num[2],pre);
            if(tr[now].ch.size()){
                tr[tmp1].ch.push_back(tr[now].ch[0]);
                tr[tmp1].ch.push_back(tr[now].ch[1]);
                tr[tmp2].ch.push_back(tr[now].ch[2]);
                tr[tmp2].ch.push_back(tr[now].ch[3]);
            }
            for(int j=0;j<(int)tr[tmp1].ch.size();j++) tr[ tr[tmp1].ch[j]].fa=tmp1;
            for(int j=0;j<(int)tr[tmp2].ch.size();j++) tr[ tr[tmp2].ch[j]].fa=tmp2;
            if(now==1){//如果是根
                int tmp=tr[now].num[1];
                tr[now].num.clear();
                tr[now].ch.clear();
                tr[now].num.push_back(tmp);
                tr[now].ch.push_back(tmp1);
                tr[now].ch.push_back(tmp2);
                tr[tmp1].fa=tr[tmp2].fa=now;
            }else{
                if(tr[pre].num.size()==1){
                    if(tr[now].num[1]>tr[pre].num[0]){
                        tr[pre].ch[1]=tmp1;
                        tr[pre].ch.push_back(tmp2);
                    }else{
                        tr[pre].ch.push_back(tr[pre].ch[1]);
                        tr[pre].ch[1]=tmp2;
                        tr[pre].ch[0]=tmp1;
                    }
                }else{
                    if(tr[now].num[1]<tr[pre].num[0]){
                        tr[pre].ch.push_back(tr[pre].ch[2]);
                        tr[pre].ch[2]=tr[pre].ch[1];
                        tr[pre].ch[1]=tmp2;
                        tr[pre].ch[0]=tmp1;
                    }else if(tr[now].num[1]>tr[pre].num[1]){
                        tr[pre].ch[2]=tmp1;
                        tr[pre].ch.push_back(tmp2);
                    }else{
                        tr[pre].ch.push_back(tr[pre].ch[2]);
                        tr[pre].ch[1]=tmp1;
                        tr[pre].ch[2]=tmp2;
                    }
                }
                tr[pre].num.push_back(tr[now].num[1]);
                getorder(tr[pre]);
                now=pre;
            }
        }
        if(tr[now].ch.size()==0){//叶子结点
                tr[now].num.push_back(x);
                getorder(tr[now]);
                break;
        }else{
            int fl=-1;
            for(int i=0;i<tr[now].num.size();i++){
                if(tr[now].num[i]>x){
                    fl=i;break;
                }
            }
            now=tr[now].ch[fl==-1?tr[now].num.size():fl];
        }
    }
}
int main() {
    scanf("%d", &t);
    while(t--){
        cnt=1;
        tr[1]={0};tr[1].num.clear();tr[1].ch.clear();
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            int x;
            scanf("%d",&x);
            add(x);
        }
        printf("Case #%d:\n",++ca);
        dfs(1);
    }
    return 0;
}

数组模拟：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
double ans1,ans2;
int n,m,t,ca,cnt;
struct node{
    int sz,num[3],ch[4],fa,dir;
}tr[100005];
void getorder(node &rt){
    sort(rt.num,rt.num+rt.sz);
}
int newnode(int x,int fa,int dir){
    tr[++cnt]={1,{x,0,0},{0,0,0,0},fa,dir};
    return cnt;
}
#define lson (tr[now].ch[0])
#define mlson (tr[now].ch[1])
#define mrson (tr[now].ch[2])
#define rson (tr[now].ch[3])
int connect(int rt1,int rt2){//把两个子树并起来
    return rt1?rt1:rt2;
}
void dfs(int now){
    for(int i=0;i<tr[now].sz;i++) printf("%d%c",tr[now].num[i],i==tr[now].sz-1?'\n':' ');
    for(int i=0;i<=3;i++){
        if(tr[now].ch[i]) dfs(tr[now].ch[i]);
    }
}
void add(int x){
    int now=1,tmp1,tmp2,fl=1;
    while(1){
        int pre=tr[now].fa;
        if(tr[now].sz==3){
            tmp1=newnode(tr[now].num[0],pre,0);
            tmp2=newnode(tr[now].num[2],pre,1);
            tr[tmp1].ch[0]=tr[now].ch[0];tr[tmp1].ch[1]=tr[now].ch[1];
            tr[tmp2].ch[0]=tr[now].ch[2];tr[tmp2].ch[1]=tr[now].ch[3];
            for(int j=0;j<2;j++) if(tr[tmp1].ch[j]) tr[ tr[tmp1].ch[j]].fa=tmp1;
            for(int j=0;j<2;j++) if(tr[tmp2].ch[j]) tr[ tr[tmp2].ch[j]].fa=tmp2;
            if(now==1){//如果是根
                tr[now].sz=1;tr[now].num[0]=tr[now].num[1];
                tr[now].ch[0]=tmp1;tr[now].ch[1]=tmp2;tr[now].ch[2]=tr[now].ch[3]=0;
                tr[tmp1].fa=tr[tmp2].fa=now;
            }else{
                if(tr[pre].sz==1){
                    if(tr[now].dir==1){
                        tr[pre].ch[1]=0;
                        tr[pre].ch[2]=tmp1;
                        tr[pre].ch[3]=tmp2;
                    }else{
                        tr[pre].ch[3]=tr[pre].ch[1];
                        tr[pre].ch[2]=0;
                        tr[pre].ch[0]=tmp1;
                        tr[pre].ch[1]=tmp2;
                    }
                }else{
                    if(tr[now].num[1]<tr[pre].num[0]){
                        tr[pre].ch[0]=tmp1;
                        tr[pre].ch[2]=connect(tr[pre].ch[1],tr[pre].ch[2]);
                        tr[pre].ch[1]=tmp2;
                    }else if(tr[now].num[1]>tr[pre].num[1]){
                        tr[pre].ch[3]=tmp2;
                        tr[pre].ch[1]=connect(tr[pre].ch[1],tr[pre].ch[2]);
                        tr[pre].ch[2]=tmp1;
                    }else{
                        if(tr[now].dir==1){
                            tr[pre].ch[1]=tmp1;tr[pre].ch[2]=tmp2;
                        }else{
                            tr[pre].ch[1]=tmp1;tr[pre].ch[2]=tmp2;
                        }
                    }
                }
                tr[pre].num[tr[pre].sz++]=tr[now].num[1];
                getorder(tr[pre]);
                now=pre;
            }
        }
        if(!lson&&!mlson&&!mrson&&!rson){//叶子结点
                tr[now].num[tr[now].sz++]=x;
                getorder(tr[now]);
                break;
        }else{
            int i=0;
            if(tr[now].num[0]>x) now=tr[now].ch[0];
            else if(tr[now].sz==2){
                if(tr[now].num[1]<x) now=tr[now].ch[3];
                else now=tr[now].ch[1]?tr[now].ch[1]:tr[now].ch[2];
            }else if(tr[now].sz==1) now=tr[now].ch[1];
            else{
                if(x>tr[now].num[2]) now=tr[now].ch[3];
                else if(tr[now].num[0]<x&&x<tr[now].num[1]) now=tr[now].ch[1];
                else now=tr[now].ch[2];
            }
        }
    }
}
int main() {
    scanf("%d", &t);
    while(t--){
        cnt=1;
        tr[1]={0,{0,0,0},{0,0,0,0},0,0};
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            int x;
            scanf("%d",&x);
            add(x);
        }
        printf("Case #%d:\n",++ca);
        dfs(1);
    }
    return 0;
}