NYOJ1006 偷西瓜(最短路)

题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=1006

这道题是一道求最短路的题， 这次用的spfa算法写的， 前几道都是用dijkstra写的， 算是练习一下。 此题让求此来回且不能有重路的最短路， 最短路都会求，最容易想到的是求两次最短路， 但是不能重路，  本题最先想到的就删边， 但是仔细想想就知道删边是很难实现， 但是我们可以转化为把边标记不能再走就可以了。

#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn = 100 + 10;
const int M = 1000 + 10;
struct Edge
{
    int from, to, dist;
    Edge(int u, int v, int d) : from(u), to(v), dist(d) {}
};
int n, m;
int d[maxn], p[maxn], vis[2*M];  //标记这个边所存的位置，来标记边不能再用
bool inq[maxn];
vector<Edge> edges;
vector<int> G[maxn];
void inin()
{
    for(int i = 1; i <= n; i++)
        G[i].clear();
    edges.clear();
}
void AddEdge(int from, int to, int dist)
{
    edges.push_back(Edge(from, to, dist));
    int k = edges.size();
    G[from].push_back(k-1);
}
void spfa(int s)
{
    memset(p, -1, sizeof(p));
    memset(inq, 0, sizeof(inq));
    memset(d, 99, sizeof(d));
    queue<int> q;
    q.push(s);
    d[s] = 0;
    inq[s] = true;
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        inq[u] = false;
        for(int i = 0; i < G[u].size(); i++)
        {
            int k = G[u][i];
            if(!vis[k])
            {
                Edge& e = edges[k];
                if(d[e.to] > d[u]+e.dist)
                {
                    d[e.to] = d[u] + e.dist;
                    p[e.to] = G[u][i];
                    if(!inq[e.to])
                    {
                        q.push(e.to);
                        inq[e.to] = true;
                    }
                }
            }
        }
    }
}
void bfs(int k)
{
    if(k == -1)
        return ;
    vis[k] = 1;
    bfs(p[edges[k].from]);
}
void del()
{
    bfs(p[n]);
    for(int i = 0; i < 2*m; i+=2)
        if(vis[i] || vis[i+1])
            vis[i] = vis[i+1] = 1;
}
int main()
{
    int u, v, w, sum;
    while(~scanf("%d", &n))
    {
        scanf("%d", &m);
        memset(vis, 0, sizeof(vis));
        sum = 0;
        inin();
        while(m--)
        {
            scanf("%d%d%d", &u, &v, &w);
            AddEdge(u, v, w);
            AddEdge(v, u, w);
        }
       spfa(1);
       del();

       if(p[n] == -1)
       {
           printf("eating\n");
           continue;
       }
       else
           sum+=d[n];
       spfa(1);
       if(p[n] == -1)
           printf("eating\n");
       else
           printf("%d\n", sum+d[n]);
    }
    return 0;
}