从二叉树一个节点到另一个节点每一步的方向
给你一棵 二叉树 的根节点 root ,这棵二叉树总共有 n 个节点。每个节点的值为 1 到 n 中的一个整数,且互不相同。给你一个整数 startValue ,表示起点节点 s 的值,和另一个不同的整数 destValue ,表示终点节点 t 的值。
请找到从节点 s 到节点 t 的 最短路径 ,并以字符串的形式返回每一步的方向。每一步用 大写 字母 ‘L’ ,‘R’ 和 ‘U’ 分别表示一种方向:
'L' 表示从一个节点前往它的 左孩子 节点。
'R' 表示从一个节点前往它的 右孩子 节点。
'U' 表示从一个节点前往它的 父 节点。
请你返回从 s 到 t 最短路径 每一步的方向。
示例 1:
输入:root = [5,1,2,3,null,6,4], startValue = 3, destValue = 6
输出:“UURL”
解释:最短路径为:3 → 1 → 5 → 2 → 6 。
示例 2:
输入:root = [2,1], startValue = 2, destValue = 1
输出:“L”
解释:最短路径为:2 → 1 。
这题思路呢,是我们先找到那个节点,我们需要找的那个连接两个初始节点和最终节点的父节点,然后求解路径就ok了
我的代码呢,其实也可以求解连接两个节点的最近祖先节点,这点还是很有意思的,同时,也可以顺势改一下,求解那个节点的高度也可以
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int find_r(struct TreeNode* root, int startValue, int destValue,struct TreeNode** p){
int a,b;
if(root){
// printf("%d ",root->val);
if(root->val==startValue||root->val==destValue){
a=find_r(root->left,startValue,destValue,p);
b=find_r(root->right,startValue,destValue,p);
if((b==1||a==1)&&(root->val==startValue||root->val==destValue)){
*p=root;
}
}
if(root->val==startValue||root->val==destValue){
return 1;
}
a=find_r(root->left,startValue,destValue,p);
b=find_r(root->right,startValue,destValue,p);
printf(" a b %d %d val %d",a,b,root->val);
if((b==1||a==1)&&(root->val==startValue||root->val==destValue)){
*p=root;
}
if(a==1&&b==1){
*p=root;
// printf("p %d ",(*p)->val);
}
if(a==1||b==1){
return 1;
}
}
return 0;
}
void dfs(struct TreeNode* root,int val,char *s,int p,char f,int *find){
if(root&&(*find)!=1){
// printf("%d ",root->val);
if(f=='l'){
s[p]='L';
}
if(f=='r'){
s[p]='R';
}
if(root->val==val){
*find=1;
s[p+1]='\0';
}
else{
dfs(root->right,val,s,p+1,'r',find);
dfs(root->left,val,s,p+1,'l',find);
}
}
}
char * getDirections(struct TreeNode* root, int startValue, int destValue){
struct TreeNode *p;
char *s=(char *)malloc(sizeof(char)*1000000);
char *s1=(char *)malloc(sizeof(char)*1000000);
s[0]='\0';
s1[0]='\0';
int * find=(int *)malloc(sizeof(int));
int i;
if(root->val==startValue||root->val==destValue){
p=root;
}
else{
int a=find_r(root,startValue,destValue,&p);
}
printf("root %d ",p->val);
dfs(p,startValue,s,0,'l',find);
*find=0;
dfs(p,destValue,s1,0,'l',find);
int len=strlen(s);
printf("%s %s",s,s1);
for(i=0;i<len-1;i++){
s[i]='U';
}
s[len-1]='\0';
strcat(s,s1+1);
return s;
//printf("%s",s1);
}