这道题和经典的背包问题很类似,不同的是在背包问题中,我们只有一种容量,而在这道题中,我们有 0 和 1 两种容量。每个物品(字符串)需要分别占用 0 和 1 的若干容量,并且所有物品的价值均为 1。因此我们可以使用二维的动态规划。
动态转移方程为
dp(i, j) = max(dp(i,j),1 + dp(i - cost_zero[k], j - cost_one[k]))
if i >= cost_zero[k] and j >= cost_one[k]
int findMaxForm(char ** strs, int strsSize, int m, int n){
int dp[101][101]={0};
int i=0,j=0,k=0,c=0;
int count0=0;
int count1=0;
for(i=0;i<strsSize;i++){//遍历字符串数组
c=0;//遍历每个字符
count0=0;
count1=0;
while(strs[i][c]!='\0'){
if(strs[i][c]-'0'==0)
count0+=1;
else
count1+=1;
c+=1;
}
for(j=m;j>=count0;j--)
for(k=n;k>=count1;k--)
dp[j][k]=dp[j][k]>(1+dp[j-count0][k-count1])?dp[j][k]:(1+dp[j-count0][k-count1]);
}
return dp[m][n];
}