Cod sursa(job #2922750)

Utilizator Iordache_CezarIordache Cezar Iordache_Cezar Data 9 septembrie 2022 21:34:48
Problema ADN Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.99 kb
#include <bits/stdc++.h>
#define INF 1e9
#define LGMAX 300008

using namespace std;
ifstream fin ("adn.in");
ofstream fout ("adn.out");

int n, cnt;
int len[LGMAX], cost[20][20], dp[1<<20][20], pre[1<<20][20];
bool e_inclus[20];
string adn[20], adn2[20];
vector <int> sol;

void citire();
int KMP(string a, string b);
bool caut_subsir(string a, string b);
void sterge();
void create_graph();
void ciclu_hamiltonian();

int main()
{
    citire();
    create_graph();
    ciclu_hamiltonian();
    return 0;
}

void citire()
{
    fin >> n;
    for (int i = 0; i < n; i++)
        fin >> adn[i];
}

int KMP(string a, string b)
{
    string s = b + a;
    for (int i = 1; i < s.size(); i++)
    {
        int j = len[i-1];
        while (j > 0 && s[i] != s[j])
            j = len[j-1];
        if (s[i] == s[j])
            j++;
        len[i] = j;
    }

    return len[s.size()-1];
}

bool caut_subsir(string a, string b)
{
    if (b.size() > a.size())
        return 0;
    string s = b + a;
    for (int i = 1; i < s.size(); i++)
    {
        int j = len[i-1];
        while (j > 0 && s[i] != s[j])
            j = len[j-1];
        if (s[i] == s[j])
            j++;
        len[i] = j;
        if (len[i] == b.size() && i >= 2*b.size()-1)
            return 1;
    }
    return 0;
}

void create_graph()
{
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            if (i != j)
            {
                if (i == 1 && j == 0)
                    i = 1;
                if (e_inclus[j])
                    continue;
                e_inclus[j] = caut_subsir(adn[i], adn[j]);
                if (e_inclus[j])
                {
                    if (adn[i].size() == adn[j].size())
                    {
                        e_inclus[min(i, j)] = 0;
                        e_inclus[max(i, j)] = 1;
                    }
                    continue;
                }
                //int c = KMP(adn[i], adn[j]);
                //cost[i][j] = c;
            }
    for (int i = 0; i < n; i++)
        if (e_inclus[i] == 0)
        {
            adn2[cnt] = adn[i];
            cnt++;
        }
    n = cnt;
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            if (i != j)
            {
                int c = KMP(adn2[i], adn2[j]);
                cost[i][j] = c;
            }
}

void ciclu_hamiltonian()
{
    int maxim = (1 << n);
    for (int mask = 0; mask < maxim; mask++)
        for (int j = 0; j < n; j++)
            dp[mask][j] = INF;

    for (int j = 0; j < n; j++)
        dp[1<<j][j] = adn2[j].size();

    for (int mask = 0; mask < maxim; mask++)
        for (int j = 0; j < n; j++)
        {
            if (mask == 3)
                int ok = 1;
            if ((mask & (1 << j)))
                for (int k = 0; k < n; k++)
                    if (k != j && (mask & (1 << k)))
                        {
                            int sz = adn2[j].size();
                            if (dp[mask ^ (1<<j)][k] + sz - cost[k][j] < dp[mask][j])
                            {
                               dp[mask][j]  = dp[mask ^ (1<<j)][k] + sz - cost[k][j];
                               pre[mask][j] = k;
                            }
                        }
        }
    int minim = INF;
    maxim--;
    int poz;
    for (int i = 0; i < n; i++)
        if (dp[maxim][i] < minim)
        {
            minim = dp[maxim][i];
            poz = i;
        }
    sol.push_back(poz);
    for (int i = 1; i < n; i++)
    {
        int copie = poz;
        poz = pre[maxim][copie];
        maxim = maxim ^ (1 << copie);
        sol.push_back(poz);
    }

    //for (int i = sol.size()-1; i >= 0; i--)
        //fout << sol[i] << ' ';
    fout << adn2[sol[sol.size()-1]];
    for (int i = sol.size()-2; i >= 0; i--)
    {
        for (int j = cost[sol[i+1]][sol[i]]; j < adn2[sol[i]].size(); j++)
            fout << adn2[sol[i]][j];
    }
}