1. 程式人生 > >劍指offer(14)連結串列中倒數第k個結點

劍指offer(14)連結串列中倒數第k個結點

題目描述
輸入一個連結串列,輸出該連結串列中倒數第k個結點。
解題思路
輸出的是 倒數第k個連結串列,可以參考Leetcode刪除連結串列倒數節點的那個演算法題
一共有兩種解法,一是兩遍遍歷的,二是一遍遍歷的。
兩次遍歷的思路就是先遍歷一遍 求出連結串列長度,然後在遍歷,輸出N-k+1的節點。這道題靠魯棒性,需要考慮各種情況,比如可能不存在倒數第k個結點。
程式碼如下

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        ListNode *res=new ListNode(0);
        ListNode *first=pListHead;
        int length=0;
        while(first!=NULL)
        {
            first=first->next;
            length++;
        }
        length-=k;
        if(length<0) return 0;
        first=pListHead;
        while(length>0)
        {
            length--;
            first=first->next;
        }
        res=first;
        return res;
    }
};

一次便利的結果,就是宣告兩個指標,兩個指標都指向頭結點,第一個指標向後移動k+1位,然後,兩個指標同步向後執行,這個時候,兩個指標之間的距離就是k,直至投一個指標挪到最後一位,此時,輸出第二個指標的結點即可。
程式碼如下

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
    class Solution {
    public:
        ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
            ListNode *res=new ListNode(0);
            ListNode *first=pListHead;
            ListNode *second=pListHead;
            while(k>0)
            {
                if(first!=NULL) 
                    first=first->next;
                else
                    return 0;
                k--;
            }
            while(first!=NULL)
            {
                first=first->next;
                second=second->next;
            }
            res=second;
            return second;
        }
    };