Codeforces 716D

题意：给定n个点,m条边，以及起点s，终点t，问你图中是否存在s->t的最短路为L，其中权值为0的可以任意修改。

思路：对给定的边分为2类，权重不为0的直接扔进去建图，权重为0的边先存起来。接着跑一遍堆优化的dij，如果dis[t]小于L，那么无论怎么修改权重0的边都没有办法凑出最短路L；

如果dis[t]=L,那么直接输出；如果dis[t]>L,则将原来事先存起来的边一条一条的加进去，权值设为1。每加一条边就跑一次dij，一旦找到dis[t]<=L，就可以终止并输出。

PS：原来堆优化的dij就是用优先队列优化，自己敲了一遍才知道，看来模板这东西得多敲才会懂。。。

#include <iostream>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <map>
#include <set>
#include <bitset>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <string>
#include <sstream>
#include <time.h>
#define x first
#define y second
#define pb push_back
#define mp make_pair
#define lson l,m,rt*2
#define rson m+1,r,rt*2+1
#define mt(A,B) memset(A,B,sizeof(A))
#define mod 1000000007
using namespace std;
typedef long long LL;
const double PI = acos(-1);
const int N=30000+10;
const int inf = 0x3f3f3f3f;
//const LL INF=0x3f3f3f3f3f3f3f3fLL;
const LL INF=1e15;
struct node
{
    int u,v,next;
    LL w;
} E[N];
int n,m,s,t,top=0,head[N];
LL L,dis[N];
bool vis[N];
vector<node> P;
void add(int u,int v,LL w)
{
    E[top].u=u;
    E[top].v=v;
    E[top].w=w;
    E[top].next=head[u];
    head[u]=top++;
}
struct Node//堆优化的dij所用的结构体
{
    int v;//点
    LL d;//距离
    Node(int v=0,LL d=0):v(v),d(d) {}//我也不懂为什么这样写，反正照写就对了
    bool operator <(const Node &r)const
    {
        return d>r.d;
    }
};
void dij()
{
    for(int i=0; i<N; i++)//先初始化
    {
        dis[i]=INF;
        vis[i]=false;
    }
    dis[s]=0;
    priority_queue<Node> Q;
    Q.push(Node(s,0));
    while(!Q.empty())
    {
        Node p=Q.top();
        Q.pop();
        if(vis[p.v])continue;
        vis[p.v]=true;
        for(int i=head[p.v]; i!=-1; i=E[i].next)
        {
            int v=E[i].v;
            if(!vis[v]&&dis[v]>dis[p.v]+E[i].w)//松弛法
            {
                dis[v]=dis[p.v]+E[i].w;
                Q.push(Node(v,dis[v]));
            }
        }
    }
}
int main()
{
#ifdef Local
    freopen("data.txt","r",stdin);
#endif
    int u,v;
    LL w;
    mt(head,-1);//邻接表初始化
    cin>>n>>m>>L>>s>>t;
    for(int i=0; i<m; i++)//先将节点分类，w!=0的先加进去建图
    {
        cin>>u>>v>>w;
        if(w)
        {
            add(u,v,w);
            add(v,u,w);
        }
        else
        {
            node p;
            p.u=u;
            p.v=v;
            p.w=w;
            p.next=-1;
            P.pb(p);
        }
    }
    dij();
    if(dis[t]<L)
    {
        puts("NO");
    }
    else if(dis[t]==L)
    {
        puts("YES");
        for(int i=0; i<top; i+=2)
        {
            printf("%d %d %I64d
",E[i].u,E[i].v,E[i].w);
        }
        for(int i=0; i<P.size(); i++)
        {
            printf("%d %d %I64d
",P[i].u,P[i].v,INF);
        }
    }
    else
    {
        int pos=0,flag=0;
        for(pos=0; pos<P.size(); pos++)
        {
            add(P[pos].u,P[pos].v,1);
            add(P[pos].v,P[pos].u,1);
            dij();
            if(dis[t]<=L)
            {
                E[top-1].w+=L-dis[t];
                E[top-2].w+=L-dis[t];
                flag=1;
                break;
            }
        }
        if(!flag)puts("NO");
        else
        {
            puts("YES");
            for(int i=0; i<top; i+=2)
            {
                printf("%d %d %I64d
",E[i].u,E[i].v,E[i].w);
            }
            for(int i=pos+1; i<P.size(); i++)
            {
                printf("%d %d %I64d
",P[i].u,P[i].v,INF);
            }
        }
    }
#ifdef Local
    cerr << "time: " << (LL) clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
#endif
}