Cod sursa(job #1038263)

Utilizator CosminRusuCosmin Rusu CosminRusu Data 21 noiembrie 2013 11:19:57
Problema Team Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.22 kb
#include <fstream>
#include <vector>
#include <bitset>
#include <queue>
#include <string.h>

using namespace std;

const char infile[] = "team.in";
const char outfile[] = "team.out";

ifstream fin(infile);
ofstream fout(outfile);

const int MAXN = 505;
const int MAXP = 55;
const int oo = 0x3f3f3f3f;

typedef vector< pair<int, int> > Graph[MAXN];
typedef vector< pair<int, int> > :: iterator It;

const inline int min(const int &a, const int &b) { if( a > b ) return b;   return a; }
const inline int max(const int &a, const int &b) { if( a < b ) return b;   return a; }
const inline void Get_min(int &a, const int b)    { if( a > b ) a = b; }
const inline void Get_max(int &a, const int b)    { if( a < b ) a = b; }

Graph G;
int P, N, M, dest[MAXP], d[MAXN][MAXN], dp[MAXP][MAXP][MAXN];
queue<int> Q;
bitset<MAXN> inQ;

inline void BFs(int S, int best[]) {
    inQ.reset();

    Q.push(S);
    best[S] = 0;
    inQ[S] = true;

    while(!Q.empty()) {
        int Node = Q.front();
        Q.pop();
        inQ[Node] = false;
        for(It it = G[Node].begin(), fin = G[Node].end(); it != fin ; ++ it)
            if(best[it->first] > best[Node] + it->second) {
                best[it->first] = best[Node] + it->second;
                if(!inQ[it->first]){
                    Q.push(it->first);
                    inQ[it->first] = true;
                }
            }
    }
}

inline int Memo(const int i, const int j, const int k) {
    if(i > j)
        return 0;
    if(dp[i][j][k] != oo)
        return dp[i][j][k];
    if(i == j)
        return d[i][k];
    for(int l = i ; l <= j ; ++ l)
        Get_min(dp[i][j][k], d[l][k] + Memo(i, l-1, dest[l]) + Memo(l + 1, j, dest[l]));
    return dp[i][j][k];
}

int main() {
    fin >> P >> N >> M;
    for(int i = 1 ; i <= M ; ++ i) {
        int x, y, z;
        fin >> x >> y >> z;
        G[x].push_back(make_pair(y, z));
        G[y].push_back(make_pair(x, z));
    }
    for(int i = 1 ; i <= P ; ++ i)
        fin >> dest[i];
    memset(d, oo, sizeof(d));
    for(int i = 1 ; i <= P ; ++ i)
        BFs(dest[i], d[i]);
    memset(dp, oo, sizeof(dp));
    fout << Memo(1, P, 1);
    fin.close();
    fout.close();
    return 0;
}