給你一個二進制字符串數(shù)組 strs 和兩個整數(shù) m 和 n 。

請你找出并返回 strs 的最大子集的大小，該子集中 最多 有 m 個 0 和 n 個 1 。

如果 x 的所有元素也是 y 的元素，集合 x 是集合 y 的 子集 。

• 示例 1：

輸入：strs = [“10”, “0001”, “111001”, “1”, “0”], m = 5, n = 3
輸出：4
解釋：最多有 5 個 0 和 3 個 1 的最大子集是 {“10”,“0001”,“1”,“0”} ，因此答案是 4 。
其他滿足題意但較小的子集包括 {“0001”,“1”} 和 {“10”,“1”,“0”} 。{“111001”} 不滿足題意，因為它含 4 個 1 ，大于 n 的值 3 。

• 示例 2：

輸入：strs = [“10”, “0”, “1”], m = 1, n = 1
輸出：2
解釋：最大的子集是 {“0”, “1”} ，所以答案是 2 。

解題思路

數(shù)組含義

dp[k][i][j]代表取前k個字符串數(shù)組的元素下，i個0，j個1的情況下，最大的子集的大小

狀態(tài)轉(zhuǎn)移

dp[k+1][i][j]=max(dp[k+1][i][j],dp[k][i-z][j-o]+1)
統(tǒng)計strs[k]中0，1的個數(shù)，z為0的個數(shù)，o為1的個數(shù)，dp[k+1][i][j]的情況是可能由dp[k][i-z][j-o]轉(zhuǎn)移而來（將strs[k]加入當前到dp[k][i-z][j-o]內(nèi)的子集中，所以子集數(shù)目加一）

代碼

func findMaxForm(strs []string, m int, n int) int {dp := make([][][]int, len(strs)+1)for i := 0; i <= len(strs); i++ {dp[i] = make([][]int, m+1)for j := 0; j <=m ; j++ {dp[i][j]=make([]int,n+1)}}cnt := func(s string) (res int) {runes := []rune(s)for _, c := range runes {if c == '1' {res++}}return}max := func(a int, b int) int {if a>b{return a} else {return b}}m2:= map[string]int{}for _,s := range strs {m2[s]= cnt(s)}for k := 0; k < len(strs); k++ {o := cnt(strs[k])z := len([]rune(strs[k]))-ofor i := 0; i <= m; i++ {for j := 0; j <= n; j++ {dp[k+1][i][j]=dp[k][i][j]if i-z>=0&&j-o>=0{dp[k+1][i][j]=max(dp[k+1][i][j],dp[k][i-z][j-o]+1)}}}}return dp[len(strs)][m][n] }

總結(jié)

以上是生活随笔為你收集整理的leetcode 474. 一和零（dp）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。