CF-Array Painting

(.-_-.)

已于 2024-02-28 19:15:31 修改

阅读量617

点赞数 8

分类专栏：日常训练文章标签： c++ 算法

于 2024-02-13 16:00:10 首次发布

本文链接：https://blog.csdn.net/lr1799322352/article/details/136107253

版权

日常训练专栏收录该内容

8 篇文章 0 订阅

订阅专栏

本文讲述了在Codeforces问题D中解决染色问题的策略，涉及连续大于0元素的子串处理、硬币消耗计算和代码实现，关键在于正确分析染色规则并优化算法。

摘要由CSDN通过智能技术生成

链接：Problem - D - Codeforces

思路历程：

这道题我在第一遍做的时候觉得好简单啊（把思路想简单了），于是打了个for循环，让值大于0的位置优先渲染其前面的格子，然后再渲染后面的格子，结果wa在第5个点，然后想了一会，茅塞顿开，不应该忽略连续的大于0的的串之间内部的作用。比如2,2的这种情况，如果按这种思路想的话就会两个互相染色，就不会产生硬币消耗，这显然不对。因此我就想把连续的大于0的串作为基本单位来进行处理。

可以发现，每个子串都需要消耗一个硬币来激活，内部可以相互染色。内部的染色都是相同的不过外部的染色就分为两种情况了：

1.如果子串内部没有元素大于1，两边中只有一边可以被染色（激活最前面的，然后每个都染色自己的后面，或者激活最后面的每个都激活自己的前面）

2.如果子串内部存在元素大于1，可以对两边进行同时染色（激活那个大于1的元素，前后同时激活，前面的激活前面，后面的激活后面）

按照这两种情况处理后再加上没有染色的方块数量即可.

代码实现：

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define all(a) a.begin(), a.end()
#define pb push_back
#define pii pair<int, int>
#define endl '\n'
#define N 200005
int a[N];//记录数值
bool vis[N];// 记录染色情况
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int i, j;
    int n;
    cin >> n;
    for (i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    if (n == 1)//如果只有一个元素就直接输出1
    {
        cout << "1" << endl;
        return 0;
    }
    int ans = 0;
    int len = 0;//记录大于0的子串长度
    bool ok = 0;//标记是否有子串内元素大于1
    for (i = 1; i <= n; i++)
    {
        if (a[i] > 0)
        {
            len++;//长度+1
            if (a[i] > 1 && i != n)//有大于1的元素,标记ok为1
                ok = 1;
        }
        else
        {
            if (len > 0)
            {
                ans++;//激活子串所需
                if (ok == 0)//内部没有大于1的情况
                {
                    int j = i - len - 1;//优先对前面的进行染色
                    if (vis[j] == 1)
                        j++;
                    for (int k = 0; k <= len; k++)
                        vis[j + k] = 1;
                }
                else//内部有大于1的情况
                {
                    int j = i - len - 1;
                    if (vis[j] == 1)
                        j++;
                    for (; j <= i; j++)//两边染色
                        vis[j] = 1;
                }
            }
            len = 0;
            ok = 0;
            continue;
        }
    }
    if (len > 0)//补充未被处理的尾部
    {
        ans++;
        if (ok == 0)
        {
            int j = i - len - 1;
            if (vis[j] == 1)
                j++;
            for (int k = 0; k <= len; k++)
                vis[j + k] = 1;
        }
        else
        {
            int j = i - len - 1;
            if (vis[j] == 1)
                j++;
            for (; j <= i; j++)
                vis[j] = 1;
        }
    }
    for (i = 1; i <= n; i++)
    {
        if (vis[i] == 0)//如果未被染色，答案+1
            ans++;
    }
    cout << ans << endl;
    return 0;
}