程序員面試金典 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;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章