反转部分单向链表

反转部分单向链表

题目描述

给定一个单链表，在链表中把第 L 个节点到第 R 个节点这一部分进行反转。

输入描述：

n 表示单链表的长度。

val 表示单链表各个节点的值。

L 表示翻转区间的左端点。

R 表示翻转区间的右端点。

输出描述：

在给定的函数中返回指定链表的头指针。

示例1

输入

5
1 2 3 4 5
1 3

输出

3 2 1 4 5

备注：

$1\le n\le 10^6$

$1\le L\le R\le n$

$-10^6\le val\le 10^6$

题解：

这题还是有点麻烦的，我们可以找到第 L 个节点的前驱节点和第 R 个节点的后继节点，正常反转第 L 个节点到第 R 个节点，然后将第 L 个节点的 next 指针指向第 R 个节点的后继节点，将第 L 个节点的前驱节点的 next 指针指向第 R 个节点，就行了。

代码：

# include <bits/stdc++.h>
using namespace std;

struct list_node{
    int val;
    struct list_node * next;
};

list_node * input_list(void)
{
    int n, val;
    list_node * phead = new list_node();
    list_node * cur_pnode = phead;
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &val);
        if (i == 1) {
            cur_pnode->val = val;
            cur_pnode->next = NULL;
        }
        else {
            list_node * new_pnode = new list_node();
            new_pnode->val = val;
            new_pnode->next = NULL;
            cur_pnode->next = new_pnode;
            cur_pnode = new_pnode;
        }
    }
    return phead;
}


list_node * reverse_list(list_node * head, int L, int R)
{
    //在下面完成代码
    if (!head || L < 1) return head;
    
    list_node *cur = head;
    list_node *l = NULL, *l_pre = NULL;
    list_node *r_next = NULL;
    
    int len = 0;
    while (cur) {
        ++len;
        if (len == L - 1) l_pre = cur;
        if (len == R + 1) r_next = cur;
        cur = cur->next;
    }
    
    l = l_pre ? l_pre->next : head;
    
    list_node *tmp_pre= r_next;
    list_node *tmp = NULL;
    
    while (l != r_next) {
        tmp = l->next;
        l->next = tmp_pre;
        tmp_pre = l;
        l = tmp;
    }
    
    if (l_pre) {
        l_pre->next = tmp_pre;
        return head;
    }
    return tmp_pre;
}

void print_list(list_node * head)
{
    while (head != NULL) {
        printf("%d ", head->val);
        head = head->next;
    }
    puts("");
}


int main ()
{
    int L, R;
    list_node * head = input_list();
    scanf("%d%d", &L, &R);
    list_node * new_head = reverse_list(head, L, R);
    print_list(new_head);
    return 0;
}