M × N Puzzle POJ - 2893

原题链接

考察： 归并排序

八数码的延伸版,这里允许了N为偶数,所以要分情况讨论

当N为偶数,向下挪一行奇偶改变,再向下挪奇偶又改变,因此答案和0的位置有关

易错：

        判断条件的时候,&1两边都要打括号不然报错

#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
const int N = 1e6+10;
int m,n;
ll a[N],tmp[N];
ll merge(int l,int r)
{
    if(l>=r) return 0;
    ll mid = (ll)l+r>>1ll;
    ll cnt = merge(l,mid) + merge(mid+1,r);
    ll i = l; ll j = mid+1; int k = 1;//在递归过程中,a数组是从l开始排序的
    while(i<=mid&&j<=r){
        if(a[i]<=a[j]) tmp[k++] = a[i++];
        else{ tmp[k++] = a[j++];cnt+=mid-i+1; }
    }
    while(i<=mid) tmp[k++] = a[i++];
    while(j<=r) tmp[k++] = a[j++];
    for(int i=l,k=1;i<=r;i++,k++) a[i] = tmp[k];//k不能从l开始,因为是从1开始赋值
    return cnt;
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d",&m,&n)!=EOF&&(n!=0||m!=0))
    {
        ll step;
        for(int i=1,k=1;i<=m;i++){
            for(int j=1;j<=n;j++){
                int x; scanf("%d",&x);
                if(x) a[k++] = x;
                else step = m-i;//当前行 
            }
        }
        ll ans = merge(1,m*n-1);
        if(n & 1) step = 0;
        printf(((step&1) == (ans&1))? "YES\n" : "NO\n");
    }
    return 0;
}