剑指offer面试题17——合并两个排序的链表

  1 #include<iostream>
  2 #include<vector>
  3 using namespace std;
  4 
  5 //链表的数据结构
  6 struct ListNode
  7 {
  8     int val;
  9     ListNode* next;
 10     ListNode(int x):val(x),next(NULL){}
 11 };
 12 
 13 //创建一个单链表
 14 ListNode* product(vector<int>& vec,int n)
 15 {
 16     if(n==0)
 17         return NULL;
 18     ListNode* ptr1=new ListNode(vec[0]);
 19     int i=1;
 20     ListNode* ptr2=ptr1;
 21     while(i<n)
 22     {
 23         ptr2->next=new ListNode(vec[i]);
 24         ptr2=ptr2->next;
 25         i++;
 26     }
 27     return ptr1;
 28 }
 29 
 30 //单链表每个结点的输出
 31 void cout_node(ListNode* root)
 32 {
 33     if(root==NULL)
 34         return;
 35     ListNode* ptr1=root;
 36     while(ptr1!=NULL)
 37     {
 38         cout<<ptr1->val<<‘ ‘;
 39         ptr1=ptr1->next;
 40     }
 41     cout<<endl;
 42     return;
 43 }
 44 
 45 ListNode* merge(ListNode* root1,ListNode* root2)
 46 {
 47     ListNode* root=NULL;
 48     if(root1==NULL&&root2==NULL)//两个链表均为空
 49         return root;
 50     if(root1==NULL)//root2为空
 51         return root2;
 52     if(root2==NULL)//root1为空
 53         return root1;
 54     ListNode* ptr1=root1;
 55     ListNode* ptr2=root2;
 56 
 57     //在确定两个链表都不为空下，将其中小的一个结点给root
 58     if(root1->val<root2->val)
 59     {
 60         root=root1;
 61         ptr1=ptr1->next;
 62     }
 63     if(root2->val<root1->val)
 64     {
 65         root=root2;
 66         ptr2=ptr2->next;
 67     }
 68     if(root1->val==root2->val)
 69     {
 70         root=root1;
 71         ptr1=ptr1->next;
 72         ptr2=ptr2->next;
 73     }
 74     ListNode* ptr0=root;
 75     ptr0->next=NULL;
 76 
 77     //开始循环
 78     while(1)
 79     {
 80         //若其中一个链表已经到了末尾就截止
 81         if(ptr1==NULL)
 82         {
 83             ptr0->next=ptr2;
 84             return root;
 85         }
 86         if(ptr2==NULL)
 87         {
 88             ptr0->next=ptr1;
 89             return root;
 90         }
 91         //否则将小的那个给root链表，将指针都往后移动
 92         if(ptr1->val<ptr2->val)
 93         {
 94             ptr0->next=ptr1;
 95             ptr1=ptr1->next;
 96             ptr0=ptr0->next;
 97             ptr0->next=NULL;
 98         }
 99         else if(ptr1->val>ptr2->val)
100         {
101             ptr0->next=ptr2;
102             ptr2=ptr2->next;
103             ptr0=ptr0->next;
104             ptr0->next=NULL;
105         }
106         else
107         {
108             ptr0->next=ptr1;
109             ptr1=ptr1->next;
110             ptr2=ptr2->next;
111             ptr0=ptr0->next;
112             ptr0->next=NULL;
113         }
114     }
115     return root;
116 }
117 
118 int main()
119 {
120     int ary1[10]={1,3,5,7,9};
121     vector<int> vec1(ary1,ary1+5);
122     ListNode* root1=product(vec1,5);
123     cout_node(root1);
124 
125     int ary2[10]={2,4};
126     vector<int> vec2(ary2,ary2+2);
127     ListNode* root2=product(vec2,2);
128     cout_node(root2);
129 
130     ListNode* root=merge(root1,root2);
131     cout_node(root);
132 
133     system("pause");
134 }