LeetCode—— 383 赎金信

问题描述

给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串，判断第一个字符串ransom能不能由第二个字符串magazines里面的字符构成。如果可以构成，返回 true ；否则返回 false。

(题目说明：为了不暴露赎金信字迹，要从杂志上搜索各个需要的字母，组成单词来表达意思。)

注意：

你可以假设两个字符串均只含有小写字母。

canConstruct("a", "b") -> false
canConstruct("aa", "ab") -> false
canConstruct("aa", "aab") -> true

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ransom-note
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：用键值对，比较费时。可以优化为数组，索引为字符，值为个数。

代码一：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        if(magazine.size() == 0 && ransomNote.size() == 0)  return true;
        map<char, int> key_count;
        for(int i = 0; i < magazine.size(); ++i)
        {
            key_count[magazine[i]] += 1;    // 如果不存在，则值为1，如果存在，则对值+1，
        }
        for(int i = 0; i < ransomNote.size(); ++i)
        {
            if(key_count.count(ransomNote[i]) == 1) // magazine中有此值，且此字符的数量要大于等于ransomNote中字符的数量
            {
                key_count[ransomNote[i]] -= 1;
                if(key_count[ransomNote[i]] < 0)
                    return false;
            }
            else
                return false;
        }
        return true;
    }
};

优化代码二：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        if(magazine.size() == 0 && ransomNote.size() == 0)  return true;
        int arr[26] = {0};
        for(int i = 0; i < magazine.size(); ++i)
        {
            arr[magazine[i] - 'a'] += 1;    // 如果不存在，则值为1，如果存在，则对值+1，
        }
        for(int i = 0; i < ransomNote.size(); ++i)
        {
            if(arr[ransomNote[i] - 'a'] > 0) // magazine中有此值，且此字符的数量要大于等于ransomNote中字符的数量
            {
                arr[ransomNote[i]-'a'] -= 1;
            }
            else
                return false;
        }
        return true;
    }
};