在图中删除连接到源顶点的边时获得额外边

时间:2013-11-11 17:27:20

标签: c++ graph

上周我发布了一个代码来计算使用Dijkastra算法的图中的最短路径,但它很长,没有人有兴趣完全读取它所以我删除它,现在我试图通过参与简化代码部分代码尚未完成,我将在此处删除部分代码,重点关注到目前为止我遇到的第一个问题。

简单地说我有一个类Graph它将由另外两个类构造,一个元素向量是一个类Edge的实例,另一个是Vertex类的元素向量,每个顶点都有一个id,每个边有两个顶点和重量。

类Graph有一个方法,它的名称最短,需要两个顶点作为参数,第一个是图的源,第二个是目的地。

到目前为止,我正在尝试消除连接到源顶点的边缘,但是我在矢量边缘仍然获得了额外的边缘,它连接到源,而与源相关的所有其他边缘都被移除。

为了演示结果,我初始化了一个图形有五个顶点vers [0],vers [1],vers [2],vers [3],vers [4],并且有10个边连接这些顶点从eds [0],eds [1],.... eds [9]。

源顶点是vers [2]由4个边连接,所以当应用最短的方法时,如下面的代码所示,我应该去除所有这4个边并以6个边结束,但结果是我摆脱了3条边缘,剩下7条边,结果如下:

Hello, This is a graph
0____1     5
0____3     4
0____4     6
1____3     5
1____4     7
2____4     8
3____4     3
size of edges 7
size of vertices 5

你可以注意到,还有一条连接到源的边是2,问题出在这边(顺便说一下,8是边的权重)

2____4     8

方法中有一些错误,特别是在for循环中,我希望你能找到我的错误。

提前致谢。 这是代码

#include<iostream>
#include<vector>
#include <stdlib.h>   // for rand()
using namespace std;

const unsigned int N = 5;

class Vertex
{
 private:
     unsigned int id;                 // the name of the vertex  
public:   
    unsigned int get_id(){return id;};
    void set_id(unsigned int value) {id = value;};

    Vertex(unsigned int init_val = 0) :id (init_val){}      // constructor   
    ~Vertex() {};                                           // destructor
};


class Edge
{
  private:
    Vertex first_vertex;                 // a vertex on one side of the edge
    Vertex second_vertex;                // a vertex on the other side of the edge
    unsigned int weight;                 // the value of the edge ( or its weight )     
  public:   
    unsigned int get_weight() {return weight;};
    void set_weight(unsigned int value) {weight = value;};

    Vertex get_ver_1(){return first_vertex;};
    Vertex get_ver_2(){return second_vertex;};

    void set_first_vertex(Vertex v1) {first_vertex = v1;};
    void set_second_vertex(Vertex v2) {second_vertex = v2;};


    Edge(const Vertex& vertex_1 = 0, const Vertex& vertex_2 = 0, unsigned int init_weight = 0)
    : first_vertex(vertex_1), second_vertex(vertex_2), weight(init_weight)
     {
     }  


    ~Edge() {} ; // destructor

};


class Graph
{
private:
     std::vector<Vertex>   vertices;
     std::vector<Edge>   edges;  


public:
     Graph(vector<Vertex> ver_vector, vector<Edge> edg_vector)
    : vertices(ver_vector), edges(edg_vector){}

     ~Graph() {}

     vector<Vertex> get_vertices(){return vertices;}
     vector<Edge> get_edges(){return edges;}

     void set_vertices(vector<Vertex> vector_value) {vertices = vector_value;}
     void set_edges(vector<Edge> vector_ed_value) {edges = vector_ed_value;}

     unsigned int shortest(Vertex src, Vertex dis);


};



unsigned int Graph::shortest(Vertex src, Vertex dis) {
        vector<Vertex> ver_out;
        vector<Edge> track;

        for(unsigned int i = 0; i < edges.size(); ++i)
            {
                if((edges[i].get_ver_1().get_id() == src.get_id()) || (edges[i].get_ver_2().get_id() == src.get_id()))
                    {
                    track.push_back (edges[i]);

                    if(edges[i].get_ver_1().get_id() == src.get_id())
                        {ver_out.push_back (edges[i].get_ver_1());}
                    else
                        {ver_out.push_back (edges[i].get_ver_2());}

                    edges.erase(edges.begin() + i ); //****
                    }
            };


    }





