题目大意:给了两个序列a、b,a序列是1 - n,b序列为随机序列(bi范围:1 - n)
可以任意交换b序列,使得以下loss函数值最小。
但题目不要求精准答案,只要满足:
所以我们可以直接贪心,先double一个s数组,让s[ i ] = sqrt( i );
然后去遍历几遍,当s[ abs(i - b[ i ]) ] + s[ abs ( j - b[ j ]) ] > s[ abs(i - b[ j ]) ] + s[ abs( j - b[ i ]) ]时交换b[ i ] 、b[ j ],尽量让loss变小。
- :???忽然悟了
代码:
#include <bits/stdc++.h>
using namespace std;
double s[1010];
int b[1010], n;
void init()
{
for(int i = 0; i < 1010; i++)
{
s[i] = sqrt(i);
}
}
void check()
{
for(int i = 0; i < n; i++)
{
for(int j = i+1; j < n; j++)
{
if(s[abs(i - b[i])] + s[abs(j - b[j])] > s[abs(i - b[j])] + s[abs(j - b[i])])
swap(b[i], b[j]);
}
}
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
init();
int T;
scanf("%d", &T);
while(T--)
{
scanf("%d", &n);
for(int i = 0; i < n; i++)
{
scanf("%d", &b[i]);
}
sort(b, b+n);
for(int i = 1; i <= 5; i++)
check();
for(int i = 0; i < n; i++)
cout<<b[i]<<" ";
cout<<endl;
}
return 0;
}