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;
}