[CF1479B1/CF1480D1] Painting the Array I

[CF1479B1/CF1480D1] Painting the Array I

Description

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

Solution

贪心，决定每个元素 (a[i]) 放在哪里，取决于 (a[i],a[i+1]) 和当前两个已有子序列的末尾

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

int n;
int a[100005];

signed main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    a[n+1]=-1;
    int top1=-1,top2=-1,ans=0;
    for(int i=1;i<=n;i++)
    {
        if(a[i]==top1 && a[i]==top2) continue;
        else if(a[i]==top1) 
        {
            top2=a[i];
            ans++;
        }
        else if(a[i]==top2)
        {
            top1=a[i];
            ans++;
        }
        else 
        {
            ans++;
            if(a[i+1]==top1 && a[i+1]!=top2) 
            {
                top1=a[i];
            }
            else if(a[i+1]!=top1 && a[i+1]==top2)
            {
                top2=a[i];
            }
            else 
            {
                top1=a[i];
            }
        }
    }
    cout<<ans<<endl;
}