问题 N: 奇数码问题

题目描述

你一定玩过八数码游戏，它实际上是在一个3*3的网格中进行的，1个空格和1~8这8个数字恰好不重不漏地分布在这3*3的网格中。
例如：
5 2 8
1 3 _
4 6 7
在游戏过程中，可以把空格与其上、下、左、右四个方向之一的数字交换（如果存在）。
例如在上例中，空格可与左、上、下面的数字交换，分别变成：
5 2 8       5 2 _      5 2 8
1 _ 3       1 3 8      1 3 7
4 6 7       4 6 7      4 6 _

奇数码游戏是它的一个扩展，在一个n*n的网格中进行，其中n为奇数，1个空格和1~n*n-1这n*n-1个数恰好不重不漏地分布在n*n的网格中。
空格移动的规则与八数码游戏相同，实际上，八数码就是一个n=3的奇数码游戏。

现在给定两个奇数码游戏的局面，请判断是否存在一种移动空格的方式，使得其中一个局面可以变化到另一个局面。

 

输入

多组数据，对于每组数据：
第1行一个整数n，n<500，n为奇数。
接下来n行每行n个整数，表示第一个局面。
接下来n行每行n个整数，表示第二个局面。
局面中每个整数都是0~n*n-1之一，其中用0代表空格，其余数值与奇数码游戏中的意义相同，保证这些整数的分布不重不漏。

 

输出

对于每组数据，若两个局面可达，输出TAK，否则输出NIE。

 

样例输入

3
1 2 3
0 4 6
7 5 8
1 2 3
4 5 6
7 8 0
1
0
0

 

样例输出

TAK
TAK

我们分析移动的结果，会影响逆序对数，当n为奇数时只需要判断两种状态逆序对的奇偶是否相同即可，n为偶数时逆序对数+空格键的行数差的奇偶是否相同。

#include<bits/stdc++.h>
#include<iostream>
using namespace std;
typedef long long ll;
ll a[250050],r[250050],ans1,ans2,ans=0,n;
void msort(int s,int t)
{
    if(s==t) return;
    int mid=(s+t)/2;
    msort(s,mid);
    msort(mid+1,t);
    int i=s,j=mid+1,k=s;
    while(i<=mid&&j<=t)
    {
        if(a[i]<a[j])
        {
            r[k]=a[i];
            k++;
            i++;
        }
        else
        {
            r[k]=a[j];
            k++;
            j++;
            ans+=mid-i+1;
        }
    }
    while(i<=mid)
    {
        r[k]=a[i];
        k++;
        i++;
    }
    while(j<=t)
    {
        r[k]=a[j];
        k++;
        j++;
    }
    for(int i=s; i<=t; i++)
        a[i]=r[i];
}
int main()
{
    ll te;
    while(cin>>n)
    {
        n*=n;
        if(n==1)
        {
            int p,q;
            scanf("%d %d",&p,&q);
            if(p==q)
            {
                printf("TAK\n");
            }
            else
            {
                printf("NIE\n");
            }
            continue;
        }
        int j=1;
        for(int i=1; i<=n; i++)
        {
            scanf("%lld",&te);
            if(te!=0)
            {
                a[j++]=te;
            }
        }
        ans=0;
        msort(1,n-1);
        ans1=ans;
        j=1;
        for(int i=1; i<=n; i++)
        {
            scanf("%lld",&te);
            if(te!=0)
            {
                a[j++]=te;
            }
        }
        ans=0;
        msort(1,n-1);
        ans2=ans;
        if(ans1%2==ans2%2)
        {
            printf("TAK\n");
        }
        else
        {
            printf("NIE\n");
        }
    }
    return 0;
}