7-1896C - Matching Arrays

题意： 两个数组$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;
}