[LeetCode 884] 两句话中的不常见单词

题目链接

https://leetcode-cn.com/problems/uncommon-words-from-two-sentences/

题目描述

句子 是一串由空格分隔的单词。每个 单词 仅由小写字母组成。

如果某个单词在其中一个句子中恰好出现一次，在另一个句子中却 没有出现 ，那么这个单词就是 不常见的 。

给你两个 句子 s1 和 s2 ，返回所有 不常用单词 的列表。返回列表中单词可以按 任意顺序 组织。

示例

示例1

输入：s1 = "this apple is sweet", s2 = "this apple is sour"
输出：["sweet","sour"]

示例2

输入：s1 = "apple apple", s2 = "banana"
输出：["banana"]

提示

• 1 <= s1.length, s2.length <= 200；
• s1 和 s2 由小写英文字母和空格组成；
• s1 和 s2 都不含前导或尾随空格；
• s1 和 s2 中的所有单词间均由单个空格分隔。

思路

在其中一个句子恰好出现一次，另一个句子没出现，其实就等价于在两个句子中只出现一次，所以可以把两个句子连起来，找只出现一次的单词。可以使用map来统计哪些单词只出现了一次。

代码

class Solution {
public:
    vector<string> uncommonFromSentences(string s1, string s2) {
        map<string, bool> singleWord; // 统计句子中每个单词是否只出现一次
        string s = s1 + " " + s2 + " ";
        string word = "";
        for (int i=0; i<s.length(); i++)
        {
            if(s[i] == ' ')
            {
                if(singleWord.find(word) == singleWord.end())   singleWord[word] = true;
                else singleWord[word] = false;
                word = "";
            }
            else    word.push_back(s[i]);
        }

        vector<string> res;
        for(map<string, bool>::iterator it=singleWord.begin(); it != singleWord.end(); it++)
        {
            if(it->second)  res.push_back(it->first);
        }
        return res;
    }
};