3809 修改数组（枚举、贪心）

1. 问题描述：

给定一个长度为 n 的正整数数组 a1，a2，…，an。你可以任意改变其中任意元素的值。但是，改变后的元素的值仍需是正整数。将一个元素的值从 a 变为 b 所需要付出的代价为 |a − b|。对于一个正整数 t，如果 |ai − t| ≤ 1，则称第 i 个元素能够与 t 匹配。现在，请你指定一个正整数 t，并且用最小的代价修改整个数组，使得数组中所有元素都能够与 t 匹配。指定的 t 不同，所需付出的最小代价也可能不同。请你合理选择正整数 t，目标是让所需付出的最小代价尽可能小。

输入格式

第一行包含整数 T，表示共有 T 组测试数据。每组数据第一行包含整数 n。第二行包含 n 个整数 a1，a2，…，an。

输出格式

每组数据输出一行结果，首先输出你选择的正整数 t，然后输出使得数组中所有元素都能够与 t 匹配，所需付出的最小代价。如果 t 有多种合理选择，则任选其一输出即可。

数据范围

1 ≤ T ≤ 20，
1 ≤ n ≤ 1000，
1 ≤ ai ≤ 100，

同一测试点内所有 n 的和不超过 1000。

输入样例：

2
3
10 1 4
5
1 1 2 2 3

输出样例：

3 7
2 0
来源：https://www.acwing.com/problem/content/description/3812/

2. 思路分析：

分析题目可以知道实际上是已知一个数轴，数轴上有若干个点，我们需要选择一个数字x，使其余数字变成当前x，x - 1，x + 1之后使得最终总共花费的代价最小，可以发现这道题目类似于104题，104题使用了绝对值不等式求解最小代价，这道题目实际上是104题的扩展，我们也可以使用类似的思路，a变成x，x + 1或者x - 1花费的最小代价可以表示为f(a，x) = max(0，abs(a - x) - 1)，所以实际上我们需要求解出这样一个x使得所有点变成这个点的花费最小，也即求解s = f1(a1，x) + f2(a2，x) + ... fn(an，x)，类似于104题我们可以对所有点按照坐标从小到大进行排序，然后首尾分组，这样中间位置的点一定是距离所有点的距离是最近的，但是这道题目要求选择的点必须是正整数的点，所以根据中间两个点的长度分情况讨论可以选择哪些点：

由上图可以发现x可以为中间a - 1，a，a + 1这些点，枚举这些点求解一个最小值即可，因为x取这三个点一定可以使得总花费是最小的。由题目可知数据规模比较小，其实我们还可以暴力枚举t的所有可能取值，由于ai在[1，100]，所以我们在这个范围内枚举t，在所有可能的的取值中计算出最小花费即可。

3. 代码如下：

暴力枚举：

class Solution:
    def process(self):
        T = int(input())
        for c in range(T):
            n = int(input())
            a = list(map(int, input().split()))
            res_c, rec_t = 10 ** 8, 0
            for t in range(1, 101):
                _sum = 0
                for j in range(n):
                    # 注意是与0取一个max, 因为最小不能是负数
                    _sum += max(0, abs(a[j] - t) - 1)
                if res_c > _sum:
                    res_c = _sum
                    rec_t = t
            print(rec_t, res_c)


if __name__ == '__main__':
    Solution().process()

贪心：

class Solution:
    def process(self):
        T = int(input())
        for c in range(T):
            n = int(input())
            a = list(map(int, input().split()))
            # a从小到大排序
            a.sort()
            mid = a[n // 2]
            res_c, res_t = 10 ** 8, 0
            # 枚举x = a - 1, a, a + 1
            for i in range(max(1, mid - 1), mid + 2):
                cost = 0
                for j in range(n):
                    cost += max(0, abs(a[j] - i) - 1)
                if cost < res_c:
                    res_c = cost
                    res_t = i
            print(res_t, res_c, end=" ")


if __name__ == '__main__':
    Solution().process()