题目:
小A有一本生命进化书,以一个树形结构记载了所有生物的演化过程。为了探索和记录其中的演化规律,小A提出了一种方法,可以以字符串的形式将其复刻下来,规则如下:初始只有一个根节点,表示演化的起点,依次记录 01 字符串中的字符,如果记录 0,则在当前节点下添加一个子节点,并将指针指向新添加的子节点;如果记录 1,则将指针回退到当前节点的父节点处。现在需要应用上述的记录方法,复刻下它的演化过程。请返回能够复刻演化过程的字符串中, 字典序最小的 01 字符串;
注意:节点指针最终可以停在任何节点上,不一定要回到根节点。
输入描述:
parents[] 数组,其中,parents[i] 表示编号 i 节点的父节点编号(根节点的父节点为 -1)。
输出描述:
返回能够复刻演化过程的字符串中, 字典序最小 的 01 字符串
输入样例:
输入样例1
<br/>[-1,0,0,2]<br/>
输入样例2
<br/>[-1,0,0,1,2,2]<br/>
输出样例:
输出样例1
<br/>00110<br/>
解释:共存在 2 种记录方案:
第 1 种方案为:0(记录编号 1 的节点) -> 1(回退至节点 0) -> 0(记录编号 2 的节点) -> 0((记录编号 3 的节点))
第 2 种方案为:0(记录编号 2 的节点) -> 0(记录编号 3 的节点) -> 1(回退至节点 2) -> 1(回退至节点 0) -> 0(记录编号 1 的节点)
返回字典序更小的 "00110"
输出样例2
<br/>00101100<br/>
运行代码:
import java.util.*;
class Main{
public static void main(String[] args){
Scanner input = new Scanner(System.in);
String str = input.next().toString();
String sr = str.substring(1,str.length()-1);
String[] arr = sr.split(",");
int[] b = new int[arr.length];
for(int j = 0; j<b.length;j++){
b[j] = Integer.parseInt(arr[j]);
}
Main ma = new Main();
System.out.println(ma.evolutionaryRecord(b));
}
HashMap<Integer,List<Integer>> child = new HashMap<>();
public String evolutionaryRecord(int[] parents){
int n = parents.length;
if (n==1)
return "";
for(int i=0 ;i<n;i++){
child.put(i,new ArrayList<>());
if(parents[i]==-1)
continue;
child.get(parents[i]).add(i);
}
StringBuilder sb = new StringBuilder(dfs(0));
int end = 0;
for(int i = sb.length()-1;i>0;i--){
if(sb.charAt(i)!='1'){
end = i;
break;
}
}
return sb.substring(0,end+1);
}
private String dfs(int i){
StringBuilder t = new StringBuilder();
List<String> path = new ArrayList<>();
for(int x:child.get(i)){
path.add("0"+dfs(x)+"1");
}
Collections.sort(path);
for(String s:path){
t.append(s);
}
return t.toString();
}
}