Олимпиада 1999г.

#include <stdio.h>
#include <math.h>
int main()
{
      freopen("input.txt","r",stdin);
      freopen("output.txt","w",stdout);
        int N, K, i, j, **mas1, **mas2, *mas_res, i_res=1, temp;
        scanf("%d %d", &N, &K);
        mas1=new int*[N];
        mas2=new int*[N];
        mas_res=new int[N];
        for(i=0; i<N; i++)
        {
            mas1[i]=new int[K];
            mas2[i]=new int[K];
            for(j=0; j<K; j++)
                scanf("%d", &mas1[i][j]);
        }
        for(i=0; i<K; i++)
        {
            temp=0;
            for(j=0; j<K; j++)
                if(mas1[N-1][j])
                    temp+=abs(i-j);
            mas2[N-1][i]=temp;
        }
        for(i=N-2; i>=0; i--)
        {
            for(j=0; j<K; j++)
            {
                temp=mas2[i+1][j];
                if(j>0 && temp>mas2[i+1][j-1])
                    temp=mas2[i+1][j-1];
                if(j<K-1 && temp>mas2[i+1][j+1])
                    temp=mas2[i+1][j+1];
                for(int y=0; y<K; y++)
                    if(mas1[i][y])
                        temp+=abs(y-j);
                mas2[i][j]=temp;
            }
        }
        temp=0;
        for(i=0; i<K; i++)
            if(mas2[0][i]<mas2[0][temp])
                temp=i;
        mas_res[0]=(N-1)*K+temp+1;
        int temp_N=1;
        while(temp_N<=N-1)
        {
            int temp2=temp;
            if(temp>0 && mas2[temp_N][temp2]>mas2[temp_N][temp-1])
                temp2=temp-1;
            if(temp<K-1 && mas2[temp_N][temp2]>mas2[temp_N][temp+1])
                temp2=temp+1;
            mas_res[i_res++]=(N-temp_N-1)*K+temp2+1;
            temp_N++;
            temp=temp2;
        }
        i_res--;
        while(i_res>=0)
            printf("%d\n", mas_res[i_res--]); 
        return 0;
}