仅验证了样例
这道题先根据他所说的建树(也可以不用,但我不建就不会做了…)。然后找到在给出的左右子树中没有出现的结点编号,这个就是根结点了。
之后我想用后序遍历做,但卡住了,因为发现如果有个结点左子树为空的话,那就和后序遍历不一样了,所以考虑用dfs做。不过用dfs写完后感觉,其实就是后序遍历的代码稍微改了一下,所以以前的思路其实没什么问题。包括和这道题很像的甲级1130,其实也是中序遍历的代码稍微改了一下。
那么既然是和后序遍历很像的dfs,那么就很容易搞了,首先如果左子树不空的话,就按照后序遍历来写。要是左子树是空的话,就先输出当前节点的字符串,然后再接着深搜遍历他的右子树就好了。最后在合适的地方放上括号就好。
#include <iostream>
#include <cstring>
using namespace std;
struct Node{
string data;
int id;
Node* left;
Node* right;
}node[25];
bool whichisroot[25];
void dfs(Node* root){
if(root==NULL){
return;
}
if(root->left!=NULL){
printf("(");
dfs(root->left);
dfs(root->right);
printf("%s",node[root->id].data.c_str());
printf(")");
}else{
printf("(");
printf("%s",node[root->id].data.c_str());
dfs(root->right);
printf(")");
}
}
int main(){
int n;
cin>>n;
memset(whichisroot,false,sizeof(whichisroot));
for(int i=1;i<=n;i++){
string data;
int l,r;
cin>>data>>l>>r;
node[i].data=data;
node[i].id=i;
if(l!=-1){
node[i].left=&node[l];
whichisroot[l]=true;
}else{
node[i].left=NULL;
}
if(r!=-1){
node[i].right=&node[r];
whichisroot[r]=true;
}else{
node[i].right=NULL;
}
}
int root;
for(int i=1;i<=n;i++){
if(whichisroot[i]==false){
root=i;
break;
}
}
dfs(&node[root]);
return 0;
}