题目:给定一个只包含0或1的数组,例如,input={1,0,0,0,1,0,0,1,0},以及一个将0转变为1的次数changeCount,计算最长连续的1的个数。
思路:
采用动态规划方法,dp[i][j]表示将第i次(0转1权力)赋予第j个元素,所得到的以j结尾的最长连续1的个数。
如果input[j]==0,dp[i][j]=dp[i-1][j]+1;
如果input[j]==1,dp[i][j] = dp[i][j - 1] + 1。
import java.util.Scanner;
public class Solution {
public static void main(String[] args) {
System.out.println(getMaxLength(new int[]{1,1,0,0,0,1,1,0,1,0,0,0,0,1 }, 4));
}
public static int getMaxLength(int[] input,int changeCount){
int m=input.length;
int[][] dp=new int[changeCount+1][m];//dp[i][j]表示将第i次(0转1权力)赋予第j个元素,所得到的以j结尾的最长连续1的格式
dp[0][0]=input[0]==1?1:0;
for(int i=1;i<input.length;i++){
if(input[i]==1){
dp[0][i]=dp[0][i-1]+1;
}
}
for(int i=1;i<=changeCount;i++){
for(int j=0;j<m;j++){
if(input[j]==0) {
if (j >= 1) {
dp[i][j] = dp[i - 1][j - 1] + 1;
} else {
dp[i][j] = 1;
}
}else{
if(j==0){
dp[i][0]=1;
}else {
dp[i][j] = dp[i][j - 1] + 1;
}
}
}
}
int ans=0;
for(int j=0;j<m;j++){
ans=dp[changeCount][j]>ans?dp[changeCount][j]:ans;
}
return ans;
}
}