7-1896C - Matching Arrays

发布时间 2023-11-26 21:31:17作者: xxj112

题意: 两个数组\(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;
}