給你一個二進制字符串數組 strs 和兩個整數 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 。
解題思路
數組含義
dp[k][i][j]代表取前k個字符串數組的元素下,i個0,j個1的情況下,最大的子集的大小
狀態轉移
dp[k+1][i][j]=max(dp[k+1][i][j],dp[k][i-z][j-o]+1)
統計strs[k]中0,1的個數,z為0的個數,o為1的個數,dp[k+1][i][j]的情況是可能由dp[k][i-z][j-o]轉移而來(將strs[k]加入當前到dp[k][i-z][j-o]內的子集中,所以子集數目加一)
代碼
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]
}