[CF1479B2/CF1480D2] Painting the Array II

[CF1479B2/CF1480D2] Painting the Array II

Description

将一个序列拆成两个子序列,然后每个子序列中相邻相同的元素只保留一个,最小化剩下元素的个数。

Solution

贪心,决定每个元素 \(a[i]\) 放在哪里,需要考虑 \(a[i]\),下个和 \(a[i]\) 相等的数的位置,以及当前已有子序列的末尾

每个元素进来的时候,如果不能直接接在一个相同的末尾后面的话,必然要毁掉一个当前的末尾延续到下一个相同元素的可能

设 A 队队尾的下个元素位置为 pA,B 队为 pB

如果 pA 小于 pB,意味着 B 队尾在今后遭遇危险的可能性更大,因此我们选择 B 插入,反之亦然

#include <bits/stdc++.h>
using namespace std;

#define int long long 
const int N = 100005;

int n,a[N];
vector <int> g[N];
int nxt[N];

signed main()
{
    ios::sync_with_stdio(false);

    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i], g[a[i]].push_back(i);

    for(int i=1;i<=n;i++) nxt[i]=n+1;
    for(int i=1;i<=n;i++) for(int j=1;j<g[i].size();j++) nxt[g[i][j-1]]=g[i][j];

    int val1=0,nxt1=n+1,val2=0,nxt2=n+1;
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        int val=a[i];
        if(val==val1) nxt1=nxt[i];
        else if(val==val2) nxt2=nxt[i];
        else 
        {
            if(nxt1>nxt2) 
            {
                val1=val;
                nxt1=nxt[i];
                ++ans;
            }
            else
            {
                val2=val;
                nxt2=nxt[i];
                ++ans;
            }
        }
    }
    cout<<ans<<endl;
}
上一篇:AcWing 3034. 望远镜 POJ3675 三角剖分求圆与多边形的面积交并


下一篇:VBS删除文件