多段图的最短路径问题动态规划算法c++代码

以下是多段图的最短路径问题动态规划算法的C++代码：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 10010;
const int M = 200010;

int n, m, k;
int h[N], e[M], ne[M], w[M], idx;
int dist[N];
int q[N], d[N], cnt[N];

void add(int a, int b, int c)
{
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx ++ ;
}

int main()
{
    scanf("%d%d%d", &n, &m, &k);

    memset(h, -1, sizeof h);
    while (m -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c);
    }

    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;

    for (int i = 1; i <= k + 1; i ++ )
    {
        int hh = 0, tt = -1;

        for (int j = 1; j <= n; j ++ )
            if (d[j] == i - 1)
                q[ ++ tt] = j;

        while (hh <= tt)
        {
            int t = q[hh ++ ];

            for (int j = h[t]; ~j; j = ne[j])
            {
                int ver = e[j];
                if (dist[ver] > dist[t] + w[j])
                {
                    dist[ver] = dist[t] + w[j];
                    d[ver] = i;
                    cnt[ver] = cnt[t] + 1;
                    if (cnt[ver] >= n)
                    {
                        puts("Yes");
                        return 0;
                    }
                    q[ ++ tt] = ver;
                }
            }
        }
    }

    puts("No");

    return 0;
}

注意：这段代码假设图中所有的边都是正边，如果有负边需要进行特殊处理。