int main()
{
cout<< "Hello, This is a graph"<< endl;

vector<Vertex> vers(5);
vers[0].set_id(0);
vers[1].set_id(1);
vers[2].set_id(2);
vers[3].set_id(3);
vers[4].set_id(4);

vector<Edge> eds(10);
eds[0].set_first_vertex(vers[0]);
eds[0].set_second_vertex(vers[1]);
eds[0].set_weight(5);   

eds[1].set_first_vertex(vers[0]);
eds[1].set_second_vertex(vers[2]);
eds[1].set_weight(9);

eds[2].set_first_vertex(vers[0]);
eds[2].set_second_vertex(vers[3]);
eds[2].set_weight(4);

eds[3].set_first_vertex(vers[0]);
eds[3].set_second_vertex(vers[4]);
eds[3].set_weight(6);

eds[4].set_first_vertex(vers[1]);
eds[4].set_second_vertex(vers[2]);
eds[4].set_weight(2);

eds[5].set_first_vertex(vers[1]);
eds[5].set_second_vertex(vers[3]);
eds[5].set_weight(5);

eds[6].set_first_vertex(vers[1]);
eds[6].set_second_vertex(vers[4]);
eds[6].set_weight(7);

eds[7].set_first_vertex(vers[2]);
eds[7].set_second_vertex(vers[3]);
eds[7].set_weight(1);

eds[8].set_first_vertex(vers[2]);
eds[8].set_second_vertex(vers[4]);
eds[8].set_weight(8);

eds[9].set_first_vertex(vers[3]);
eds[9].set_second_vertex(vers[4]);
eds[9].set_weight(3);


unsigned int path;

Graph graf(vers, eds);
path = graf.shortest(vers[2], vers[4]);


cout<<graf.get_edges()[0].get_ver_1().get_id() <<"____"<<graf.get_edges()[0].get_ver_2().get_id() <<"     "<<graf.get_edges()[0].get_weight()<< endl;  //test
cout<<graf.get_edges()[1].get_ver_1().get_id() <<"____"<<graf.get_edges()[1].get_ver_2().get_id() <<"     "<<graf.get_edges()[1].get_weight()<< endl;  //test
cout<<graf.get_edges()[2].get_ver_1().get_id() <<"____"<<graf.get_edges()[2].get_ver_2().get_id() <<"     "<<graf.get_edges()[2].get_weight()<< endl;  //test
cout<<graf.get_edges()[3].get_ver_1().get_id() <<"____"<<graf.get_edges()[3].get_ver_2().get_id() <<"     "<<graf.get_edges()[3].get_weight()<< endl;  //test
cout<<graf.get_edges()[4].get_ver_1().get_id() <<"____"<<graf.get_edges()[4].get_ver_2().get_id() <<"     "<<graf.get_edges()[4].get_weight()<< endl;  //test
cout<<graf.get_edges()[5].get_ver_1().get_id() <<"____"<<graf.get_edges()[5].get_ver_2().get_id() <<"     "<<graf.get_edges()[5].get_weight()<< endl;  //test
cout<<graf.get_edges()[6].get_ver_1().get_id() <<"____"<<graf.get_edges()[6].get_ver_2().get_id() <<"     "<<graf.get_edges()[6].get_weight()<< endl;  //test
//cout<<graf.get_edges()[7].get_ver_1().get_id() <<"____"<<graf.get_edges()[7].get_ver_2().get_id() <<"     "<<graf.get_edges()[7].get_weight()<< endl;  //test
//cout<<graf.get_edges()[8].get_ver_1().get_id() <<"____"<<graf.get_edges()[8].get_ver_2().get_id() <<"     "<<graf.get_edges()[8].get_weight()<< endl;  //test
//cout<<graf.get_edges()[9].get_ver_1().get_id() <<"____"<<graf.get_edges()[9].get_ver_2().get_id() <<"     "<<graf.get_edges()[9].get_weight()<< endl;  //test


cout<<"size of edges"<<graf.get_edges().size()<< endl;
cout<<"size of vertices"<<graf.get_vertices().size()<< endl;

return 0;
}

1 个答案:

答案 0 :(得分:1)

这是因为你正在有效地跳过Graph::shortest for循环中的一些向量元素,因为即使你i当前元素,你也在增加erase。将其更改为类似的内容以解决问题:

for (unsigned int i = 0; i < edges.size();) { // no ++i here
    if ((edges[i].get_ver_1().get_id() == src.get_id()) || (edges[i].get_ver_2().get_id() == src.get_id())) {
        track.push_back(edges[i]);

        if (edges[i].get_ver_1().get_id() == src.get_id()) {
            ver_out.push_back(edges[i].get_ver_1());
        } else {
            ver_out.push_back(edges[i].get_ver_2());
        }

        edges.erase(edges.begin() + i);
    } else {
        ++i; // increment only if not erasing
    }
}

或者根据评论,使用迭代器:

for (auto i = edges.begin(); i != edges.end();) {
    if ((i->get_ver_1().get_id() == src.get_id()) || (i->get_ver_2().get_id() == src.get_id())) {
        track.push_back(*i);

        if (i->get_ver_1().get_id() == src.get_id()) {
            ver_out.push_back(i->get_ver_1());
        } else {
            ver_out.push_back(i->get_ver_2());
        }

        i = edges.erase(i);
    } else {
        i++;
    }
}

您在该函数中也缺少return语句。