Hemose Shopping（cf-1592b）（递推）

cf

洛谷

#include <bits/stdc++.h>
#include <iostream>
#include<unordered_map>
#define x first
#define y second

#define  ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);

using namespace std;

typedef  long  long LL ;
typedef  unsigned long  long ULL ;
typedef pair<int,int> PII ;
typedef pair<LL,int> PLI ;
const int N = 2e5+10,M = 1000 * 100000 + 10 ,mod = 1000000007,INF1 = 0x3f3f3f3f;
const LL INF2 = 1e17;


//能否通过交换两个距离大于等于 x 的数，使得数组可以按照非递减的顺序来排序。
// 题目规定 a,b 的距离为 |a-b|
int  n,x,a[N],b[N];

bool check()
{
    // 判断中间的 两段 n-x 外的部分 
    // 1 + n-x
    for(int i=n-x+1;i<=x;i ++ )
        if(a[i] != b[i]) return 0;
    return 1;
}
inline void solve() {
    cin>>n>>x;
    for(int i=1;i<= n;i ++ )cin>> a[i],b[i] =a[i];

    if(x *2  <=n)
    {
        cout <<"YES" << endl;return ;
    }

    sort(b+1,b+n+1);
    // 这里  x* 2 >= n  ==== x >= n/2
    if(!check()) cout << "NO" << endl;
    else cout << "YES" <<endl;

}
signed main()
{
    ios
    int T;cin>>T;
    while(T -- )
    {
        solve();

    }


    return 0;
}