牛妹爱数列

牛妹爱数列

https://ac.nowcoder.com/acm/problem/205087

题目

有一个长度为 n 的序列 a,保证它是一个 01 序列,并执行以下两种操作:

  1. 单点修改:将位置 x 上的数翻转(0 变 1,1 变 0);
  2. 前缀修改:将位置 1~x 上的数翻转(每个数都 0 变 1,1 变 0)。

求:使得数列上的所有数都变为 0 的最小翻转次数。

解题思路

动态规划

表示将序列中前 个数都变为 0 的最小翻转次数。

表示将序列中前 个数都变为 1 的最小翻转次数。

遍历序列,如果

  • 要想将序列前 个数变为 0,可以使用单点修改直接将 变为 0,翻转次数为
    或者使用前缀修改,翻转次数为
    所以,
  • 要想将序列前 个数变为 1,可以不需更改,翻转次数为
    或者使用单点修改,翻转次数为
    所以,

同理,如果

根据上述的状态转移方程可以看出,当前的状态只与前一个状态有关。
所以,可以对上面的 DP 进行空间上的优化,使空间复杂度从 降为 ,只需使用 2 个变量实时更新前一个状态。

C++代码

#include<iostream>
using namespace std;

int main(){
    int n;
    cin >> n;
    int a;
    int pre[2] = {0};
    int cur[2] = {0};
    for(int i=1; i<=n; ++i){
        cin >> a;
        if(a==1){
            cur[0] = min(pre[0]+1, pre[1]+1);
            cur[1] = min(pre[0]+1, pre[1]);
        }
        else{
            cur[0] = min(pre[0], pre[1]+1);
            cur[1] = min(pre[0]+1, pre[1]+1);
        }
        pre[0] = cur[0];
        pre[1] = cur[1];
    }
    cout << cur[0] << endl;
    return 0;
}
全部评论

相关推荐

喜欢走神的孤勇者练习时长两年半:爱华,信华,等华,黑华
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-20 19:57
已编辑
某大厂 golang工程师 23.0k*16.0, 2k房补,年终大概率能拿到
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务