hdu 1874 畅通工程续

来源:互联网 发布:宽带连接软件下载 编辑:程序博客网 时间:2024/06/06 20:09

10分钟写的代码,改了2个小时。。。。。

一直无限WA。。。。终于是AC了

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
#define ls rt<<1
#define rs ls1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-9)
#define type int
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define inf 100000007
#define pi acos(-1.0)
#define M 1000+5
int mp[M][M],fn[M],d[M];
void Init(){
int i,j;
for(i=0;i<M;i++){
for(j=0;j<M;j++)
mp[i][j]=inf;
d[i]=inf;
fn[i]=0;
}
}
int Dij(int n,int ss,int dd){
int i,j,minx,k;
for(i=0;i<n;i++)d[i]=mp[ss][i];
d[ss]=0;fn[ss]=1;
for(i=1;i<n;i++){
minx=inf;k=0;
for(j=0;j<n;j++)
if(fn[j]==0&&d[j]<minx){
minx=d[j];k=j;
}
fn[k]=1;
for(j=0;j<n;j++)
if(fn[j]==0&&d[j]>d[k]+mp[j][k])
d[j]=d[k]+mp[j][k];
}
return d[dd];
}
int main(){
    int i,j,n,m,x,y,z,ss,dd;
    while(scanf("%d %d",&n,&m)!=EOF){
Init();
while(m--){
scanf("%d %d %d",&x,&y,&z);
if(z<mp[x][y])mp[x][y]=mp[y][x]=z;
}
scanf("%d %d",&ss,&dd);
x=0;
if(ss!=dd){
x=Dij(n,ss,dd);
x=x<inf ? x:-1;
}
printf("%d\n",x);
}
    return 0;
}