Cod sursa(job #1469604)

Utilizator otto1Palaga Vicentiu-Octavian otto1 Data 8 august 2015 21:33:27
Problema Zone Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.02 kb
#include<stdio.h>
#include<algorithm>
using namespace std;

FILE *f = fopen("zone.in","r");
FILE *g = fopen("zone.out","w");

#define Max9 11
#define MaxN 600
#define ll long long

int N;
ll C[Max9];
int viz[Max9];
int A[MaxN][MaxN];
ll Best[MaxN][MaxN];
int Linie[5];
int Coloana[5];
bool gata;

void citire(void)
{
    fscanf(f,"%d",&N);
    for(int i=1;i<=9;i++)
        fscanf(f,"%lld ",&C[i]);
    for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++)
            fscanf(f,"%d ",&A[i][j]);
}

void Preprocesare(void)
{
    for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++)
            Best[i][j] = 1LL*A[i][j] + Best[i-1][j]+Best[i][j-1]-Best[i-1][j-1];
}

inline ll SumaSubmatrice(int x1,int y1,int x2,int y2)
{
    return Best[x2][y2]+Best[x1][y1]-Best[x1][y2]-Best[x2][y1];
}

inline int ExistaInVector(ll c)
{
    for(int i=1;i<=9;i++)
        if(!viz[i] && C[i] == c)
            return i;

    return 0;
}

inline void Initializare(void)
{
    for(int i=1;i<=9;i++)
        viz[i] = 0;
}

inline bool EsteBunaDistribuirea(int l1,int l2,int c1,int c2)
{
    int a;
    Linie[1] = l1; Linie[2] = l2; Linie[3] = Coloana[3] = N;
    Coloana[1] = c1; Coloana[2] = c2;

    for(int i=1;i<=3;i++)
        for(int j=1;j<=3;j++)
        {
            a = ExistaInVector(SumaSubmatrice(Linie[i-1],Coloana[j-1],Linie[i],Coloana[j]));
            if(!a || viz[a]) return false; viz[a] = 1;
        }

    return true;
}

void Rezolvare(void)
{
    Preprocesare();

    sort(C+1,C+10);

    for(int i=1;i<N && !gata;i++)
        for(int j=1;j<N && !gata;j++)
            if(ExistaInVector(Best[i][j]))
                for(int k=i+1;k<N && !gata;k++)
                    for(int l=j+1;l<N && !gata;l++)
                    {
                        gata = EsteBunaDistribuirea(i,k,j,l);
                        Initializare();
                    }
}

int main()
{
    citire();
    Rezolvare();

    fprintf(g,"%d %d %d %d\n",Linie[1],Linie[2],Coloana[1],Coloana[2]);
}