LeetCode - 19. Remove Nth Node From End of List

19. Remove Nth Node From End of List #

Problem's Link
#

 ----------------------------------------------------------------------------

Mean: 

给定一个链表，删除倒数第n个结点.

analyse:

简单粗暴 :)

Time complexity: O(N)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-02-17-13.51
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long(LL);
typedef unsigned long long(ULL);
const double eps(1e-8);

struct ListNode
{
   int val;
   ListNode *next;
   ListNode(int x) : val(x), next(NULL) {}
};

class Solution
{
public:
   ListNode* removeNthFromEnd(ListNode* head, int n)
   {
       vector<int> list;
       while(head)
       {
           list.push_back(head->val);
           head=head->next;
       }
       int si=list.size();
       if(si<=1) return head;
       // calculate the idx
       int idx=si-n;
       ListNode* ret;
       ListNode* tmp=ret;
       bool isFirst=true;
       for(int i=0; i<si; ++i)
       {
           if(i==idx)
               continue;
           if(isFirst)
           {
               tmp=new ListNode(list[i]);
               ret=tmp;
               isFirst=false;
           }
           else
           {
               tmp->next=new ListNode(list[i]);
               tmp=tmp->next;
           }
       }
       return ret;
   }
};

int main()
{
   Solution solution;
   int n,num,tmp;
   while(cin>>n>>num)
   {
       ListNode* head=NULL,*root=NULL;
       for(int i=0; i<n; ++i)
       {
           cin>>tmp;
           if(i==0)
           {
               head=new ListNode(tmp);
               root=head;
           }
           else
           {
               head->next=new ListNode(tmp);
               head=head->next;
           }
       }

       ListNode* ans=solution.removeNthFromEnd(root,num);
       while(ans)
       {
           cout<<ans->val<<" ";
           ans=ans->next;
       }
       cout<<endl;
       cout<<"End."<<endl;
   }
   return 0;
}
/*

*/