0
在解决此问题中:http://www.cs.duke.edu/csed/newapt/drawtree.html我编写了下面的代码,但它似乎运行速度太慢。有没有更快的方法检查所有的孩子节点W/O使用FOR循环?队列会有帮助吗?如何提高递归函数的运行时间
public class DrawTree {
HashMap<String, ArrayList<String>> map =
new HashMap<String, ArrayList<String>>();
ArrayList<String> drawing = new ArrayList<String>();
String root;
public String[] draw(int[] parents, String[] names) {
for(int x=0; x<parents.length; x++)
{
int parentindex = parents[x];
//root name
if(parentindex==-1)
{
root=names[x];
if(!map.containsKey(names[x]))
{
map.put(names[x], new ArrayList<String>());
}
continue;
}
//add parent, child to map
if (!map.containsKey(names[parentindex]))
map.put(names[parentindex],
new ArrayList<String>());
map.get(names[parentindex]).add(names[x]);
}
sketch("",root,false);
return drawing.toArray(new String[drawing.size()]);
}
//***IMPROVE RUN TIME - different algorithm??***
//method takes root and prefix?
public void sketch(String parent, String child, boolean addPipe){
StringBuilder toAdd = new StringBuilder();
//don't need to add connector pipe
if(!addPipe)
{
//number of spaces to add to prefix
int spaces = parent.indexOf('-')+1;
//add spaces to prefix
while(spaces>0)
{
toAdd.append(" ");
spaces--;
}
toAdd.append("+-"+child);
}
//index of pipe in parent, -1 if parent doesn't have pipe
int parentPipe = parent.indexOf('|');
//need to add connector pipe & parent has pipe
// (is a child of a subtree)
if(parentPipe>0)
{
//number of spaces to add to prefix
int spaces = parent.indexOf('-')+1;
//add spaces to prefix
while(spaces>0)
{
if(spaces==parentPipe) toAdd.append('|');
else toAdd.append(" ");
spaces--;
}
toAdd.append("+-"+child);
}
//need to add pipe and parent doesn't have pipe
if(addPipe && parentPipe<0)
{
int spaces = parent.indexOf('-')+1;
while(spaces>0)
{
if(spaces==2) toAdd.append('|');
else toAdd.append(" ");
}
toAdd.append("+-"+child);
}
//add child to list of tree drawing
String node = toAdd.toString();
drawing.add(node);
//System.out.println(node);
//loop through list of children, passing each recursively
//...count level?
if(map.containsKey(child))
{
//System.out.println("map works");
for(int x = 0; x<map.get(child).size(); x++)
{
boolean pipe = false;
if(x<(map.get(child).size()-1)) pipe=true;
//System.out.println(map.get(child).get(x));
sketch(node, map.get(child).get(x), pipe);
}
}
}
此问题更适合[codereview.se](http://codereview.stackexchange.com/)。 – cheeken
定义为“慢”的基础是什么?你有没有特别的时机? – kosa
map.containsKey(key)有一个改进。它几乎与map.get(key)一样。调用containsKey并逐个获取是没有意义的。更好地使用'List list = map.get(key); if(list == null){list = new ArrayList (); map.put(key,list);} list.add(something);' –