Crazy Rows

代码如下:

/*
    Crazy Rows 
*/ 
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAX_N=41;
int M[MAX_N][MAX_N]; 

int a[MAX_N];

int main(void)
{
    freopen("D:\1.TXT","r",stdin);
    int N;
    cin >> N;
    cout << "N=" << N << endl;
    for(int i=0;i<N;i++)
    {
        for(int j=0;j<N;j++)
        {
            scanf("%1d",&M[i][j]);
        }
    }
    
    int res = 0;
    for(int i=0;i<N;i++)
    {
        a[i]=-1;
        for(int j=0;j<N;j++)
        {
            if(M[i][j]==1) a[i]=j;
        }
    }
    
    for(int i=0;i<N;i++)
    {
        int pos=-1;
        for(int j=i;j<N;j++)
        {
            if(a[j]<=i) 
            {
                pos=j;
                break;
            }
        }
        
        for(int j=pos;j>i;j--)
        {
            swap(a[j],a[j-1]);
            res++;
        }
        //printf("H"); 
    }

    printf("%d
",res);
    return 0;
}