Dijkstra 算法求非負權圖的最短路徑

/**/
/*Dijkstra*/
#include <iostream>
#include <fstream>
using namespace std;
#define MAXN 1000000

fstream fin;
void dijkstra(int **a,int n,int *d);
int main()
{
    fin.open("1008.txt",ios::in);
    int **a;
 
    int n;
    fin>>n;
    a=new int*[n];
    for(int i=0;i<n;i++)
      a[i]=new int[n];
      
    int *d=new int[n];
    
    for(int i=0;i<n;i++)
      for(int j=0;j<n;j++)
        a[i][j]=0;
        
    int k;
    fin>>k;
    for(int i=0;i<k;i++)
    {
        int x,y,weight;
        fin>>x>>y>>weight;
        a[x-1][y-1]=weight;
    }
    
    for(int i=0;i<n;i++)
      {
          for(int j=0;j<n;j++)
           cout<<a[i][j]<<" ";
          cout<<endl;
       }    
        
    dijkstra(a,n,d);
    cout<<"----------------------"<<endl;
    for(int i=0;i<n;i++)
       cout<<d[i]<<" ";
    cout<<endl; 
     
    system("pause");
    return 0;
}

void dijkstra(int **a,int n,int *d)
{
    bool *f=new bool[n];
    memset(f,0,sizeof(f)+1);
    /*ÕÒµ½d×îСµÄµã*/
    for(int i=1;i<n;i++)
      d[i]=MAXN;
    
    d[0]=0;
    int u=0;
    
    for(int i=0;i<n;i++)
    { 
        int min=MAXN;
        for(int j=0;j<n;j++)
        {
           if(!f[j]&&d[j]<min)
            {
               min=d[j];
               u=j;
            }
        } 
    
        f[u]=1;
        for(int j=0;j<n;j++)
        {
            //¸üÐÂ
            if(a[u][j]&&!f[j]) 
            {
                if(d[j]>d[u]+a[u][j])
                   d[j]=d[u]+a[u][j];
            }
             
        }   
    } 
    delete f;
}

/*1008.txt*/

5
7
1 2 2
1 3 6
1 4 3
2 3 4
2 4 5
4 5 7
3 5 8

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章