Dreamoon Likes Coloring 【CF 1329 A】

傳送門

思路：“Dreamoon will choose a number pipi from range [1,n−li+1](inclusive) and will paint all cells from pipi to pi+li−1(inclusive) in ii-th color.”可以知道從[1, n - li - 1]任意位置往後染pi個格子為第ith種顏色。

容易想到，如果∑li < n，說明"-1"。

如果∑li>=n，因為我們不知道怎麼染色才好，但我們知道SUM = ∑li，即我們目前還可以染色SUM塊。不如我們類貪心的思想染色，這樣我們可以分成兩種情況:

假設now為當前的pi，len為剩餘未染色的塊

①SUM - now >= len - 1

說明我們只用當前顏色染色1塊，之後SUM-now的個數也可以染色剩餘的部分，那麼 SUM -= now ，len -= 1

②SUM - now < len - 1

說明我們如果用當前顏色只染色1塊，則SUM - now 不能染色剩餘的 len - 1，那麼我們需要讓

SUM - now == len - 1 - ? (==>) ? = (len - 1) - (SUM - now),則當前顏色需要染色 ? + 1個才行。SUM -= now, len -= (1 + ?)

這樣我們可以用pre_s，pre_d記錄之前的開始位置和染色長度。

當然我們不能忘記一個條件“每種顏色只能在[1, n - li - 1]開始往後染色”，如果（n - li - 1） < pre_s + pre_d，

說明我們無法完成滿足題意的染色，因為我們前面是儘可能少的染色且滿足題目要求，如果仍然無法滿足，說明沒有可行解。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
#include <cstring>
#include <functional>
#define LL long long
using namespace std;

const int N = 1e5 + 10;
int L[N], inx[N];

void solve ()
{
    int n, m, len;
    scanf("%d%d", &n, &m);

    LL Sum_d = 0;
    for(int i = 1; i <= m; ++i) {
        scanf("%d", L + i);
        Sum_d += L[i];
    }

    int pre_s, pre_d;
    pre_s = 1; pre_d = 0;
    len = n;
    for(int i = 1; i <= m; ++i) {
        //printf("start = %d  L = %d\n", pre_s + pre_d, n - L[i] + 1);
        if(n - L[i] + 1 < pre_s + pre_d) { break; }

        if(Sum_d - L[i] >= len - 1) {
            Sum_d -= L[i];
            len -= 1;
            pre_s = pre_s + pre_d;
            pre_d = 1;
        } else {
            int tmp_d = (len - 1) - (Sum_d - L[i]);
            if(tmp_d + 1 > L[i]) { break; }
            Sum_d -= L[i];
            len -= (1 + tmp_d);
            pre_s = pre_s + pre_d;
            pre_d = (1 + tmp_d);
        }
        inx[i] = pre_s;
    }

   // cout << "len = " << len << endl;

    if(len > 0) {
        printf("-1\n");
    } else{
        for(int i = 1; i <= m; ++i) { printf("%d ", inx[i]); }
        printf("\n");        
    }


}

int main()
{
    solve();

    return 0;
}