Files
data-structures-and-algorithms/homework5/ListGraph.h
T

137 lines
3.9 KiB
C++

//
// Created by 31416 on 24-12-4.
//
#ifndef LIST_GRAPH_H
#define LIST_GRAPH_H
#include <sstream>
#include<vector>
#include "GraphExceptions/InsertExistedConnectException.h"
#include "GraphExceptions/NodeIdOutOfRangeException.h"
#include "GraphExceptions/SameNodeConnectException.h"
using namespace std;
/*
* 简述:
* 节点=数据域+指针域
* 第一个节点存储id
* 第二个节点存储值
* 后续节点存储连接关系
* 数据域:
* data:可存id,可存其他节点的id,可存数据
* weight:边上的权
*/
template<typename T>
class ListGraph {
private:
struct node{
T data;
T weight;
struct node* next;
};
vector<node*> nodes;
bool flag; //定义是否有向
bool isWeighted; //定义是否有权
public:
ListGraph() {
nodes.clear();
flag = false;
isWeighted = false;
}
ListGraph(bool flag) {
nodes.clear();
this->flag = flag;
isWeighted = false;
}
ListGraph(bool flag,bool isWeighted) {
nodes.clear();
this->flag = flag;
this->isWeighted = isWeighted;
}
void insert(T data) {
node* insert = new node();
insert->data=nodes.size();
insert->weight = 0;
insert->next=new node();
insert->next->data=data;
insert->next->next=NULL;
nodes.push_back(insert);
}
void connect(int id1,int id2,T weight=0) { //flag区分是否为有向图
if ((id1>nodes.size()-1||id2>nodes.size()-1)||(id1<0)||(id2<0)) {
throw NodeIdOutOfRangeException("您提供的id超出范围",0X002);
}
if(id1==id2) {
throw SameNodeConnectException("禁止连接图中的同一个点",0x001);
}
node *current = nodes.at(id1);
int current_id = 0;
while(current->next!=NULL) {
if((current->next->data==id2)&&(current_id>1)) {
if (current->next->weight==weight) {
throw InsertExistedConnectException("禁止插入已存在的连接",0x003);
}
current->next->weight=weight;
}
current = current->next;
current_id++;
}
current->next=new node();
current->next->data=id2;
current->next->weight = weight;
current->next->next=NULL;
if (!flag) {
node *current2 = nodes.at(id2);
int current2_id = 0;
while(current2->next!=NULL) {
if((current2->next->data==id2)&&(current2_id>1)) {
if (current2->next->weight==weight) {
throw InsertExistedConnectException("禁止插入已存在的连接",0x003);
}
current2->next->weight=weight;
}
current2 = current2->next;
current2_id++;
}
current2->next=new node();
current2->next->data=id1;
current2->next->weight = weight;
current2->next->next=NULL;
}
}
string to_string() {
stringstream ss;
for (int i=0;i<=nodes.size()-1;i++) {
node* current = nodes.at(i);
while(current!=NULL) {
if(current->next==NULL) {
if (isWeighted) {
ss<<current->data<<"|"<<current->weight;
}
else {
ss<<current->data;
}
}else {
if (isWeighted){
ss<<current->data<<"|"<<current->weight<<" ";
}
else {
ss<<current->data<<" ";
}
}
current = current->next;
}
ss<<endl;
}
return ss.str();
}
};
#endif //LIST_GRAPH_H