Cod sursa(job #1351956)

Utilizator firutibogdanFiruti Bogdan-Cristian firutibogdan Data 21 februarie 2015 11:46:53
Problema Floyd-Warshall/Roy-Floyd Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 kb
#include<fstream>
using namespace std;
int infinit=20000000;
int b[3][5002][12];
int n,k,i,a[3][5005],max0,max1,max2,j,p;
int main()
{
     ifstream fin("joc13.in");
    ofstream fout("joc13.out");
    fin>>n>>k;
    for(i=1;i<=2;i++)
    {
        for(j=1;j<=n;j++)
        {
            fin>>a[i][j];
        }
    }
    for(p=0;p<=k;p++)
    {
        b[1][1][p]=-infinit;
        b[2][1][p]=-infinit;
    }
    b[1][1][1]=a[1][1];
    b[2][1][1]=a[1][1]+a[2][1];
    for(j=2;j<=n;j++)
    {
        max0=-infinit;
        max1=-infinit;
        for(p=2;p<=k;p++)
        {
            if(b[1][j][p]>-infinit)
            {
                b[1][j][p]=b[1][j-1][p-1]+a[1][j];
                if(b[1][j][p]>max0)
                {
                    max0=b[1][j][p];
                }
            }
            if(b[2][j][p]>-infinit)
            {
                b[2][j][p]=b[2][j-1][p-1]+a[2][j];
                if(b[2][j][p]>max1)
                {
                    max1=b[2][j][p];
                }
            }
        }
        b[1][j][1]=max1+a[1][j];
        b[2][j][1]=max0+a[2][j];
    }
    max2=-infinit;
    for(p=1;p<=k;p++)
    {
        if(b[2][n][p]>max2)
        {
            max2=b[2][n][p];
        }
    }
    fout<<max2;
    fin.close();
    fout.close();
    return 0;
}