AcWing108 奇数码问题

这是数码问题的一个特殊情况，n为奇数的正方形，然后有一个元素是0，其余的都是正数，给定两个数码游戏的局面，问是否存在一种移动空格的放实，使其中一个局面可以变化成另外一个局面，

这个问题是问你有没有解，不是问解是什么，当且仅当，两个局面转化成的序列（这里的序列是指去掉零的，然后排列的）逆序对的个数奇偶性相同，这个问题有解，所以就把这个问题转化为了求逆序对的个数，

然后这个只是奇数码问题，还是正方形的n×n，还有更一般的情况，n×m，然后n和m的奇偶性任意，因此还有更多的结论

#include <bits/stdc++.h>

#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(), (x).end()

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pa;

const int N = 250005;

int cnt1, cnt2;

void merge(int *a, int l, int mid, int r, int& cnt) {
    int i = l, j = mid + 1;
    int temp[N], index = 0;
    while (i <= mid && j <= r) {
        if (a[i] <= a[j]) temp[index++] = a[i++];
        else temp[index++] = a[j++], cnt += mid - i + 1;
    }
    while (i <= mid) {
        temp[index++] = a[i++];
    }
    while (j <= r) {
        temp[index++] = a[j++];
        cnt += mid - i + 1;
    }
    for (int i = 0; i < index; i++) {
        a[l + i] = temp[i];
    }
}

void merge_sort(int *a, int l, int r, int& cnt) {
    if (l < r) {
        int mid = (l + r) / 2;
        merge_sort(a, l, mid, cnt);
        merge_sort(a, mid + 1, r, cnt);
        merge(a, l, mid, r, cnt);
    }
}

int main() {
    int n;
    while (cin >> n) {
        cnt1 = 0; cnt2 = 0;
        int a[N], b[N];
        int index1 = 0, index2 = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                int x;
                cin >> x;
                if (x) a[index1++] = x;
            }
        }
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                int x;
                cin >> x;
                if (x) b[index2++] = x;
            }
        }
        merge_sort(a, 0, n * n - 2, cnt1);
        merge_sort(b, 0, n * n - 2, cnt2);
        if ((cnt1 + cnt2) % 2) cout << "NIE" << endl;
        else cout << "TAK" << endl;
    }
    return 0;
}