Cod sursa(job #711310)

Utilizator alexalghisiAlghisi Alessandro Paolo alexalghisi Data 11 martie 2012 21:21:06
Problema Elimin Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.46 kb
#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int h[530][530];
bool gen[20];
int n,m,r,c;
vector<int> v;
long long maxim=-1<<30;

void back(int niv,int nr)
{
	if(niv==n+1)
	{	
		if(nr==c)
		{
			v.clear();
			
			long long S=0;
			for(int i=1;i<=m;i++)
			{
				S=0;
				for(int j=1;j<=n;j++)
					if(gen[j]==0)
						S+=h[i][j];
				v.push_back(S);
			}
			S=0;
			
			sort(v.begin(),v.end());
			for(int i=v.size()-1;i>=c;i--)
				S+=v[i];
			maxim=max(S,maxim);
		}
		return;
	}
	if(nr<=c)
	{
		gen[niv]=1;
		back(niv+1,nr+1);
		gen[niv]=0;
	}
	back(niv+1,nr);
}
void back2(int niv,int nr) // r linii
{
	if(niv==n+1)
	{	
		if(nr==r)
		{		
			v.clear();
			
			long long S=0;
			for(int i=1;i<=n;i++)
			{
				S=0;
				for(int j=1;j<=m;j++)
					if(gen[j]==0)
						S+=h[j][i];
				v.push_back(S);
			}
			S=0;
			
			sort(v.begin(),v.end());
			for(int i=v.size()-1;i>=c;i--)
				S+=v[i];
			maxim=max(S,maxim);
		}
		return;
	}
	if(nr<=r)
	{
		gen[niv]=1;
		back(niv+1,nr+1);
		gen[niv]=0;
	}
	back(niv+1,nr);
}

int main()
{	
	freopen("elimin.in","r", stdin);
	freopen("elimin.out","w", stdout);
	scanf("%d %d %d %d\n",&m,&n,&r,&c);
	for(int i=1;i<=m;i++)
		for(int j=1;j<=n;j++)
			scanf("%d ",&h[i][j]);
	if(n<=m) // nr coloanelor
	{
		back(1,0);
		printf("%lld",maxim);
	}
	else
	{
		back2(1,0);
		printf("%lld",maxim);
	}
	return 0;
}