题意: 两个数组\(a和b\),对\(b\)任意排序, 使得\(a[i]>b[i]的个数为x\),要求输出能满足的数列。
思路: 一个任意排序,相当于两个任意排序,都升序,发现规律,\(让排序后的b数组,循环右移x位置\) ,满足条件则输出,否则一定不满足。
代码:
点击查看代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e6+7;
// x-c 一定是b的倍数,并且是一个a次方数
const int inf=1e12;
int a[N],ad[N];
int b[N];
void solve()
{
int n,x;
cin>>n>>x;
for(int i=0;i<n;i++)
{
cin>>a[i];
}
for(int i=0;i<n;i++) cin>>b[i];
iota(ad,ad+n,0);
sort(ad,ad+n,[&](int l,int r){
return a[l]<a[r];
});
sort(b,b+n);
int w=x;
for(int i=0;i<n;i++)
{
int j=(i+w)%n;
if(a[ad[i]]>b[j]) x--;
}
for(int i=0;i<n;i++)
{
int j=(i+w)%n;
a[ad[i]]=b[j];
}
if(x==0)
{
cout<<"YES\n";
for(int i=0;i<n;i++) cout<<a[i]<<" \n"[i==n-1];
}
else cout<<"NO\n";
}
signed main()
{
int t=1;
cin>>t;
while(t--)
{
solve();
}
return 0;
}