Cod sursa(job #921278)

Utilizator Mihai22eMihai Ionut Enache Mihai22e Data 20 martie 2013 21:24:09
Problema Bibel Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.34 kb
#include <cstring>

#include <fstream>
#include <vector>
using namespace std;

const int MAX_N = 17;
const int INF = 0x3f3f3f3f;
const int MAX_C = (1 << MAX_N);

int N, t,  Nr = 1;
int Cost[MAX_N][MAX_N], A[MAX_N][2], B[MAX_C][MAX_N], Min[MAX_N], last[MAX_N][2];
vector < int > v[MAX_N];

inline int dist(int x1, int x2, int y1, int y2)
{
    return (x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2);
}

int main()
{
    ifstream f("bibel.in");
    ofstream g("bibel.out");

    memset(Cost, INF, sizeof(Cost));
    while(t != 2)
    {
        f >> t;
        if(!t)
            f >> A[N][0] >> A[N][1], ++N;
        else if(t == 1)
        {
             int Sum = INF;
             for(int i = 0; i < N; ++i)
                for(int j = 0; j < N; ++j)
                    if(i != j)
                    {
                        Cost[i][j] = Cost[j][i] = dist(A[i][0], A[j][0], A[i][1], A[j][1]);
                        v[i].push_back(j), v[j].push_back(i);
                    }
                
             for(int i = 0; i < (1 << N); ++i)
                 for(int j = 0; j < N; ++i)
                     B[i][j] = INF;
             for(int i = 0; i < N; ++i)
                for(int j = 0; j < Nr; ++j)
                    B[1 << i][i] = min(B[1 << i][i], Min[j] + dist(last[j][0], A[i][0], last[j][1], A[i][1]));
                
                   
             for(int i = 1; i < (1 << N); ++i)
                for(int j = 0; j < N; ++j)
                    if(i & (1 << j))
                    {
                        for(int k = 0; k < v[j].size(); ++k)
                            if(i & (1 << v[j][k])) 
                                B[i][j] = min(B[i][j], B[i ^ (1 << j)][ v[j][k] ] + Cost[ v[j][k] ][j]); 
                    }   

            int Tmp = INF;
            for(int i = 0; i < N; ++i)
                    Tmp = min(Tmp, B[(1 << N) - 1][i]);
            if(Tmp  < Sum)
                Sum = Tmp;
            for(int i = 0; i < N; ++i)
                    Min[i] = B[(1 << N) - 1][i];
            
            memset(Cost, INF, sizeof(Cost));
            Nr = N;
            for(int i = 0; i < Nr; ++i)
                last[i][0] = A[i][0], last[i][1] = A[i][1];
            for(int i = 0; i < N; ++i)
                v[i].clear();
            N = 0;
            
            g << Sum << '\n';
        }   
    } 
    
    f.close();
    g.close();

    return 0;
}