【ACWing】108. 奇数码问题

题目地址：

https://www.acwing.com/problem/content/description/110/

你一定玩过八数码游戏，它实际上是在一个$3\times 3$的网格中进行的，$1$个空格和$1\sim 8$。这$8$个数字恰好不重不漏地分布在这$3\times 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\times n$的网格中进行，其中$n$为奇数，$1$个空格和$1\sim n^2-1$这$n^2-1$个数恰好不重不漏地分布在$n\times n$的网格中。空格移动的规则与八数码游戏相同，实际上，八数码就是一个$n=3$的奇数码游戏。现在给定两个奇数码游戏的局面，请判断是否存在一种移动空格的方式，使得其中一个局面可以变化到另一个局面。

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

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

数据范围：
$1\le n<500$

两个局面可达等价于排成$n^2-1$的一行的时候逆序对的个数的奇偶性相同。可以用归并排序求逆序对个数。代码如下：

#include <iostream>
using namespace std;
using ll = long long;

const int N = 3e5 + 10;
int n;
int x[N], y[N], b[N];

ll merge_sort(int *a, int l, int r) {
  ll res = 0;
  if (l >= r) return res;
  int mid = l + r >> 1;
  res += merge_sort(a, l, mid) + merge_sort(a, mid + 1, r);
  int i = l, j = mid + 1, idx = l;
  while (i <= mid && j <= r) {
    if (a[i] <= a[j])
      b[idx++] = a[i++];
    else
      res += mid - i + 1, b[idx++] = a[j++];
  }
  while (i <= mid) b[idx++] = a[i++];
  while (j <= r) b[idx++] = a[j++];
  for (int i = l; i <= r; i++) a[i] = b[i];
  return res;
}

int main() {
  while (~scanf("%d", &n)) {
    for (int i = 1, j = 1; i <= n * n; i++) {
      int c;
      scanf("%d", &c);
      if (c) x[j++] = c;
    }
    for (int i = 1, j = 1; i <= n * n; i++) {
      int c;
      scanf("%d", &c);
      if (c) y[j++] = c;
    }
    n = n * n - 1;
    
    puts((merge_sort(x, 1, n) % 2 == merge_sort(y, 1, n) % 2) ? "TAK" : "NIE");
  }
}

每组数据时间复杂度$O(n^2)$，空间$O(n^2)$。