Knowledge Test about Match 贪心乱搞 匹配

Posted 2021-08-18 爷灬傲奈我何123

链接
题意：
a数组是0～n-1，让b数组和a来匹配，任意两个数的代价是sqrt(|a[j]-b[j]|)，求加起来的最小值
思路：
由于sqrt函数的性质，不能无脑sort，由于n加起来40000，每组最多1000，算下来最多4e7左右吧，然后b的值都在0～n-1里面，那就启发我们，每组里面n^2,可以先枚举差值d然后暴力匹配。
代码：

#include<iostream>
using namespace std;
const int N = 10010;
int a[N];
bool st[N];
int ans[N];
bool anss[N];
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		double res=0;
		int n;
		cin>>n;
		for(int i=0;i<=n-1;i++)
			st[i]=0,anss[i]=0;
		for(int i=1;i<=n;i++)
			cin>>a[i];
		for(int i=0;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				if(anss[j])	continue;
				if(a[j]-i>=0 && !st[a[j]-i])
				{
					st[a[j]-i]=1;
					ans[a[j]-i]=a[j];
					//ans[j]=a[j]-i;
					anss[j]=1;
					continue;
				}
				if(a[j]+i<=n-1 && !st[a[j]+i])
				{
					st[a[j]+i]=true;
					anss[j]=1;
					ans[a[j]+i]=a[j];
				}
			}
		}
		for(int i=0;i<=n-1;i++)
			cout<<ans[i]<<" ";
		cout<<endl;
	}

	return 0;
}