Cod sursa(job #2965521)

Utilizator AACthAirinei Andrei Cristian AACth Data 15 ianuarie 2023 15:06:30
Problema ADN Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.25 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("adn.in");
ofstream g("adn.out");
#define cin f
#define cout g

int n;
vector < string > adn;
void read()
{
	f >> n;
	adn.resize(n);
	for(auto &it : adn)
		f >> it;
}
struct elem{
	int val,nod,mask;
};
bool kth(int mask, int k)
{
    return (mask & (1 << k)) != 0;
}
bool substringKMP(string &small, string &big)
{
	if(small.size() > big.size())return false;
	int target = small.size();
	string s = small + "#" + big;
	vector <int> pi (s.size ());
    pi[0] = 0;
    for (int i = 1; i < s.size (); i++) {
        int cur = pi[i - 1];
        while (cur && s[i] != s[cur])
            cur = pi[cur - 1];
        if (s[i] == s[cur])
            cur++;
        pi[i] = cur;
        if (pi[i] == target)
            return true;
    }
    return false;
}

void trimADN()
{
	// remove duplicates
	sort(adn.begin(), adn.end());
	adn.erase(unique(adn.begin(), adn.end()),adn.end());
	vector < bool > bad(adn.size());
	for(int i=0;i<adn.size();++i)
		for(int j = i+1;j<adn.size();++j)
		{
			if(substringKMP(adn[i],adn[j]))
				bad[i] = true;
			if(substringKMP(adn[j],adn[i]))
				bad[j] = true;
		}
	vector < string > finalADN;
	for(int i=0;i<adn.size();++i)
		if(!bad[i])
			finalADN.push_back(adn[i]);
	adn = finalADN;
}
int determineCostAdn(string& s1,string& s2)
{
	string s = s2 + "#" + s1;
	vector <int> pi (s.size ());
    pi[0] = 0;
    for (int i = 1; i < s.size(); i++) {
        int cur = pi[i - 1];
        while (cur && s[i] != s[cur])
            cur = pi[cur - 1];
        if (s[i] == s[cur])
            cur++;
        pi[i] = cur;
    }
    return pi[s.size()-1];
}
int cost[20][20];
int dp[19][(1 << 18) + 1];
elem parent[19][(1 << 18) + 1];
void precomputeCosts()
{
	int i,j;
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			if(i != j)
				cost[i][j] = determineCostAdn(adn[i],adn[j]);
}
void solve()
{
    trimADN();
    n = adn.size();
    precomputeCosts();
    int all = (1 << n) - 1;
   	parent[0][0] = {-1,-1,-1};
	for(int mask = 0;mask <= all;mask++)
        for(int nod = 0;nod<n;nod++)
        {
        	elem state;
        	state.val = dp[nod][mask];
        	state.nod = nod;
        	state.mask = nod;
            for(int nxt = 0;nxt < n;nxt ++)
                if (!kth(mask,nxt))
                {
                    int ncost = state.val + cost[nod][nxt];
                    int nmask = mask | (1 << nxt);
                    if(ncost > dp[nxt][nmask])
                    {
                        dp[nxt][nmask] = ncost;
                        parent[nxt][nmask] = state;
                    }
                }
       	}
    elem best = {-1,-1,-1};
	for(int i=0;i<n;i++)
		if(dp[i][all] > best.val)
			best = {dp[i][all],i,all};
	vector < int > sol;
	for(elem nod = best;nod.nod != -1;nod = parent[nod.nod][nod.mask])
		sol.push_back(nod.nod);
    reverse(sol.begin(), sol.end());
	// rebuild the solution
	cout << adn[sol[0]];
	for(int i = 1;i<sol.size();i++)
	{
		int dif = cost[sol[i-1]][sol[i]];
		for(int j = dif;j < adn[sol[i]].size();++j)
			cout<<adn[sol[i]][j];
	}

}
void restart()
{

}
int32_t main()
{
    {
        read();
        solve();
        restart();
    }
    return 0;
}