题目传送门
思路
根据这题的题意,先拿一个结构体将题中的 a a a 数组存起来,里面存两个变量 v a l , i d val,id val,id 分别表示值和下标,然后创建一个 b b b 数组和 a n s ans ans 数组。
然后把这两个数组排序, a a a 数组按照值从小到大排序, b b b 数组也按照值从小到大排序,再用一个 f l a g flag flag 存储是否合法,初始化为 f a l s e false false。
接着我们取出 b b b 数组第 1 ∼ x 1 \thicksim x 1∼x 个数判断一下是否大于等于 a n − x + i a_{n-x+i} an−x+i,如果大于,把 f l a g flag flag 赋值为 1 1 1,然后直接 b r e a k break break。
接着我们取出 b b b 数组第 x + 1 ∼ n x+1 \thicksim n x+1∼n 个数判断一下是否小于 a i − x a_{i-x} ai−x,如果小于,把 f l a g flag flag 赋值为 1 1 1,然后直接 b r e a k break break。
如果 f l a g flag flag 是 t r u e true true,直接输出 N o No No。
如果 f l a g flag flag 是 f a l s e false false,先输出 Y e s Yes Yes,再将 b i b_i bi 赋值给 a n s a n − m + i . i d ans_{a_n-m+i.id} ansan−m+i.id,然后下半部分也是同理的。
接下来讲一下证明。
因为我们要使 a i > b i a_i \gt b_i ai>bi 的对数刚好有 x x x 个,所以我们要将 a a a 数组前 x x x 大与 b b b 数组前 x x x 小放一起才可以符合题意。
代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
struct node{
int val,id;
}a[200005];
int t,n,m;
int b[200005],ans[200005];
bool cmp(node x,node y){
return x.val<y.val;
}
signed main(){
cin>>t;
while(t--){
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i].val;
a[i].id=i;
}
for(int i=1;i<=n;i++){
cin>>b[i];
}
sort(a+1,a+n+1,cmp);
sort(b+1,b+n+1);
int flag=0;
for(int i=1;i<=m;i++){
if(b[i]>=a[n-m+i].val){
flag=1;
break;
}
}
for(int i=m+1;i<=n;i++){
if(b[i]<a[i-m].val){
flag=1;
break;
}
}
if(flag){
cout<<"No"<<endl;
}else{
cout<<"Yes"<<endl;
for(int i=1;i<=m;i++){
ans[a[n-m+i].id]=b[i];
}
for(int i=m+1;i<=n;i++){
ans[a[i-m].id]=b[i];
}
for(int i=1;i<=n;i++){
cout<<ans[i]<<" ";
}
cout<<endl;
}
}
return 0;
}