在链表中删除倒数第K个节点

题目描述

给出一个单链表，返回删除单链表的倒数第 K 个节点的链表。

输入描述：

n 表示链表的长度。

val 表示链表中节点的值。

输出描述：

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

示例1

输入

5 4
1 2 3 4 5

输出

1 3 4 5

备注：

$1\le K\le n\le 10^6$

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

题解：

首先指针 p 从链表头节点开始，走到第 K 个结点停止，然后另一个指针 q 从头节点开始，p 继续遍历直到链表尾部，q 跟着移动，在 p 为链表最后一个节点时，q 就是要删除的倒数第 K 个结点。

代码：

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

struct list_node{
    int val;
    struct list_node * next;
}; //链表的节点

int K;

list_node * input_list(void) //读入链表
{
    int n, val;
    list_node * phead = new list_node();
    list_node * cur_pnode = phead;
    scanf("%d %d", &n, &K);
    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 * remove_last_kth_node(list_node * head, int K)
{
    //在下面完成代码
    int n = 1;
    list_node * p = head;
    while (p && n < K) {
        n += 1;
        p = p->next;
    }
    if (!p) return head;
    list_node * q = head;
    list_node * q_pre = head;
    while (p->next) {
        p = p->next;
        q_pre = q;
        q = q->next;
    }
    q_pre->next = q->next;
    delete q;
    return head;
}

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

int main ()
{
    list_node * head = input_list(); // 链表的头节点
    list_node * rhead = remove_last_kth_node(head, K);
    print_list(rhead);
    return 0;
}