[逆序对] poj 2893 M × N Puzzle

/**
[逆序对] poj 2893 M × N Puzzle
某状态的奇偶性定义为逆序对(不包括0的)总数的奇偶性。
此题目终态为偶数
首先，0的左右移动不改变奇偶性。
N为奇数，上下移动不改变奇偶性，故逆序数为偶的YES
N为偶数，上下移动逆序数变化为±1，此时还要考虑0的竖直距离，逆序数%2 == 距离%2 时YES
*/
#include 
#include 

#define lowbit(i) (i) & (-i)
#define N 1000007
int a[N],b[N],k;
void update(int i)
{
    for(; i <=k; i += lowbit(i))
        b[i] ++;
}
int sum(int i)
{
    int ans = 0;
    for(; i > 0; i -= lowbit(i))
        ans += b[i];
    return ans;
}
int cal()
{
    int i,res = 0;
    for(i = 0; i < k; ++i)
    {
        update(a[i]);
        res += i + 1 - sum(a[i]);
    }
    return res;
}
int main()
{
    int n,m,cnt,step,i,j,s;
    while(scanf("%d%d",&m,&n) && m)
    {
        memset(b,0,sizeof(b));
        k = 0;
        for(i = 0; i < m; ++i)
            for(j = 0; j < n; ++j)
            {
                scanf("%d",&s);
                if(s == 0)
                    step = m - i - 1;
                else
                    a[k++] = s;
            }
        cnt = cal();
        if(n & 1)
            step = 0;
        printf(step % 2 == cnt % 2? "YES\n" : "NO\n");
    }
    return  0;
}