Cod sursa(job #638089)

Utilizator paul_gabryelPaul Buda paul_gabryel Data 20 noiembrie 2011 18:49:30
Problema Ferma2 Scor 90
Compilator cpp Status done
Runda .com 2011 Marime 0.96 kb

#include <cstdio>
#include <fstream>

using namespace std;

#define N 1001

int l[N][N],c[N][N],d[N][N],sol,n,k;

inline void read ()
{
	ifstream in ("ferma2.in");
	in>>n>>k;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=i;++j)
		{
			in>>c[i][j];
			d[i][j]=d[i-1][j-1]+c[i][j];
			l[i][j]=l[i][j-1]+c[i][j];
			c[i][j]+=c[i-1][j];
		}
}

inline int dig (int m)
{
	int s=0;
	for(int i=n;m;--i,--m)
		s+=d[n][i];
	return s;
}

inline int col (int q,int m)
{
	int s=0;
	for(int i=1;m;++i,--m)
		s+=c[n][i]-c[q+i-1][i];
	return s;
}

inline int lin (int q,int w,int m)
{
	int s=0;
	for(int i=n;m;--i,--m)
		s+=l[i][i]-l[i][w]-(l[i][i]-l[i][i-q]);
	return s;
}

inline void solve ()
{
	for(int i=0;i<=k;++i)
		for(int j=0;i+j<=k;++j)
		{
			int s=0,q=k-(i+j);
			if(i)
				s+=dig(i);
			if(j)
				s+=col(i,j);
			if(q)
				s+=lin(i,j,q);
			if(s>sol)
				sol=s;
		}
}

int main ()
{
	read ();
	printf("\n");
	solve ();
	freopen ("ferma2.out","w",stdout);
	printf("%d",sol);
	return 0;
}