c++针对特定数据结构创建堆

make_heap

https://en.cppreference.com/w/cpp/algorithm/make_heap

struct ds 
{
	double value; 
	int idx;       
	ds(double v, int index) : value(v), idx(index){}
};
   
// 自定义比较函数   
class CMP {
public:
	bool operator()(const ds& a, const ds& b) {
		return a.value< b.value;
	}
};

std::vector<ds> edgeWeights;
std::make_heap(edgeWeights.begin(), edgeWeights.end(), CMP());  // max heap

while (edgeWeights.size() > 0) {
        bool bHeap = std::is_heap(edgeWeights.begin(), edgeWeights.end(), CMP());
	if (!bHeap) {
	    std::make_heap(edgeWeights.begin(), edgeWeights.end(), CMP());
	    // bHeap = std::is_heap(edgeWeights.begin(), edgeWeights.end(), CMP());
	}
	std::pop_heap(edgeWeights.begin(), edgeWeights.end(), CMP());
	ds top = edgeWeights.back();
	edgeWeights.pop_back();

        edgeWeights.push_back(ds(0.0, 1, eId));
	std::push_heap(edgeWeights.begin(), edgeWeights.end(), CMP());
	continue; 
}

priority_queue

https://en.cppreference.com/w/cpp/container/priority_queue


struct ds
{
	double value;
	int idx;
	ds(double d, int index) : value(d), idx(index) {}
};

struct CMP : public std::binary_function<ds, ds, bool>
{	
	bool operator()(const ds& _Left, const ds& _Right) const
	{	
		return _Left.value > _Right.value;   // min heap
	}
};
typedef std::priority_queue<ds, std::vector<ds>, CMP > DistancePriorityQueue;

priQueue.push(ds(2.0, 0));

while (!priQueue.empty())
{
	ds di = priQueue.top();
	priQueue.pop();
}
			
posted @   半夜打老虎  阅读(40)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 25岁的心里话
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 零经验选手,Compose 一天开发一款小游戏!
· 因为Apifox不支持离线,我果断选择了Apipost!
· 通过 API 将Deepseek响应流式内容输出到前端
点击右上角即可分享
微信分享提示