HDU 3790 (Djikstra)

题目链接如下:

http://acm.hdu.edu.cn/showproblem.php?pid=3790

这题用最短路径Djikstra就可以解决,用Floyd会超时!

代码如下:

#include <iostream>
using namespace std;

const int INF = 999999999;
const int N = 1001;
struct
{
  int distance;
  int price;
} map[N][N];
int dist[N];
int price[N];
int visited[N];

void djikstra(int s, int t, int n);

int main()
{
  int n, m;
  int i, j, a, b, d, p;
  int s, t;
  while(scanf("%d%d", &n, &m) != EOF && (n || m))
  {
    for(i = 1; i <= n; i++)
    {
      for(j = 1; j <= n; j++)
      {
        map[i][j].distance = INF;
        map[i][j].price = INF;
      }
    }
    for(i = 1; i <= n; i++)
    {
      map[i][i].distance = 0;
      map[i][i].price = 0;
    }
    for(i = 1; i <= m; i++)
    {
      scanf("%d%d%d%d", &a, &b, &d, &p);
      if(map[a][b].distance > d)
       {
        map[a][b].distance = map[b][a].distance = d;
        map[a][b].price = map[b][a].price = p;
      }
    }
    scanf("%d%d", &s, &t);
    djikstra(s, t, n);
  }
  return 0;
}

void djikstra(int s, int t, int n)
{
  int i, j;
  int min, k;
  memset(visited, 0, sizeof(visited));
  for(i = 1; i <= n; i++)
  {
    dist[i] = map[s][i].distance;
    price[i] = map[s][i].price;
  }
  visited[s] = 1;
  for(i = 1; i <= n; i++)
  {
    min = INF;
    for(j = 1; j <= n; j++)
    {
      if(!visited[j] && dist[j] < min)
      {
        min = dist[j];
        k = j;
      }
    }
    if(min == INF)
    {
      break;
    }
    visited[k] = 1;
    for(j = 1; j <= n; j++)
    {
      if(!visited[j])
      {
        if(dist[j] > dist[k] + map[k][j].distance)
        {
          dist[j] = dist[k] + map[k][j].distance;
          price[j] = price[k] + map[k][j].price;
        }
        else if(dist[j] == dist[k] + map[k][j].distance)
        {
          if(price[j] > price[k] + map[k][j].price)
          {
            price[j] = price[k] + map[k][j].price;
          }
        }
      }
    }
  }
  printf("%d %d\n", dist[t], price[t]);
}

你可能感兴趣的:(HDU)