CareerCup-4.2

简介:

Given a directed graph, design an algorithm to find out whether there is a route between two nodes.

开始想用Dijkstra算法,后来发现根本不需要最短只求连通性。回头还得回忆下Dijkstra,写起来有点生疏了。 

#include <iostream>
#include <stack>
using namespace std;
#define Data int
class Graph
{
public:
    Graph(int size):mSize(size)
    {
        data = new Data*[size];
        for(int i=0; i<size; i++)
        {
            data[i] = new Data[size];
            for(int j=0; j<size; j++)
            {
                data[i][j] = 0;
            }
        }
    };
    void addLink(int src, int dst, int weight=1)
    {
        data[src][dst] = weight;
    };
    bool isReached(int src, int dst)
    {
        if(src == dst) return true;
        bool reached[mSize];
        for(int i=0; i<mSize; i++)
            reached[i] = false;
        stack<Data> stack;
        stack.push(src);
        reached[src] = true;
        while(true)
        {
            Data d = stack.top();
            cout<<"Reach: "<<d<<endl;
            if(d == dst)
                return true;
            stack.pop();
            for(int i=0; i<mSize; i++)
            {
                if(data[d][i] > 0 && !reached[i])
                    stack.push(i);
            };
            if(stack.empty()) break;
        };
        return false;
    };
private:
    int mSize;
    Data** data;
};

int main()
{
    Graph graph(4);
    graph.addLink(0, 3);
    graph.addLink(3, 2);
    cout<<boolalpha<<graph.isReached(0, 1)<<endl;
    system("pause");
}

目录
相关文章
|
机器学习/深度学习
|
机器学习/深度学习