程序员面试金典 16.15

Master Mind:根据答案solution以及猜测guess,判断猜中和伪猜中的数量。

先计算猜中,同时统计未猜中的频次,最后根据频次统计伪猜中。

class Solution {
public:
    vector<int> masterMind(string solution, string guess) {
        vector<int> ans(2, 0);
        map<char, int> m;
        for(size_t s = 0; s < solution.size(); s++)
        {
            if(solution[s] == guess[s]) ans[0]++;
            else m[solution[s]]++;
        }
        for(size_t g = 0; g < guess.size(); g++)
        {
            if(guess[g] != solution[g]){
                auto iter = m.find(guess[g]);
                if(iter != m.end()){
                    ans[1]++;
                    iter->second--;
                    if(iter->second == 0) m.erase(iter);
                }
            }
        }
        return ans;
    }
};

你可能感兴趣的:(《程序员面试金典》)