Hướng giải của Xóa phần tử thứ N từ cuối


Nhớ rằng hướng dẫn giải này chỉ nên sử dụng khi bế tắc, và tuyệt đối không nên sao chép mã nguồn kèm theo. Hãy tôn trọng tác giả bài tập và người viết hướng dẫn giải.
Nộp mã nguồn lời giải chính thức trước khi giải bài tập đó có thể khiến bạn bị ban.
Lời giải

Dùng hai con trỏ: cho first đi trước second đúng K bước, sau đó di chuyển cả hai cho đến khi first kết thúc. Khi đó second ở node trước node cần xóa.

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

struct Node {
    int data;
    Node* next;
    Node(int val) : data(val), next(nullptr) {}
};

int main() {
    int n, k; cin >> n >> k;
    Node *head = nullptr, *tail = nullptr;
    for (int i = 0; i < n; ++i) {
        int x; cin >> x;
        Node* node = new Node(x);
        if (!head) head = tail = node;
        else tail->next = node, tail = node;
    }
    Node dummy(0);
    dummy.next = head;
    Node *first = &dummy, *second = &dummy;
    for (int i = 0; i <= k; ++i) first = first->next;
    while (first) {
        first = first->next;
        second = second->next;
    }
    Node* to_del = second->next;
    second->next = second->next->next;
    delete to_del;
    for (Node* p = dummy.next; p; p = p->next) {
        cout << p->data;
        if (p->next) cout << " -> ";
    }
}
class Node:
    def __init__(self, val):
        self.data = val
        self.next = None

n, k = map(int, input().split())
a = list(map(int, input().split()))
head = tail = None
for val in a:
    node = Node(val)
    if head is None: head = tail = node
    else: tail.next = node; tail = node

dummy = Node(0)
dummy.next = head
first = second = dummy
for _ in range(k + 1):
    first = first.next
while first:
    first = first.next
    second = second.next
second.next = second.next.next

p = dummy.next
while p:
    print(p.data, end='')
    if p.next:
        print(' -> ', end='')
    p = p.next

Nhận xét

Không có ý kiến tại thời điểm này.