双指针+贪心，CF 1849D - Array Painting

一、题目

1、题目描述

2、输入输出

2.1输入

2.2输出

3、原题链接

1849D - Array Painting

---

二、解题报告

1、思路分析

首先注意到一点，对于非0连续段，我们只需要1的花费就能把整段染红

如果段内无2，我们可以额外染红段左或段右的某个0值

如果段内有2，我们可以染红段左右的两个0值

这样就很好处理了

我们遍历数组，记录0的个数

遇到非0连续段，我们看段内是否有2，有2把两边的0都染红

否则，我们优先染红左边，次优先染红右边

2、复杂度

时间复杂度： O(N)空间复杂度：O(N)

3、代码详解

 ​

#include <bits/stdc++.h>
using i64 = long long;
using i128 = __int128;
using PII = std::pair<int, int>;
const i64 inf = 1e18 + 7, P = 1e9 + 7;


void solve() {
    int n;
    std::cin >> n;
    std::vector<int> a(n);
    for (int i = 0; i < n; i ++ ) std::cin >> a[i];
    int res = 0;
    for (int i = 0; i < n; ) {
        if (!a[i]) {
            ++ res;
            ++ i;
            continue;
        }
        bool has2 = false;
        int st = i;
        for (; i < n && a[i]; i ++ )
            has2 = has2 || (a[i] == 2);
        ++ res;
        if (st && !a[st - 1]) {
            -- res;
            if (!has2) continue;
        }
        if (i < n) {
            a[i] = 1;
            ++ i;
        }
    }
    std::cout << res;
}

int main(int argc, char** argv) {
    std::ios::sync_with_stdio(false), std::cin.tie(0), std::cout.tie(0);
    int _ = 1;
    // std::cin >> _;
    while (_ --)
        solve();
    return 0;
}