Cod sursa(job #2538859)

Utilizator MoodyFaresFares Mohamad MoodyFares Data 5 februarie 2020 11:28:25
Problema Gather Scor 70
Compilator cpp-64 Status done
Runda simulare_miri Marime 2.96 kb
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

const int MAX_K = 16;
const int MAX_N = 750;
const int INF = 0x7fffffff;

struct Edge {
  int node;
  int length;
  int cap;

  bool operator <(const Edge &other) const {
    return this->length > other.length;
  }
};

int n;
int cell[1 + MAX_K];
int getCon[1 + MAX_N];
vector<Edge> g[1 + MAX_N];
int minPath[1 + MAX_K][1 + MAX_K][1 + MAX_K];
bool vis[MAX_N];
int dp[1 + MAX_K][1 + (1 << MAX_K)];

void dijkstra_minPath(int root, int minCap) {
  priority_queue<Edge> pq;

  pq.push({root, 0});
  while (!pq.empty()) {
    Edge temp = pq.top();
    pq.pop();

    int node, dist;

    node = temp.node;
    dist = temp.length;

    if (vis[node])
      continue;

    vis[node] = true;
    if (getCon[node] != 0)
      minPath[getCon[root]][getCon[node]][minCap] = dist;

    for (auto e : g[node]) {
      int u, l, cap;

      u = e.node;
      l = e.length;
      cap = e.cap;

      if (!vis[u] && cap >= minCap)
        pq.push({u, dist + l});
    }
  }

  for (int i = 1; i <= n; i++)
    vis[i] = false;
}

int main() {
  freopen("gather.in", "r", stdin);
  freopen("gather.out", "w", stdout);

  int k, m;

  scanf("%d%d%d", &k, &n, &m);

  cell[1] = 1;
  getCon[1] = 1;
  for (int i = 2; i <= k + 1; i++) {
    scanf("%d", &cell[i]);
    getCon[cell[i]] = i;
  }
  k++;

  for (int i = 1; i <= m; i++) {
    int u, v, l, c;

    scanf("%d%d%d%d", &u, &v, &l, &c);
    g[u].push_back({v, l, c});
    g[v].push_back({u, l, c});
  }

  for (int i = 1; i <= k; i++) {
    for (int j = 1; j <= k; j++) {
      for (int minCap = 0; minCap <= k - 1; minCap++)
        minPath[i][j][minCap] = INF;
    }
  }

  for (int i = 1; i <= k; i++) {
    for (int minCap = 0; minCap <= k - 1; minCap++)
      dijkstra_minPath(cell[i], minCap);
  }

  for (int i = 1; i <= k; i++) {
    for (int mask = 0; mask < (1 << k); mask++)
      dp[i][mask] = INF;
  }

  for (int i = 1; i <= k; i++)
    dp[i][1] = minPath[1][i][0];

  for (int mask1 = 0; mask1 < (1 << (k - 1)); mask1++) {
    int mask = mask1 * 2 + 1;
    if (!(mask & 1))
      continue;

    int countBytes = 0;
    for (int byte = 1; byte < k; byte++) {
      if (mask & (1 << byte))
        countBytes++;
    }

    for (int i = 2; i <= k; i++) {
      for (int byte = 1; byte < k; byte++) {
        if (mask & (1 << byte)) {
          for (int j = 2; j <= k; j++) {
            if (dp[j][mask ^ (1 << byte)] != INF && minPath[j][byte + 1][countBytes - 1] != INF && minPath[byte + 1][i][countBytes] != INF)
              dp[i][mask] = min(dp[i][mask], dp[j][mask ^ (1 << byte)] + minPath[j][byte + 1][countBytes - 1] * countBytes + minPath[byte + 1][i][countBytes] * (countBytes + 1));
          }
        }
      }
    }
  }

  int ans;

  ans = INF;
  for (int i = 2; i <= k; i++)
    ans = min(ans, dp[i][(1 << k) - 1]);

  printf("%d\n", ans);
  return 0;
}