Ones and Zeroes

2017-06-04  本文已影响4人  我叫胆小我喜欢小心

题目来源
给你一堆0和1组成的字符串,以及m个1,n个0,问m个1,n个0最多能组成多少个字符串。
没想到怎么用DP来做--,实在太弱了。
实际上就是一个背包问题呀,每个字符串要或不要,要的话得消耗多少个0多少个1。

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>> dp(m+1, vector<int>(n+1, 0));
        
        for (auto str : strs) {
            int numZeroes = 0, numOnes = 0;
            for (int i=0; i<str.size(); i++) {
                if (str[i] == '0')
                    numZeroes++;
                else
                    numOnes++;
            }
            for (int i=m; i>=numZeroes; i--) {
                for (int j=n; j>=numOnes; j--) {
                    dp[i][j] = max(dp[i][j], dp[i - numZeroes][j - numOnes] + 1);
                }
            }
        }
        return dp[m][n];
    }
};
上一篇 下一篇

猜你喜欢

热点阅读