慢慢把在leetcode上寫的代碼搬過來(不知道leetcode是啥的童鞋請點擊http://leetcode.com/),由於對於題目的理解或者coding技巧隨時變化,所以,這一系列的博客可能會被老是修改 = =!增刪都會比較頻繁,見諒。另外題目分類本來就很難,排除掉我漏分錯分的情況,很多題目本身就屬於幾個類別(比如可能它在排序類,又在遞歸類,還在查找類等等等等),所以一個你覺得是A類的,在A類裏面沒有找到的話~可以試試其他。
題目描述很簡單:Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
也是很經典的題了,一般都是用一個大小爲K的最小堆來維護,時空複雜度O(nlogk)+O(k) (空間複雜度我一般去掉輸入輸出必須的空間,因爲這部分空間無論你怎麼優化都沒法避免的)。
code:
//優先隊列的使用
class Solution {
public:
struct cmp{
bool operator() (ListNode* x,ListNode* y){
return x->val>y->val; //最小堆
}
};
ListNode *mergeKLists(vector<ListNode *> &lists) {
priority_queue<ListNode*,vector<ListNode*>,cmp> pq;
ListNode* head = NULL,**headP = &head;
for(int i=0;i<lists.size();i++)
if(lists[i])
pq.push(lists[i]);
while(!pq.empty()){
ListNode* tmp = pq.top();
pq.pop();
if(tmp->next) pq.push(tmp->next);
*headP = tmp;
headP = &((*headP)->next);
}
*headP = NULL;
return head;
}
};
堆就用C++的priority_queue實現,注意我代碼裏面用了個二維指針,爲啥要用它,實際上它的作用跟dummy head是一樣的,就是新開一個節點指向頭,避免頭指針特判的情況。
題目描述:Merge two sorted linked lists and return it as a new list. The new list should be made by splicing together the nodes of the first two lists.
這個題是上面那個題的簡化版,由於只有兩個list,就不需要用堆來維護了。
code:
class Solution {
public:
ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) {
ListNode* dummy = new ListNode(1),*ite = dummy;
while(l1&&l2){
if(l1->val<l2->val){
ite->next = l1;
l1=l1->next;
}else{
ite->next = l2;
l2=l2->next;
}
ite=ite->next;
}
l1 = l2?l2:l1;
while(l1){
ite->next=l1;
l1=l1->next;
ite=ite->next;
}
ite->next = NULL;
ite = dummy->next;
delete dummy;
return ite;
}
};
由於有是鏈表的題,還是有鏈表頭特判的問題。這次我用的是dummy
head(而不是指針的指針),可以比較一下差別。
題目描述:Given two sorted integer arrays A and B, merge B into A as one sorted array.
跟上面的大同小異,指針換成數組了。
class Solution {
public:
void merge(int A[], int m, int B[], int n) {
int end = n+m-1;
int ia=m-1,ib=n-1;
while(ia>=0&&ib>=0)
{
if(A[ia]>B[ib])
A[end--] = A[ia--];
else
A[end--] = B[ib--];
}
while(ia>=0)
A[end--] = A[ia--];
while(ib>=0)
A[end--] = B[ib--];
}
};
=======================以後陸續補充==================以後陸續補充==================以後陸續補充==================