问题1:找到链表中倒数第 k 个元素。
由于链表无法得知其长度,一般的想法是首先遍历一遍得到长度len,然后再从头走len-k;就得到了倒数第k个元素。效率太慢。有一种比较好的方法。设2个指针pFirst,pSecond,首先让pFirst走k步。
让两个指针有 k 步的遍历步距差值,这样当第一个指针指向最后时,第二个指针正好指向倒数第 k 个。其实遍历的次数和算法一相同。网上有一些文章误认为这种方式更高效,其实没有太本质的变化,只有常数的变化。
具体来说,就是先让 ptrFirst 遍历了 k 个节点后,两个指针再开始一起遍历节点。
问题2:找到链表的中间元素。
与问题1类似,让pFirst以两部速度遍历,当pFirst到尾时,pSecond指向的就是中间元素。该题与判断链表是否有环类似。
iNode * GetMiddleNode ( iNode *head ) { iNode *p1 = head; iNode *p2 = p1; while( p2 ) { p2 = p2->next; if(p2) { p2 = p2->next; p1=p1->next; } } return p1; }
该思想还可以扩展成如何找到倒数1/k的元素。