Cod sursa(job #2520833)

Utilizator ivan.tudorIvan Tudor ivan.tudor Data 9 ianuarie 2020 20:11:05
Problema ADN Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.06 kb
#include <iostream>
#include<cstdio>
#include<fstream>
#include<string>
using namespace std;
ifstream fin("adn.in");
ofstream fout("adn.out");
const int N=18;
int cost[N][N];
string s[N];
int dp[N][1<<N];
char prv[N][1<<N];
const int LENMAX=30005;
const int INF=300000005;
int lsp[LENMAX];
int included[N];
string aux;
int calculatelps(string a,string b,int id1,int id2){
  aux=a+b;
  lsp[0]=lsp[1]=0;
  for(int i=2;i<aux.size();i++){
    int poz=lsp[i-1];
    while(aux[poz+1]!=aux[i] && poz){
      poz=lsp[poz];
    }
    if(aux[poz+1]==aux[i]){
      lsp[i]=poz+1;
    }
    else
      lsp[i]=0;
    if(lsp[i]==a.size()-1){
      included[id2]|=(1<<id1);
    }
  }
  return lsp[aux.size()-1];
}
string subs;
int n;
void afis(int id,int mask){
  if(prv[id][mask]==-1){
    subs=s[id].substr(1);
    fout<<subs;
    return;
  }
  int noumask=mask;
  for(int i=0;i<n;i++){
    if(noumask&(1<<i) && included[id]&(1<<i))
      noumask^=(1<<i);
  }
  afis(prv[id][mask],noumask);
  subs=s[id].substr(s[id].size()-cost[prv[id][mask]][id]);
  fout<<subs;
}
int main()
{
  fin>>n>>ws;
  for(int i=0;i<n;i++){
    fin>>s[i];
    s[i]=" "+s[i];
  }
  for(int i=0;i<n;i++){
    included[i]|=(1<<i);
    for(int j=0;j<n;j++){

      if(i!=j){
        cost[i][j]=s[j].size()-1-calculatelps(s[j],s[i],j,i);
      }
    }
  }
  for(int i=0;i<n;i++){
    for(int m=0;m<1<<n;m++){
      dp[i][m]=INF;
    }
  }
  for(int i=0;i<n;i++){
    dp[i][included[i]]=s[i].size()-1;
    prv[i][included[i]]=-1;
  }
  for(int mask=0;mask<1<<N;mask++){
    for(int i=0;i<n;i++){
      if(mask & (1<<i) && dp[i][mask]!=INF){
        for(int j=0;j<n;j++){
          if(!(mask&(1<<j)) && dp[j][mask|included[j]]>dp[i][mask]+cost[i][j]){
            dp[j][mask|included[j]]=dp[i][mask]+cost[i][j];
            prv[j][mask|included[j]]=i;
          }
        }
      }
    }
  }
  int sol=INF,idsol;
  for(int i=0;i<n;i++){
    if(sol>dp[i][(1<<n)-1]){
      sol=dp[i][(1<<n)-1];
      idsol=i;
    }
  }
  afis(idsol,(1<<n)-1);
  return 0;
}