优化二进制序列
Description
你有一个包含 0 和 1 的二进制序列,现在想要在这个序列中进行一次优化。你可以翻转一个 0 成为 1,但只能进行一次操作。你的目标是找到连续 1 的最长子序列,使得经过一次翻转后,连续 1 的个数最多。你能找到最长的连续 1 子序列吗?
Input
第一行输入一个整数n,表示数组的长度。
第二行输入n个整数(不是0就是1),中间用空格隔开。
Output
输出一个整数,表示数组中连续1的最大个数。
Sample
代码
import java.util.Scanner;
public class Main {
public static void main(String[] args) throws Exception {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] nums = new int[n];
for(int i = 0; i < n; i++) {
nums[i] = scanner.nextInt();
}
System.out.println(process(nums));
}
public static int process(int[] num){
int dp0 = 0;
int dp1 = 0;
int result = -1;
for(int j = 0;j<num.length;j++){
if(num[j]==1){
dp1++;
dp0++;
}else {
dp1 = dp0+1;
dp0 = 0;
}
result = Math.max(dp1,Math.max(dp0,result));
}
return result;
}
}
思路
首先考虑动态规划;
使用两个变量dp0,dp1分别记录当前位置i不含有0的最大子数组长度、含一个0的最大子数组长度。
当i= 1 dp0 = dp0 +1,dp1 = dp1+1
当i =0,,dp0 = 0,dp1 = dp0 + 1.
由于dp1之前已经含有一个0了,所有再遇到0需要更新成dp0(不含0的)并且加1(相当于当前dp1又含了个0),然后dp0 初始化成0(因为dp0不能含有0)
另外定义一个变量result保存dp1的最大值。
总体是dp0和dp1的不断替换,需要注意当为0时,dp1要先赋值,不然dp1 = dp0+1也就是0+1=1
注意
这个规模输入是1<=n<=100000
在输入的时候,发现字符串分割读取时有runtime error,猜想分割不适合此类大规模问题。
切记使用scanner.nextInt()进行数字读取,不要使用nextLine然后分割