CodeForces 518E Arthur and Questions（贪心 + 思维）题解

#include<cmath>
#include<set>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include <iostream>
#include<algorithm>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e5 + 10;
const ull seed = 131;
const int INF = 0x3f3f3f3f;
const int MOD = 1000000009;
int n, k, none[maxn];
ll a[maxn];
int id(int j, int k, int i){return j * k + i;}
void solve(int ii, int l, int r){
    int m = (l + r) / 2;
    int num = 0;
    for(int i = m; i <= r; i++)
        a[id(i, k, ii)] = num++;
    num = 0;
    for(int i = m; i >= l; i--)
        a[id(i, k, ii)] = num--;
    int dis;
    if(l != 0){
        dis = a[id(l - 1, k, ii)] - a[id(l, k, ii)];
        if(dis >= 0){
            dis++;
            for(int i = l; i <= r; i++)
                a[id(i, k, ii)] += dis;
        }
    }
    if(id(r + 1, k, ii) <= n){
        dis = a[id(r, k, ii)] - a[id(r + 1, k, ii)];
        if(dis >= 0){
            dis++;
            for(int i = l; i <= r; i++)
                a[id(i, k, ii)] -= dis;
        }
    }
}
int main(){
    char o[20];
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i++){
        scanf("%s", o);
        if(o[0] == ‘?‘){
            none[i] = 1;
        }
        else{
            sscanf(o, "%lld", &a[i]);
        }
    }
    for(int i = 1; i <= k; i++){
        int l = 0, r = 0, ok = 0;
        for(int j = 0; j * k + i <= n; j++){
            if(none[id(j, k, i)] && (j == 0 || !none[id(j - 1, k, i)])){
                l = j;
                ok = 1;
            }
            if(none[id(j, k, i)]){
                r = j;
            }
            else{
                if(ok){
                    solve(i, l, r);
                    ok = 0;
                }
            }
        }
        if(ok) solve(i, l, r);
    }

    ll tot = 0, pre;
    for(int i = 1; i <= k; i++){
        tot += a[i];
    }
    pre = tot;
    for(int i = k + 1; i <= n; i++){
        tot = tot - a[i - k] + a[i];
        if(tot <= pre){
            printf("Incorrect sequence\n");
            return 0;
        }
        pre = tot;
    }

    for(int i = 1; i <= n; i++){
        if(i != 1) printf(" ");
        printf("%d", a[i]);
    }
    printf("\n");
    return 0;
}