解决链表相关的算法题时,在循环的判断语句中,选择pNode==nullptr 与 pNode->next==nullptr的问题上举棋不定,处理起来很不顺心。
所以,记录下 选择pNode==nullptr 还是 pNode->next==nullptr 的依据
下面以一道算法题展开讲解,题目为 从尾到头打印链表
链表节点定义如下:
struct ListNode
{
int m_nKey;
ListNode * m_pNext;
}
思路为:从头到尾遍历每一个节点,将每一个节点压入栈中,然后,依次打印栈顶、弹出栈顶元素,直到栈为空。
这里就涉及到遍历每一个节点,在循环的判断语句中 选择pNode==nullptr 还是 pNode->next==nullptr 的问题就来了
选择依据其实很简单,就是最后的pNode节点是否需要在循环体中被处理
在本例子中,利用循环体处理压栈操作,显然,最后一个节点是需要被压栈处理的,所以选择pNode==nullptr
实现代码如下:
void PrintListReversingly_Iteratively(ListNode * pHead)
{
std::stack<ListNode*> nodes;
ListNode * pNode = pHead;
while(pNode != nullptr){
nodes.push(pNode);
pNode = pNode->m_pNext;
}
while(!nodes.empty()){
pNode = nodes.top();
cout<<pNode->m_nKey<<" ";
nodes.pop();
}
}
再举例,如何获取指向链表最后一个节点的指针?
显然也是利用循环体,在循环体中切换到下一节点,我们不想到最后一个节点时还切换到下一个节点,所以pNode->next==nullptr
ListNode * pNode = pHead;
while(pNode->next==nullptr){
pNode = pNode->m_pNext;
}
这样,在退出循环体后,pNode指向链表的最后一个节点
结论:
在遍历链表时,若最后一个节点需要在循环体中被处理,选择pNode==nullptr
若最后一个节点不需要在循环体中被处理,选择pNode->next==nullptr
而无论选择哪一种,由于是遍历,循环体中都是采用pNode = pNode->m_pNext;