牛客-鏈表中倒數第k個節點

牛客-鏈表中倒數第k個節點

思路1:

• 設鏈表的長度爲len
• 倒數第k個就是正數第len-k+1個

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/

ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
         if (pListHead==NULL) {
            return NULL;
        }
        int len=0;
        ListNode *p = pListHead;
        while(p!=NULL) {
            len++;
            p=p->next;
        }
        if(k>len) {
            return NULL;
        }
        int i=len-k;
        ListNode *q = pListHead;
        while(i>0) {
            i--;
            q=q->next;
        }

        return q;
    }

思路2:

• 讓p指針先走k-1步
• 然後q和p指針同步走，知道p走到盡頭，那麼q指向的就是倒數第k個節點

 ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
         if (pListHead==NULL || k<=0) {
            return NULL;
        }

        ListNode *p = pListHead;
        ListNode *q = pListHead;
        
        while((k-1)>0) {
            k--;
            if(p->next == NULL) {
                return NULL;
            }
            p=p->next;
        }

        while(p->next!=NULL) {
            p=p->next;
            q=q->next;
        }
        return q;
    }