在竞赛中如果系统栈很小的话,过深的递归会让栈溢出,这个时候我们就要自己手写栈,将递归转化成手工栈。
方法其实也很简单。
基本思路上,我们就是用栈不断的pop,push。但是何时push,何时pop呢?
在《算法导论》上对深度优先遍历树的讲解中,在深度遍历中,会对每个节点进行染色,白色为没有被访问过;灰色为被访问过,但是该节点的所有子树还没有完成访问;黑色,节点被访问过,而且该节点的所有子树都被完全的访问。
所以,我们就通过颜色标记来进行判断了。
整体的框架如下:
memset(vis,0,sizeof(vis));
stack S;
S.push(root)
while(!S.empty()){
int u = S.top();
if(vis[u] == 1)// if node is gray, then color black
{
vis[u] = 2;
// do things after dfs children.
S.pop();
}
else if(vis[u] == 0)// if node is white, then color gray
{
vis[u] = 1;
// do things before dfs children.
for all children v
S.push(v);
}
}