Codeforces Round #700 (Div. 2) D2 Painting the Array II(贪心)

题意:
给出一个数组,要求将这个数组分为两个子序列。定义子序列的价值为:将连续相同的数合并为一个后,剩下数的个数。求两个子序列价值相加最小。

题解:
其实和d1的解法差不多,基本一样。

设x为第一个序列的最后一个元素,y为第二个序列的最后一个元素。那么可以分类讨论:
1.x==y
此时a[i]不管放哪个都没影响。
2.a[i]==x&&a[i]!=y
此时应该放x后面,才是最优的,因为能消一个是一个。
3.a[i]!=x&&a[i]==y
同上。
4.a[i]!=x&&a[i]!=y
那么这个时候就不能随便放了。设nex[i]表示 i 下次出现的位置,如果nex[x]<nex[y],说明要把a[i]放在y后面,这样做就能使得后面的x可以消掉,反之放在x后面。

代码:

#include<bits/stdc++.h>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
#define DEV_RND ((int)rand()*RAND_MAX+rand())
#define RND(L,R) (DEV_RND%((R)-(L)+1)+(L))
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>pii;
const int MAXN=1e5+5;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
std::vector<int> v[MAXN];
int a[MAXN];
int num[MAXN];
int main()
{
    
    
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
    
    
        cin>>a[i];
        v[a[i]].push_back(i);
    }
    int ans=0;
    int x=0,y=0;
    for(int i=1;i<=n;i++)
    {
    
    
        num[a[i]]++;
        if(x==y)
        {
    
    
            if(x!=a[i])
            {
    
    
                ans++;
                x=a[i];
            }
        }
        else
        {
    
    
            if(a[i]==x)
            {
    
    
                //ans++;
                x=a[i];
            }
            else if(a[i]==y)
            {
    
    
                //ans++;
                y=a[i];
            }
            else
            {
    
    
                if(num[x]>=v[x].size())
                {
    
    
                    ans++;
                    x=a[i];
                }
                else if(num[y]>=v[y].size())
                {
    
    
                    ans++;
                    y=a[i];
                }
                else if(v[x][num[x]]<v[y][num[y]]) 
                {
    
    
                    ans++;
                    y=a[i];
                }
                else
                {
    
    
                    ans++;
                    x=a[i];
                }
            }
        }
    }
    printf("%d\n",ans);
}

猜你喜欢

转载自blog.csdn.net/weixin_45755679/article/details/113761851