POJ 1797 Heavy Transportation

SPFA水题。记录一下到这个结点的最大承载能力。因为一个小错误而WA了一晚上,悲催,T_T......

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;

const int maxn=1000+10;
struct Edge{int from,to,w;}e[1000010];
int dis[maxn],vis[maxn];
vector<Edge>G[maxn];
int n,m,tot;

void SPFA()
{
    queue<int>Q;
    memset(dis,0,sizeof(dis));
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    Q.push(1);
    while(!Q.empty())
    {
        int h=Q.front();Q.pop();vis[h]=0;
        for(int i=0;i<G[h].size();i++)
        {
            Edge &e=G[h][i];
            if(h==1)
            {
                dis[e.to]=e.w;
                Q.push(e.to);
                vis[e.to]=1;
            }
            else
            {
                if(min(dis[h],e.w)>dis[e.to])
                {
                    dis[e.to]=min(dis[h],e.w);
                    if(vis[e.to]==0)
                    {
                        Q.push(e.to);
                        vis[e.to]=1;
                    }
                }
            }
        }
    }
}
int main()
{
    int R,T;
    scanf("%d",&R);
    for(T=1;T<=R;T++)
    {
        scanf("%d%d",&n,&m);
        tot=0;
        for(int i=0;i<=n;i++) G[i].clear();
        while(m--)
        {
            int from,to,w;
            scanf("%d%d%d",&from,&to,&w);

            e[tot].from=from;e[tot].to=to;e[tot].w=w;
            G[e[tot].from].push_back(e[tot]);
            tot++;

            e[tot].from=to;e[tot].to=from;e[tot].w=w;
            G[e[tot].from].push_back(e[tot]);
            tot++;
        }
        SPFA();
        printf("Scenario #%d:\n",T);
        printf("%d\n",dis[n]);
        printf("\n");
    }
    return 0;
}

 

posted @ 2015-07-31 22:23  Fighting_Heart  阅读(138)  评论(0编辑  收藏  举报