[模拟] aw3565. 完美矩阵(绝对值不等式+边界处理)

1. 题目来源

链接：3565. 完美矩阵

前导题：[绝对值不等式] 货仓选址(绝对值不等式+贪心)

2. 题目解析

有点脑筋急转弯的意思，转过来了之后还得处理边界问题。

若行列是回文序列，则 {i, j}, {i, m - j + 1}, {n - i + 1, j}, {n - i + 1, m - j + 1}，在此 i、j 从 1 开始。

这四个位置的点均相等，即关于中线、横线对称，关于中心对称。

当长为偶数奇数时，左右将少一个数。当宽为奇数时，上下将少一个数，当长、宽均为奇数时，中间那个数就被晾下了。

总之，当我们遇到不足 4 个数的特殊情况时，显然需要特殊考虑边界。

处理边界问题，同样思想，两种方式：

• 遍历整个二维数组，将回文序列的数统一压缩到右上方。这样自然如果只有两个对应数的话，该对应位置就只存两个数，如果只有一个数的话，就只存一个数。这就是 mrk 大佬的处理方式。这样就将问题直接转化成货仓选址问题了。
• 遍历上 1/4 二维数组，对称求出对应的四个位置。可能会出现位置重复，但是可以使用 set 进行去重。这就是 y总 的处理方式。

---

若想将其变为 x，则最小代价为 min(|a-x|+|b-x|+|c-x|+|d-x|)，显然满足绝对值不等式，应该取中位数。

假设数组下标从 0 开始，那么中位数分为两种情况，数组长为奇数、偶数，假设长度为 n：

• 奇数：n/2。
• 偶数：左中位数：(n+1)/2-1，右中位数 n/2。
  • 针对左中位数这个式子，在求奇数长度时，也能正确求得中位数，因为是上取整，取到中位数前一位数，再减一，取到中位数。
  • 这两个式子可以记一下，适用于求数组中位数，及左中位数！

数据会爆 long long。

---

时间复杂度：$O(nm)$
空间复杂度：$O(nm)$

---

// 学习 mrk 大佬的代码
// 直接将问题转化为货仓选址问题...tql
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

typedef long long LL;

const int N = 105;

int n, m, a[N][N];
vector<int> g[N][N];

int main() {
    int T; scanf("%d", &T);
    while (T--) {
        scanf("%d%d", &n, &m);
        // mrk 大佬处理边界问题属实巧妙！
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                scanf("%d", &a[i][j]);
                g[min(i, n - i + 1)][min(j, m - j + 1)].pb(a[i][j]);    // 压缩到左上角 1/4 矩形中
            }
        }

        LL ans = 0;
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                if (!g[i][j].size()) continue;
                sort(g[i][j].begin(), g[i][j].end());
                // int t = g[i][j][(g[i][j].size() + 1) / 2 - 1];   // 偶数情况，取左边中位数
                int t = g[i][j][g[i][j].size() / 2];                // 偶数情况，取右边中位数
                for (int k = 0; k < g[i][j].size(); k++) 
                    ans += abs(t - g[i][j][k]);

                g[i][j].clear();                                    // 多组测试数据
            }
        }
        printf("%lld\n", ans);
    }
    return 0;
}

// y总的代码，采用下标的形式固定传入四个值
// 再用 set 进行判重，也是非常的香啊
#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>
#include <vector>

using namespace std;

typedef long long LL;
typedef pair<int, int> PII;

#define x first
#define y second

const int N = 105;

int n, m;
int a[N][N];

LL cal(set<PII> S) {
    vector<int> q;
    for (auto s : S) q.push_back(a[s.x][s.y]);
    sort(q.begin(), q.end());

    int t = q[q.size() / 2];
    LL res = 0;
    for (int i = 0; i < q.size(); i ++ ) 
        res += abs(t - q[i]);
    return res;
}

int main() {
    int T;
    scanf("%d", &T);

    while (T -- ) {
        scanf("%d%d", &n, &m);
        for (int i = 1; i <= n; i ++ ) 
            for (int j = 1; j <= m; j ++ ) {
                scanf("%d", &a[i][j]);
            }

        LL res = 0;

        for (int i = 1; i <= n - i + 1; i ++ ) 
            for (int j = 1; j <= m - j + 1; j ++ ) {
                res += cal({{i, j}, {i, m - j + 1}, {n - i + 1, j}, {n - i + 1, m - j + 1}});
            }

        printf("%lld\n", res);
    }
    return 0;
}