Question

我正在尝试使用堆来解决问题“合并K列表”，它正在合并k个已排序的链表并将其作为一个排序列表返回。通常我创建一个min heap来存储所有列表节点，并使用预定义函数LessThanLinkedList（）进行比较。但我发现第62行和第75行中的pop_heap（）操作永远不会起作用。虽然我使用预定义的比较函数作为参数，但它不会删除堆的顶部。以下是我的代码。我使用visual studio 2010作为IDE。谁知道原因？非常感谢你的帮助！

#include <stdio.h>
#include <stdlib.h>
#include <vector>
#include <queue>
#include <list>
#include <numeric>

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

class Solution {
public:

  static bool LessThanLinkedList( const ListNode * l1, const ListNode * l2) 
  {
     return( l1->val > l2->val );
  }

  ListNode *mergeKLists(vector<ListNode *> &lists) {

     int idx;
     bool ball_list_null;
     ListNode * pNode;
     ListNode *new_head;

     ball_list_null = true;

     for( idx = 0; idx < lists.size(); idx++ )
     {
         if( NULL != lists[idx] )
         {
             ball_list_null = false;
             break;
         }
     }
     if( true == ball_list_null )
         return(NULL);

     vector< ListNode* > list_heap;
     for( idx = 0; idx < lists.size(); idx++ )
     {
         if( NULL != lists[idx] )
         {
             pNode = lists[idx];
             while( NULL != pNode )
             {
                 list_heap.push_back( pNode );
                 pNode = pNode->next;
             }
         }
     }

     make_heap( list_heap.begin(), list_heap.end(), LessThanLinkedList );

     if(list_heap.size() > 0) 
     {
         new_head = list_heap[0];
         pop_heap( list_heap.begin(), list_heap.end(), LessThanLinkedList );//not work
     }

     if( list_heap.size() == 0 )
     {
         new_head->next = NULL;
     }
     else
     {
         pNode = new_head;
         while( list_heap.size() >0 )
         {
             pNode->next = list_heap[0];
             pop_heap( list_heap.begin(), list_heap.end(), LessThanLinkedList ); // not work
             pNode = pNode->next ;
         }
         pNode->next = NULL;
     }
     return( new_head );

  }
};

void main()
{
  Solution xpfsln;
  ListNode *l1,*l2,*l3,*l4,*l5,*l6,*l7,*head;
  l1 = new ListNode(1);
  l2 = new ListNode(2);
  l3 = new ListNode(3);

  l1->next = l2;
  l2->next = l3;
  l3->next = NULL;

  vector<ListNode *> list_vec;
  list_vec.push_back(l1);

  head = xpfsln.mergeKLists( list_vec );
}

Answer 1

pop_heap不会从容器中删除元素。它不能，因为它甚至不能访问容器，只是它的元素。它的作用（当然假设[begin, end)形成一个有效的非空堆）重新排列元素，使得堆的第一个元素移动到该范围的最后一个元素，并离开{{1} }作为有效的堆。如果您想要从容器中删除元素，则只需在调用[begin, end-1)后删除容器的最后一个元素（例如，通过调用pop_back()）。

pop_heap

C ++ STL pop_heap不起作用

1 个答案: