Codeforces Round #372 (Div. 2) -- D. Complete The Graph（SPFA）

大体题意：

给你一个无向图，有n 个顶点和m 个边，一些边的权重是正数，一些边的权重是0，表示已经删除，告诉你起始位置和终止位置和最短路L，要求把已经删除的边（权值为0）重建，自己赋值，使得最短路依然是L。

思路：

单源最短路，其实想清楚了还是很简单的！

先求一遍最短路，当发现最短路L2  小于L时，那么肯定是无解的，因为你此时最短路已经小于L了，在加入一些重建的边，只能会越来越小，因此不可能达到L！

如果L2 == L  我们就可以直接输出了，把0边随便输出一个inf 无穷大就好（相当于此路不通！！）

如果L2 < L的话，我们就要加0边了！

枚举每一个0边，加入一个0边就在求一次最短路，直到加入某个边使得最短路不大于L即可！把此时的位置记录下来！

那么先输出已知边，在输出记录位置的0边之前的边，权值为1，然后根据L和最短路的关系算出记录位置的0边的权重！

最后剩下的0边 随意输出一个inf即可！

#include <set>
#include <map>
#include <queue>
#include <deque>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define L(i) i<<1
#define R(i) i<<1|1
#define INF  0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-9
#define maxn 1000010
#define MOD 1000000007

struct Edge
{
    int to,next;
    long long val;
    bool operator<(const Edge &a)const
    {
        return val < a.val;
    }
} edge[20010];
int n,m,s,t,tot;
int head[10010],vis[10010];
long long L,dis[10010];

void init()
{
    tot = 0;
    memset(head,-1,sizeof(head));
}
void add_edge(int u,int v,long long val)
{
    edge[tot].to = v;
    edge[tot].next = head[u];
    edge[tot].val = val;
    head[u] = tot++;
}
void SPFA()
{
    queue<int> q;
    memset(vis,0,sizeof(vis));
    memset(dis,INF,sizeof(dis));
    q.push(s);
    vis[s] = 1;
    dis[s] = 0;
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        vis[u] = 0;
        for(int i = head[u]; i != -1; i = edge[i].next)
        {
            int v = edge[i].to;
            if(dis[v] > dis[u] + edge[i].val)
            {
                dis[v] = dis[u] + edge[i].val;
                if(!vis[v])
                {
                    q.push(v);
                    vis[v] = 1;
                }
            }
        }
    }
}
int x[10010],y[10010];
long long tmp[10010];

int main()
{
    int T;
    while(scanf("%d%d%lld%d%d",&n,&m,&L,&s,&t) != EOF)
    {
        init();
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d%lld",&x[i],&y[i],&tmp[i]);
            if(tmp[i] != 0)
            {
                add_edge(x[i],y[i],tmp[i]);
                add_edge(y[i],x[i],tmp[i]);
            }
        }
        SPFA();
        if(dis[t] < L)
            printf("NO\n");
        else if(dis[t] == L)
        {
            printf("YES\n");
            for(int i = 0; i < m; i++)
                printf("%d %d %lld\n",x[i],y[i],!tmp[i]?1000000000000000000:tmp[i]);
        }
        else
        {
            int flag = -1;
            for(int i = 0; i < m; i++)
            {
                if(tmp[i] != 0)
                    continue;
                tmp[i] = 1;
                add_edge(x[i],y[i],tmp[i]);
                add_edge(y[i],x[i],tmp[i]);
                SPFA();
                if(dis[t] <= L)
                {
                    flag = i;
                    break;
                }
            }
            if(flag != -1)
            {
                printf("YES\n");
                for(int i = 0; i < flag; i++)
                    printf("%d %d %lld\n",x[i],y[i],tmp[i]);
                printf("%d %d %lld\n",x[flag],y[flag],L-dis[t]+1);
                for(int i = flag + 1; i < m; i++)
                    printf("%d %d %lld\n",x[i],y[i],!tmp[i]?1000000000000000000:tmp[i]);
            }
            else
                printf("NO\n");
        }
    }
    return 0;
}