Problem p00200 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <iostream>
#include <map>
#include <queue>
#include <vector>
using namespace std;
#define fr first
#define sc second
typedef pair<int, int> Pi;
struct edge {
  int to, element /*時間もしくは金額*/;
};
typedef vector<vector<edge>> Graph;

int dijkstr(Graph &info, int s, int g) {
  vector<int> min_cost;
  priority_queue<Pi, vector<Pi>, greater<Pi>> que;
  min_cost.resize(info.size(), -1);
  que.push(Pi(0, s)); // firstが時間(金額) , secondが場所
  min_cost[s] = 0;
  while (!que.empty()) {
    Pi p = que.top();
    que.pop();
    if (p.sc == g)
      return p.fr; // ゴールに着いたら
    if (p.fr > min_cost[p.sc])
      continue; // 現在地が前にきたときの最小より大きかった場合continue
    for (int i = 0; i < info[p.sc].size(); i++) { // 現在地からの道の数だけ回る
      edge &e = info[p.sc][i];
      if (min_cost[e.to] == -1 ||
          p.fr + e.element <
              min_cost
                  [e.to]) { // 次の場所はまだコストを調べていない ||
                            // 次の場所の最小値よりも移動後のコストが小さい場合。
        min_cost[e.to] = p.fr + e.element;
        que.push(Pi(min_cost[e.to], e.to));
      }
    }
  }
  return min_cost[g];
}

int main() {
  int n, m;
  while (1) {
    cin >> n >> m;
    if (n == 0 && m == 0)
      break;
    Graph info[2];
    info[0].resize(n);
    info[1].resize(n);

    int a, b, COST, TIME;
    for (int i = 0; i < n; i++) {
      cin >> a >> b >> COST >> TIME;
      info[0][a].push_back((edge){b, COST});
      info[0][b].push_back((edge){a, COST});
      info[1][a].push_back((edge){b, TIME});
      info[1][b].push_back((edge){a, TIME});
    }

    int k, p, q, r;
    cin >> k;
    for (int i = 0; i < k; i++) {
      cin >> p >> q >> r;
      cout << dijkstr(info[r], p, q) << endl;
    }
  }
}

Error: TLE

Fixed submission:

#include <iostream>
#include <map>
#include <queue>
#include <vector>
using namespace std;
#define fr first
#define sc second
typedef pair<int, int> Pi;
struct edge {
  int to, element /*時間もしくは金額*/;
};
typedef vector<vector<edge>> Graph;

int dijkstr(Graph &info, int s, int g) {
  vector<int> min_cost;
  priority_queue<Pi, vector<Pi>, greater<Pi>> que;
  min_cost.resize(info.size(), -1);
  que.push(Pi(0, s)); // firstが時間(金額) , secondが場所
  min_cost[s] = 0;
  while (!que.empty()) {
    Pi p = que.top();
    que.pop();
    if (p.sc == g)
      return p.fr; // ゴールに着いたら
    if (p.fr > min_cost[p.sc])
      continue; // 現在地が前にきたときの最小より大きかった場合continue
    for (int i = 0; i < info[p.sc].size(); i++) { // 現在地からの道の数だけ回る
      edge &e = info[p.sc][i];
      if (min_cost[e.to] == -1 ||
          p.fr + e.element <
              min_cost
                  [e.to]) { // 次の場所はまだコストを調べていない ||
                            // 次の場所の最小値よりも移動後のコストが小さい場合。
        min_cost[e.to] = p.fr + e.element;
        que.push(Pi(min_cost[e.to], e.to));
      }
    }
  }
  return min_cost[g];
}

int main() {
  int n, m;
  while (1) {
    cin >> n >> m;
    if (n == 0 && m == 0)
      break;
    Graph info;
    info.resize(n);

    int a[3010], b[3010], COST[3010], TIME[3010];
    for (int i = 0; i < n; i++) {
      cin >> a[i] >> b[i] >> COST[i] >> TIME[i];
      // info[a].push_back( edge{b, COST, TIME} );
      // info[b].push_back( edge{a, COST, TIME} );
    }

    int k, p, q, r;
    cin >> k;
    for (int i = 0; i < k; i++) {
      cin >> p >> q >> r;

      if (r == 0) { // 最小金額
        for (int i = 0; i < n; i++) {
          info[a[i]].push_back((edge){b[i], COST[i]});
          info[b[i]].push_back((edge){a[i], COST[i]});
        }
      } else if (r == 1) { // 最短時間
        for (int i = 0; i < n; i++) {
          info[a[i]].push_back((edge){b[i], TIME[i]});
          info[b[i]].push_back((edge){a[i], TIME[i]});
        }
      }
      cout << dijkstr(info, p, q) << endl;
    }
  }
}