首页 > 代码库 > dfs手写栈模板

dfs手写栈模板

在竞赛中如果系统栈很小的话,过深的递归会让栈溢出,这个时候我们就要自己手写栈,将递归转化成手工栈。 
方法其实也很简单。 
基本思路上,我们就是用栈不断的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);    }}

dfs手写栈模板