码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode

时间:2018-08-12 12:09:16      阅读:142      评论:0      收藏:0      [点我收藏+]

标签:nbsp   solution   int   ini   class   init   span   div   while   

链表题。很简单但是debug了非常久。

愚蠢如我...把所有地址存在数组里。空间复杂度...

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
        ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode *p=head,*q;
        ListNode *s[100000];
        int m=0;
        while(p){
            s[m]=p;m++;
            p=p->next;
        }
        if (m==n) return head->next;
        if(n==1) (*s[m-2]).next=NULL;
        else {
            p=s[m-n-1];
            q=p->next;
            (*p).next=(*q).next;
        }
        return head;
    }
};

 

非常巧妙的方法,保持两个指针差n个位置,使后面的指针指向最后一个,前面的指针即为倒数第n.

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

 

LeetCode

标签:nbsp   solution   int   ini   class   init   span   div   while   

原文地址:https://www.cnblogs.com/travelller/p/9462183.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!