Leetcode | Remove Nth Node From End of List

Given a linked list, remove the nth node from the end of list and return its head.

For example,

Given linked list: 1->2->3->4->5, and n = 2.

After removing the second node from the end, the linked list becomes 1->2->3->5.

Method I

没去想快慢指针的做法。 而是用了一个vector以便可以根据下标定位到具体的节点。

这里为了删除节点方便，用了指针的指针，ListNode**，这样直接修改当前位置为下一个位置就行了，不管它是不是head。

注意的是，在循环里面，临时变量p必须也是ListNode**的类型，不能是ListNode*，后者会造成&p得到的是临时地址；

class Solution {
public:
    ListNode *removeNthFromEnd(ListNode *head, int n) {
        if (head == NULL) return NULL;
        vector<ListNode**> nodes;
        ListNode** p = &head;
        while (*p != NULL) {
            nodes.push_back(p);
            p = &((*p)->next);
        }
        *nodes[nodes.size() - n] = (*nodes[nodes.size() - n])->next;
        return head;
    }
};

Method II

用两个指针，第一个指针先跑了n次。然后第二个再跑。这样当第一个指针跑到头时，第二个指针就跑到从后往前数第n个指针了。

class Solution {
public:
    ListNode *removeNthFromEnd(ListNode *head, int n) {
        if (head == NULL) return NULL;
        ListNode** p1 = &head, **p2 = &head;
        
        for (int i = 0; i < n; ++i) {
            if (!*p1) return NULL;
            p1 = &((*p1)->next);
        }
        
        while (*p1) {
            p2 = &((*p2)->next);
            p1 = &((*p1)->next);
        }
        *p2 = (*p2)->next;
        
        return head;
    }
};

果然这些链表类的题要一起做，不然都想不到好办法。。。