cf

H. K and Medians

题意

给出1到n的数，可以任意取K位只要递增的数列（可以不连续）保留中位数，问是否可以变成m位的数组

题解

实际上只要考虑最后一个操作，

反过来相当于m位数组填坑，最后一个操作，在某个数左边右边添加任意（k - 1）/ 2位，就可以保证由整个数组的空位也可以，还要保证坑的数量mod (k - 1) = 0

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX = 2e5 + 10;
int a[MAX];

const int INF = 1e9 + 7;

int main() {
    int T;
    scanf("%d", &T);
    while(T--) {
        int n, k, m;
        scanf("%d%d%d", &n, &k, &m);
        vector<int> ans;
        scanf("%d", &a[0]);
        if(a[0] > 1)
            ans.push_back(a[0] - 1);
        int end = m;
        for(int i = 1; i < m; i++) {
            scanf("%d", &a[i]);
            if(a[i] > a[i - 1] + 1) {
                ans.push_back(a[i] - a[i - 1] - 1);
            }
        }
        if(n > a[m - 1])
            ans.push_back(n - a[m - 1]);
        ll tot = 0;
        for(auto x : ans) {
            tot += x;
        }
        ll L = 0, R = 0;
        bool flag = 0;
        for(auto x : ans) {
            L += x;
            R = tot - L;
            if(L - k / 2 >= 0 && R - k / 2 >= 0 && tot % (k - 1) == 0) {
                flag = 1;
                break;
            }
        }
        if(flag) printf("YES\n");
        else printf("NO\n");
    }
}