Cod sursa(job #2633977)

Utilizator CraniXortDumitrescul Eduard CraniXort Data 9 iulie 2020 14:44:44
Problema Team Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.07 kb
#include <bits/stdc++.h>
#define maxn 55
#define maxv 505

std::ifstream fin("team.in");
std::ofstream fout("team.out");

int x[maxn];
int dist[maxv][maxv];
int dest[maxn];
int dp[maxn][maxn][maxn];

void computeMinDist(int V){
    int i, j, k;
    for (k=1; k<=V; k++){
        for (i=1; i<=V; i++){
            if (i == k) continue;
            for (j=1; j<=V; j++){
                if (j == k or j == i) continue;
                if (dist[i][k] == -1 or dist[k][j] == -1) continue;
                if (dist[i][j] == -1) dist[i][j] = 1e9;

                dist[i][j] = std::min (dist[i][j], dist[i][k] + dist[k][j]);
            }
        }
    }
}

int main()
{
    int n, V, E, src, dst, cost, i, j, k, u;
    fin >> n >> V >> E;

    for (i=1; i<=V; i++)
    for (j=1; j<=V; j++)
        dist[i][j] = -1;

    for (i=1; i<=V; i++)
        dist[i][i] = 0;

    for (i=0; i<E; i++) {
        fin >> src >> dst >> cost;
        dist[src][dst] = cost;
        dist[dst][src] = cost;
    }

    computeMinDist(V);

    /*
    for (i=1; i<=V; i++, fout << '\n')
        for (j=1; j<=V; j++)
        fout << dist[i][j] << ' ';
        */

    for (i=1; i<=n; i++)
        fin >> dest[i];

    for (i=1; i<=n; i++)
        dp[i][i][i] = 0;

    for (int l=0; l<=n; l++)
    for (i=1; i<=n; i++){
        j = i + l;
        for (k=1; k<=n; k++){
            if (k == i and k == j)
            continue;
            dp[i][j][k] = 1e9;
            for (u=i; u<=j; u++){
                //std::cout << dp[i][u-1][u] << ' ' << dp[u+1][j][i]  << ' ' << dist[dest[u]][dest[k]] << dp[i][u-1][u] + dp[u+1][j][i] + dist[dest[u]][dest[k]] << '\n';
                //std::cout << std::min (dp[i][j][k], dp[i][u-1][u] + dp[u+1][j][u] + dist[dest[u]][dest[k]]) << '\n';
                dp[i][j][k] = std::min (dp[i][j][k], dp[i][u-1][u] + dp[u+1][j][u] + dist[dest[u]][dest[k]]);
            }
        }
    }



    int ans = 1e9;
    for (i=1; i<=n; i++)
    ans = std::min (ans, dp[1][n][i] + dist[1][dest[i]]);

    fout << ans;

    return 0;
}