Cod sursa(job #919232)

Utilizator CosminRusuCosmin Rusu CosminRusu Data 19 martie 2013 15:17:47
Problema Ferma2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>

using namespace std;

ifstream cin("ferma2.in");
ofstream cout("ferma2.out");
int a[1002][1002], s[1002][1002];
int main()
{
    int n, k, maxim=0;
    cin>>n>>k;
    for(int i=1;i<=n;++i)
        {
            for(int j=1;j<=i;++j)
            {
                cin>>a[i][j];
                s[i][j]=s[i][j-1]+a[i][j];
            }
            maxim+=a[i][i];
        }
    cin.close();
    int bestSum=(1<<31)-1;
    for(int i=n-k;i<=n;++i)
        for(int j=n-k;j<=i;++j)
        {
            int sum=0;
            for(int ii=i, jj=j, jj1=1 ; jj1<=n-k ; ++jj1, --ii, --jj)
                sum+=s[ii][jj]-s[ii][jj-(n-k-jj1+1)];
            bestSum=min(bestSum, sum);
        }
    cout<<maxim-bestSum<<"\n";
    cout.close();
    return 0;
}