面试100题之17

题目：输入一个链表的头结点，反转该链表，并返回反转后链表的头结点。链表结点定义如下：
struct ListNode
{ int m_nKey;ListNode* m_pNext; };

分析：通常在单链表的首元结点（存放第一个数据元素的结点）之前附设一个头结点（数据域什么都不放），称之为带头结点的单链表，反之就是不带头结点的单链表。如果是带头结点的链表反转后头结点应该还放在最开始。

带头结点的单链表反转：

#include<stdafx.h>
#include<iostream>
#include <deque>
using namespace std;
struct ListNode
{ int m_nKey;
ListNode* m_pNext;
};

void InvertList(ListNode *head)
{
    ListNode* pre=NULL;
    ListNode* p=head->m_pNext;
    ListNode* pnext;
    if(p==NULL)
        return;
    while(p!=NULL)
    {
        pnext=p->m_pNext;
        p->m_pNext=pre;
        pre=p;
        p=pnext;
    }
    head->m_pNext=pre;
}
void main()
{
    ListNode a[5];
    ListNode head;//头结点
    a[0].m_nKey=0;
    a[1].m_nKey=1;
    a[2].m_nKey=2;
    a[3].m_nKey=3;
    a[4].m_nKey=4;
    a[0].m_pNext=&a[1];
    a[1].m_pNext=&a[2];
    a[2].m_pNext=&a[3];
    a[3].m_pNext=&a[4];
    a[4].m_pNext=NULL;
    head.m_pNext=&a[0];
    ListNode *tmp=head.m_pNext;
    for(int i=0;i<5;i++)
    {
        cout<<tmp->m_nKey<<endl;
        tmp=tmp->m_pNext;
    }
    InvertList(&head);
    tmp=head.m_pNext;
    for(int i=0;i<5;i++)
    {
        cout<<tmp->m_nKey<<endl;
        tmp=tmp->m_pNext;
    }
}

不带头结点的单链表反转

#include<stdafx.h>
#include<iostream>
#include <deque>
using namespace std;
struct ListNode
{ int m_nKey;
ListNode* m_pNext;
};

ListNode * InvertList(ListNode *head)
{
    ListNode* pre=NULL;
    ListNode* p=head;
    ListNode* pnext;
    if(p->m_pNext==NULL)
        return p;
    while(p!=NULL)
    {
        pnext=p->m_pNext;
        p->m_pNext=pre;
        pre=p;
        p=pnext;
    }
    return pre;
}
void main()
{
    ListNode a[5];
    a[0].m_nKey=0;
    a[1].m_nKey=1;
    a[2].m_nKey=2;
    a[3].m_nKey=3;
    a[4].m_nKey=4;
    a[0].m_pNext=&a[1];
    a[1].m_pNext=&a[2];
    a[2].m_pNext=&a[3];
    a[3].m_pNext=&a[4];
    a[4].m_pNext=NULL;
    ListNode *tmp=&a[0];
    for(int i=0;i<5;i++)
    {
        cout<<tmp->m_nKey<<endl;
        tmp=tmp->m_pNext;
    }
    ListNode *newhead=InvertList(&a[0]);
    tmp=newhead;
    for(int i=0;i<5;i++)
    {
        cout<<tmp->m_nKey<<endl;
        tmp=tmp->m_pNext;
    }
}