set容器插入自定义类型排序

桃扇骨 2023-05-31 10:22 22阅读 0赞

## 背景 ##

做实验的时候发现图网络数据边不是按照生成时间排序的  
因此本文背景就是讲网络数据的边按照时间排序。  
数据格式：src dest time  
想到使用set自动排序。  
因此写了程序

## 第一次预处理 ##

定义边类和排序规则

class Edge
    { 
    public:
    	Edge(int src,int dest,int weight){ 
    		this->src=src;
    		this->dest=dest;
    		this->weight=weight;
    	}
    
    	int src;
    	int dest;
    	int weight;
    };
    
    class compareEdge
    { 
    public:
    	bool operator()(const Edge& e1,const Edge& e2)//first weight
    	{ 
    		return e1.weight<e2.weight;
    	}
    };

main函数中读数据到自定义的set<Edge,compareEdge> 类型的变量中  
使用迭代器一次读取就可以了

#### 问题： 但是这样发现预处理后边数目减少很多 ####

#### 原因： set容器中key值不重复 程序运行时讲weight相同的边去掉了 ####

## 第二次预处理 ##

class compareEdge
    { 
    public:
    	bool operator()(const Edge& e1,const Edge& e2)//first weight
    	{ 
    		if(e1.weight==e2.weight)
    		{ 
    			if(e1.src==e2.src)
    			{ 
    				return e1.dest<e2.dest;
    			}
    			return e1.src<e2.src;
    		}
    		return e1.weight<e2.weight;
    	}
    };

主要修改了compareEdge的排序。  
首先按照权重排序，如果权重相同，按照源节点排序，如果源节点相等。按照目的节点排序。  
运行的结果存在weight相同过的边了。  
![在这里插入图片描述][20191024170541748.png]

[20191024170541748.png]: /images/20230531/27f38cc5642845cfb70f0075e8a787c0.png