2021牛客暑期多校训练营1 K题

K题: Knowledge Test about Match

原题链接:https://ac.nowcoder.com/acm/contest/11166/K

题目大意

给定两个长度为$n$的数组$a$,$b$:
a = { 0 , 1 , 2 , 3 , . . . , n − 2 , n − 1 } a=\{0,1,2,3,...,n-2,n-1\} a={0,1,2,3,...,n−2,n−1}
b = { b 0 , b 1 , b 2 , b 3 , . . . , b n − 2 , b n − 1 } ( 对 于 0 ≤ i ＜ n ， 保 证 0 ≤ b i ＜ n ) b=\{b_0,b_1,b_2,b_3,...,b_{n-2},b_{n-1}\}(对于0\le i＜n，保证0\le b_i＜n ) b={b0​,b1​,b2​,b3​,...,bn−2​,bn−1​}(对于0≤i＜n，保证0≤bi​＜n)
现在你可以任意调整数组$b$内的元素顺序，使得他们的loss函数最小。
求排序后的b数组。
定义数组$a$和$b$的loss函数为:
$$f(a,b)=\sum _{i=0}^{n-1}\sqrt{\left|a_i-b_i\right|}$$
由于解决这个问题的时间有限，不一定要求出最优解。
共$T$组测试数据，设第$k$组测试的标准答案的loss函数为$f_k^{\ast }$，你的输出结果的loss函数为 $f_k\hat{}$，则只需满足以下不等式即可:
$$\frac{1}{T}\sum _{k=1}^T\frac{f_k\hat{}-f_k^{\ast }}{f_k^{\ast }}\le 4\%$$

题解

不难发现，对于loss函数(参考根式的函数图像)，在 A方案:多个中等大小的绝对值差 与 B方案:部分小绝对值差+部分大绝对值差 中，B方案更优。我们采取贪心策略， 若同一个数在$b$数组中出现多次，则优先将其填入到最近的空位中。

参考代码

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e3+5;
int T,n,b[MAXN],c[MAXN];
int main()
{
	cin>>T;
	while(T--){
		scanf("%d",&n);
		memset(b,0,sizeof(b)),memset(c,0,sizeof(c));
		int p,u=0;
		for(int i=0;i<n;i++)scanf("%d",&p),c[p]++,b[p]=p;//记录每个数出现的次数，并将已有数直接填入
		for(int s=1;s<n;s++){//从小到大枚举间距
			for(int i=0;i<n;i++){//先判断该数是否仍有多余，再判断距离为s的点是否在数组范围内(合法)，再判断是否为空位，然后进行填入，更新c
				if(c[i]>1)if(i-s>=0)if(!c[i-s])c[i-s]++,c[i]--,b[i-s]=i;
				if(c[i]>1)if(i+s<n)if(!c[i+s])c[i+s]++,c[i]--,b[i+s]=i;
			}
		}
		for(int i=0;i<n;i++){
			printf("%d ",b[i]);
		}
		puts("");
	}
	return 0;
}