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");
    }
}
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页