AcWing 3809. 修改数组

原题链接

Description

给定一个长度为 $n$ 的正整数数组 $a_1,a_2,\cdots,a_n$。

你可以任意改变其中任意元素的值。

但是，改变后的元素的值仍需是正整数。

将一个元素的值从 $a$ 变为 $b$ 所需要付出的代价为 $|a−b|$。

对于一个正整数 $t$，如果 $|a_i−t|≤1$，则称第 $i$ 个元素能够与 $t$ 匹配。

现在，请你指定一个正整数 $t$，并且用最小的代价修改整个数组，使得数组中所有元素都能够与 $t$ 匹配。

指定的 $t$ 不同，所需付出的最小代价也可能不同。

请你合理选择正整数 $t$，目标是让所需付出的最小代价尽可能小。

Input

第一行包含整数 $T$，表示共有 $T$ 组测试数据。

每组数据第一行包含整数 $n$。

第二行包含 $n$ 个整数 $a_1,a_2,\cdots,a_n$。

Output

每组数据输出一行结果，首先输出你选择的正整数 $t$，然后输出使得数组中所有元素都能够与 $t$ 匹配，所需付出的最小代价。

如果 $t$ 有多种合理选择，则任选其一输出即可。

Data Range

$1≤T≤20$,
$1≤n≤1000$,
$1≤a_i≤100$,
同一测试点内所有 $n$ 的和不超过 1000。

Solution

简单题， 根据数据范围可以发现， $1 \le a_i \le 100$, 如果$t$直接暴力枚举，时间复杂度约等于$t*n*T = 2 * 10^6$，是可以ac的，所以直接对$t$从1~100枚举

Code

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdio>

using namespace std;
const int N = 10010;
int q[N];

int main()
{
    int T;
    cin >> T;
    while(T --)
    {
        int n;
        cin >> n;
        
        for(int i = 0; i < n; i ++)
        {
            cin >> q[i];
        }
        int res = 0x3f3f3f3f, j = 0;
        for(int t = 1; t <= 100; t ++)
        {
            int num = 0;
            for(int i = 0; i < n; i ++)
            {
                if(abs(t - q[i]) > 1) num += abs(t - q[i]) - 1; 
            }
            
            if(num < res)
            {
                res = num;
                j = t;
            }
        }
        cout << j << " " << res << endl;
    }
    return 0;
}