CH 0503 奇数码问题(进阶指南， 逆序数，归并)

算法竞赛进阶指南，38 页， 排序， 归并求逆序数

本题要点：
1、 不考虑空格，把每一个数字写到一个数组 a 上
空格与一个数字左右交换，逆序数不变。空格与一个数字上下交换，相当于把某个数字 （比如 a）往前移动了 n - 1 个位置
或则往后移动了 n - 1 个位置。
2、 这里以 a 往前移动了 n - 1 个位置为例，这n - 1 个数，假设有x个大于a ， y个小于a（x + y == n - 1）,
移动前的逆序数： 不变的部分 + x
移动后的逆序数： 不变的部分 + y
两种转态的逆序数的差值 ：差值 = |x - y|。
显然，x + y 和 x - y 的奇偶性一样。
3、 计算每个序列 的逆序数，这里使用了归并排序来计算；

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
using namespace std;

const int MaxN = 510;
int g1[MaxN][MaxN];
int g2[MaxN][MaxN];
int n, cnt;
int a[MaxN * MaxN], b[MaxN * MaxN];

void merge(int L, int mid, int R)
{
	int i = L, j = mid + 1;
	for(int k = L; k <= R; ++k)
	{
		if(j > R || (i <= mid && a[i] <= a[j]))
		{
			b[k] = a[i++];
		}else{
			b[k] = a[j++], cnt += mid - i + 1;
		}
	}
	for(int k = L; k <= R; ++k)
	{
		a[k] = b[k];
	}
}

void mergeSort(int L, int R)
{
	if(L < R)
	{
		int mid = (L + R) / 2;
		mergeSort(L, mid);
		mergeSort(mid + 1, R);
		merge(L, mid, R);
	}
}

void solve()
{
	int num = 0, s1 = 0, s2 = 0;
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
		{
			if(g1[i][j] != 0)
			{
				a[++num] = g1[i][j];
			}
		}
	cnt = 0;
	mergeSort(1, num);
	s1 = cnt;
	cnt = 0, num = 0;
	for(int i = 1; i <= n; ++i)
		for(int j = 1; j <= n; ++j)
		{
			if(g2[i][j] != 0)
			{
				a[++num] = g2[i][j];
			}
		}
	cnt = 0;
	mergeSort(1, num);
	s2 = cnt;
	if(abs(s1 - s2) & 1)
	{
		printf("NIE\n");
	}else{
		printf("TAK\n");
	}
}

int main()
{
	while(scanf("%d", &n) != EOF)
	{
		for(int i = 1; i <= n; ++i)
			for(int j = 1; j <= n; ++j)
			{
				scanf("%d", &g1[i][j]);
			}
		for(int i = 1; i <= n; ++i)
			for(int j = 1; j <= n; ++j)
			{
				scanf("%d", &g2[i][j]);
			}
		solve();
	}
	return 0;
}

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

/*
TAK
TAK
*/