【14】链表中倒数第k个结点

【14】链表中倒数第k个结点

时间限制：1秒

空间限制：32768K

本题知识点： 链表

题目描述

输入一个链表，输出该链表中倒数第k个结点

牛客网题目链接：点击这里题目

vs2010全部代码

#include<iostream>
using namespace std;

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* P=pListHead;
ListNode* End=pListHead;
int count=0;
while(End)
{
End=End->next;
count++;
if(count>k)
{
P=P->next;
}
}
if(count<k) return NULL;
else return P;
}
};
//测试用例:6,{1,2,3,4,5}
//对应输出应该为:{}

int main()
{
Solution S1;
ListNode* L1=NULL;
ListNode* PL=NULL;
ListNode* s=NULL;
//创建链表
int n;
cin>>n;
int tmp;
cout<<"输入第"<<1<<"个结点值";
cin>>tmp;
s=new ListNode(tmp);
PL=s; L1=PL;
for(int i=1; i<n; i++)
{
int tmp;
cout<<"输入第"<<i+1<<"个结点值";
cin>>tmp;
s=new ListNode(tmp);
PL->next=s;
PL=PL->next;
}
//输出链表
cout<<endl;
PL=L1;
for(int i=0; i<n; i++)
{
cout<<PL->val<<"->";
PL=PL->next;
}

cout<<endl;
cout<<S1.FindKthToTail(L1,5)->val<<endl;

}

改进的代码：

/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};*/
class Solution {
public:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
if(!pListHead || k==0) return NULL;

ListNode* p1=pListHead;k--;
ListNode* p2=NULL;
while(k--){
p1=p1->next;
if(!p1) return NULL;
}
p2=pListHead;
while(p1 && p1->next){
p1=p1->next;
p2=p2->next;
}
return p2;
}
};

考虑三种情况：

1. 输入的pLIstHead为空指针

2. 输入的pListHead为头结点的链表的结点总数少于k

3. 输入的参数k为0.