Cod sursa(job #131214)

Utilizator savimSerban Andrei Stan savim Data 3 februarie 2008 13:43:00
Problema Elimin Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.02 kb
#include <stdio.h>

#define nmax 31
#define mmax 7926

int p,q,max=0,i,j,k,n,m,r,c,nr,co;
int a[nmax][mmax];
int fol[nmax],s[mmax],o[mmax];
void inter(int p, int q)
{
     int r=(p+q)/2,x,y,k=0,i;
     x=p;y=r+1;
     while (x<=r && y<=q)     
     {
         k++;
         if (s[x]<s[y])      
         {
            o[k]=s[x];x++;
         }
         else
         {
            o[k]=s[y];y++;    
         }
     }
     while (x<=r)
     {
         k++;o[k]=s[x];x++;      
     }
     while (y<=q)
     {
         k++;o[k]=s[y];y++;      
     }
     k=0;
     for (i=p; i<=q; i++)
     {
         k++;s[i]=o[k];    
     }

}     
void merge(int p, int q)
{
     int r=(p+q)/2;
     if (p==q) return;
     merge(p,r);
     merge(r+1,q);    
     inter(p,q);
     if (p+1>=q) return;
}
int main()
{

	freopen("elimin.in","r",stdin);
	freopen("elimin.out","w",stdout);

	scanf("%d%d%d%d",&n,&m,&r,&c);
    if (n<=m)
	for (i=1; i<=n; i++)
		for (j=1; j<=m; j++)
			scanf("%d",&a[i][j]);
    else
    {
		p=1;q=0;
		for (i=1; i<=n; i++)
			for (j=1; j<=n; j++)
			{
				if (q<m) q++;
				else { p++;q=1;}
				scanf("%d",&a[q][p]);
			}
		k=n;n=m;m=k;k=r;r=c;c=k;
	}

	   for (k=0; k<=(1<<n)-1; k++)
	   {
		   for (i=1; i<=n; i++)
			   fol[i]=0;

		   nr=0;co=k;
		   while (co>0)
		   {
			   nr+=co%2;co=co/2;
		   }
		   if (nr==r)
		   {
			   for (i=1; i<=m; i++)
				  s[i]=0;
			   co=k;i=0;
			   while (co>0)
			   {
				  i++;
				  fol[i]=co%2;co=co/2;
			   }

			   for (j=1; j<=m; j++)
			   {
				   s[j]=0;
				   for (i=1; i<=n; i++)
						if (fol[i]==0) s[j]+=a[i][j];
			   }
/*			  for (i=1; i<=m-1; i++)
				   for (j=i+1; j<=m; j++)
					  if (s[i]>s[j])
					  {
						int x=s[i];
						s[i]=s[j];
						s[j]=x;
					  } */

               merge(1,m);  
			   int sum=0;
			   for (i=c+1; i<=m; i++)
					sum+=s[i];
			   if (sum>max) max=sum;
		   }
	   }

	printf("%d\n",max);
        
    
    
    return 0;    
}