算法Day 13 优化二进制序列

优化二进制序列

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然后分割