寻找最短的周期,包括一个特定的边缘

时间:2018-08-01 23:58:34

标签: c++ graph shortest-path directed-graph weighted-graph

您好,我想解决此网站http://rosalind.info/problems/cte/上的“通过给定边缘的最短周期”问题。 假设我们的特定优势(这是我们在此问题中的第一优势)为“ E”。 我写了一个程序来解决这个问题,我的算法是在'E'的end_node上使用DFS,直到遇到'E'的start_node。对于该网站上给出的示例,它工作正常,但是当我使用大量数据时,它就会循环播放。 我尝试了许多简单的有向图示例,但没有发现为什么会循环。 有人可以告诉我是否有循环吗?

这是我的程序:

#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
#include <map>
#include <set>
#include <string>
#include <queue>
#include <list>
#include <sstream>
#include <iomanip>
#include <fstream>
#include <stack>


using namespace std;

#define ll long long int
#define pb push_back
#define mk make_pair

const ll maxn = 1e5 + 3;
const ll inf = 1e12 + 10;

vector <ll> cost;

void is_cyclic(int i, bool * recstack,ll cnt,vector <pair<ll,ll>> * vec,ll ind) {
    recstack[i] = true;
    vector <pair<ll, ll>>::iterator it = vec[i].begin();
    for (; it != vec[i].end(); it++) {
        if (it->first == ind) {
            cnt += it->second;
            cost.pb(cnt);
        }
        else {
            if (recstack[it->first] == false) {
                is_cyclic(it->first, recstack, cnt + it->second, vec, ind);
            }
        }
    }
    recstack[i] = false;
}

int main() {
    ll k;
    //file >> k;
    //while (k--) {
        ll n, m;
        cin >> n >> m;
        ll s;
        vector <pair<ll, ll>> * vec;
        vec = new vector<pair<ll, ll>>[n];
        for (int j = 0; j < m; j++) {
            ll a, b, c;
            cin >> a >> b >> c;
            if (j == 0) {
                s = a - 1;
            }
            vec[a - 1].pb(mk(b - 1, c));
        }
        bool * recstack = new bool[n];
        for (int j = 0; j < n; j++) {
            recstack[j] = false;
        }
        ll cnt = 0;
        pair <ll, ll> p;
        p = vec[s][0];
        cnt += p.second;
        recstack[s] = true;
        is_cyclic(p.first, recstack, cnt, vec,s);
        if (cost.size() == 0) {
            cout << -1 << " ";
        }
        else {
            sort(cost.begin(), cost.end());
            cout  << cost[0] << " ";
        }
        cost.clear();
    //}
    return 0;
}

1 个答案:

答案 0 :(得分:0)

首先,您应该使用BFS而非DFS来计算未加权图中的最短路径。

第二,几乎不可能弄清楚您的代码是如何工作的。如果您想获得有意义的答案,建议您清理代码。这里有一些指针:

  • 请使用描述变量作用/含义的变量名。像std::vector<…> vec这样的变量名使您的代码很难理解。
  • 添加一些注释以告诉我们您的代码在哪里做什么。
  • 请不要使用#define pb push_back之类的思想。这使得几乎所有不是您的人都无法阅读您的代码。
  • 您使用了许多不必要的指针/堆分配(并且您没有delete。)示例是vecrecstack(使std::vector<bool>… )
  • 使用std::numeric_limits而不是定义自己的maxn等。
  • 使用.push_back(std::make_pair(a,b))代替.emplace_back(a,b)
相关问题