Lưu ý: Các code mẫu dưới đây chỉ mang tính tham khảo và có thể không AC được bài tập này

Code mẫu của ladpro98

#include <bits/stdc++.h>
const int MN = 500005;
using namespace std;
int m, n;
int next[MN], bit[MN], point[MN];

void Upd(int i) {
    while (i <= m * n) {
        bit[i]++; i += i & -i;
    }
}

int Get(int i) {
    int s = 0;
    while (i) {
        s += bit[i];
        i -= i & -i;
    }
    return s;
}

int main()
{
    int i, j, kind, pos = 0; long long cnt = 0;
    scanf("%d %d", &m, &n); scanf("%d", &kind);
    for(j=1; j<=n; j++) for(i=1; i <= m; i++) {
        next[(i - 1) * n + j] = ++pos;
        point[pos] = (i - 1) * n + j;
    }
    if (kind == 1) {
        for(i=1; i<= m * n; i++)
        if (i != next[i]) {
            cnt++;
            point[next[i]] = point[i];
            swap(next[point[i]], next[i]);
        }
    }
    else {
        for(i =  m * n; i; i--) {
            cnt += Get(next[i]);
            Upd(next[i]);
        }
    }
    cout << cnt;
    return 0;
}