NowCoder 练习 —— 牛妹爱数列（DP）

题目链接: 牛妹爱数列

一、题目描述

他手里有一个长度为n的序列a，保证它是一个01序列，并执行以下两种操作：

• 单点修改：将位置x上的数翻转（0变1,1变0）；
• 前缀修改：将位置1~x上的数翻转（每个数都0变1,1变0）。

他现在想要最小化翻转次数，使得数列上的所有数都变为0。

二、输入描述:

• 第一行，输入一个数n
• 第二行，输入n个数，第i个数表示a_i
• 数据保证1≤n≤10⁵, 0≤a_i≤1

三、输出描述:

输出最小翻转次数

示例1：

• 输入
  10
  1 0 1 1 0 0 0 1 0 0
• 输出
  3

四、思路

• 因为修改支持单点修改，区间反转只能前缀和反转，所以考虑DP；
• dp[i][j]表示前i个数为j时需要的最少操作数，0 ≤ j ≤ 1；
• 状态转移方程为：
  • a[i] == 0时，dp[i][0] = min(dp[i-1][0], dp[i-1][1]+1) dp[i][1] = min(dp[i-1][0]+1, dp[i-1][1]+1);
  • a[i] == 1时，dp[i][0] = min(dp[i-1][0]+1, dp[i-1][1]+1) dp[i][1] = min(dp[i-1][0]+1, dp[i-1][1]);
• 输出 min(dp[n][0], dp[n][1]+1);

五、Code

#include <bits/stdc++.h>

using namespace std;
#pragma GCC optimize(2)
#define ios std::ios::sync_with_stdio(false);cin.tie(0), cout.tie(0);
#define ll long long
const int Max = 1e5+3;
const int mod = 1e9+7;

int a[Max], dp[Max][2], n;

int main()
{
	ios
	
	cin >> n;
	for(int i=1; i<=n; i++)
		cin >> a[i];
	dp[0][0] = dp[0][1] = 0;
	for(int i=1; i<=n; i++)
	{
		if(a[i]) {
			dp[i][0] = min(dp[i-1][0]+1, dp[i-1][1]+1);
			dp[i][1] = min(dp[i-1][0]+1, dp[i-1][1]);
		}
		else {
			dp[i][0] = min(dp[i-1][0], dp[i-1][1]+1);
			dp[i][1] = min(dp[i-1][0]+1, dp[i-1][1]+1);
		}
	}
	int tmp = min(dp[n][0], dp[n][1]+1);
	cout << tmp << endl;
	return 0;
}

---

蒟蒻一只，欢迎指正