首页 > 代码库 > ZOJ--1655--Transport Goods【dijkstra】

ZOJ--1655--Transport Goods【dijkstra】

题意:某国首都正被攻打,需要运送物资到首都,告诉你n个点,编号1~n,n是首都,剩下的点各有wi重量的物资,m条路,每条路有个货物损失比例,现需要求出最多能运送多少货物到首都。


其实转换一下就是一个最短路问题,边的权值是损失比例,找损失比例最小的那条路,则能运送的货物最多。

dist数组存放运成功的比例,初始化为0表示运不成。


WA了N发,各种double类型都用int定义的,而且它给的样例即使定义成int对结果也没影响。。。

#include<cstring>
#include<string>
#include<fstream>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cctype>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<stack>
#include<ctime>
#include<cstdlib>
#include<functional>
#include<cmath>
using namespace std;
#define PI acos(-1.0)
#define MAXN 1010
#define eps 1e-11
#define INF 0x7FFFFFFF
#define long long ll;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

double edge[105][105];
double dist[105];
int vis[105];
int w[105];
int n, m;
void dijkstra(int v){
    int i, j;
    memset(vis,0,sizeof(vis));
    memset(dist,0,sizeof(dist));
    vis[v] = 1;
    dist[v] = 1;
    int k = v;
    for(i=0;i<n;i++){
        for(j=1;j<=n;j++){
            if(!vis[j]&&edge[k][j]!=-1&&dist[j]<dist[k]*edge[j][k]){
                dist[j] = dist[k] * edge[j][k];
            }
        }
        double temp = 0;
        for(j=1;j<=n;j++){
            if(!vis[j]&&dist[j]-temp>eps){
                k = j;
                temp = dist[j];
            }
        }
        vis[k] = 1;
    }
}
int main(){
    int i, j, a, b;
    double c, ans;
    while(scanf("%d%d",&n,&m)!=EOF){
        ans = 0;
        for(i=0;i<n-1;i++){
            scanf("%d",&w[i+1]);
        }
        for(i=1;i<=n;i++){
            for(j=1;j<=n;j++){
                edge[i][j] = -1;
            }
        }
        for(i=0;i<m;i++){
            scanf("%d%d%lf",&a,&b,&c);
            if(1-c>edge[a][b])
                edge[a][b] = edge[b][a] = 1 - c;
        }
        dijkstra(n);
        for(i=1;i<n;i++)    ans += dist[i]*w[i];
        printf("%.2lf\n",ans);
    }
    return 0;
}