leetcode链表--9、reverse-linked-list-ii（翻转区间的链表）

题目描述

 

Reverse a linked list from position m to n. Do it in-place and in one-pass.

For example:Given1->2->3->4->5->NULL, m = 2 and n = 4,return1->4->3->2->5->NULL.

Note:

Given m, n satisfy the following condition:1 ≤ m ≤ n ≤ length of list.

 

解题思路：

1）找到要翻转区间的第一个结点并记录该结点的前一个结点pre

2）找到要翻转区间的最后一个结点

3）翻转区间的链表，并保证后续的链表正确连接在链上

4）将翻转的第一个结点的前一结点与链表连接（分为两种情况，一种pre为空则返回翻转后链表，一种不为空，则连接，然后返回head）

#include <iostream>
#include <malloc.h>
using namespace std;

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};
/*在链表的末端插入新的节点，建立链表*/
ListNode *CreateList(int n)
{
    ListNode *head;
    ListNode *p,*pre;
    int i;
    head=(ListNode *)malloc(sizeof(ListNode));
    head->next=NULL;
    pre=head;
    for(i=1;i<=n;i++)
    {
        p=(ListNode *)malloc(sizeof(ListNode));
        cin>>p->val;
        pre->next=p;

        pre=p;
    }
    p->next=NULL;

    return head->next;//不带空的头结点
}
/*-------------------------输出链表-----------------------------------*/
void PrintList(ListNode *h)
{
    ListNode *p;

    p=h;//不带空的头结点
    while(p)
    {
        cout<<p->val<<" ";
        p=p->next;
        cout<<endl;
    }
}
class Solution {
public:
    //例 1->2->3->4->5->6->NULL   m=2  n=5
    ListNode *reverseBetween(ListNode *head, int m, int n) {
        if(head == NULL)
            return NULL;

        ListNode *pNode = head;
        ListNode *pre =NULL;

        for(int i=1;i<m;i++)
        {
            pre = pNode;
            pNode = pNode->next;
        }
        ListNode *last = pNode;
        for(int i=m;i<n;i++)
        {
            last = last->next;
        }
        //翻转pNode last之间的结点
        for(int i=0;i<n-m;i++)
        {
            ListNode *temp = pNode->next;
            pNode->next = last->next;
            last->next = pNode;

            pNode = temp;
        }
        if(pre == NULL)
            return last;
        else
        {
            pre->next = last;
            return head;
        }
    }


};
int main()
{
    int n1;
    ListNode *h;
    cout<<"输入链表1的结点数目"<<endl;
    cin>>n1;
    h = CreateList(n1);
    cout<<"链表1为："<<endl;
    PrintList(h);
    Solution s;
    s.reverseBetween(h,2,5);
    cout<<"翻转后："<<endl;
    PrintList(h);

}