力扣500

一、题目描述

给定字符串数组，判断每个字符串是否由同一行构成

二、解题思路

将三行分为三组，如果一个字符在第一行找到就把数字1插入到set容器，第二行就插入2，第三行就插入3。由于set容器不会有重复的元素，最后判断set容器的长度是否为1，长度为1就说明是由同一行的字母组成。

三、代码

class Solution
{
public:
    vector<string> findWords(vector<string>& words)
    {
        //给键盘中的三行定义字符数组
        set<char> row1 = {'q','w','e','r','t','y','u','i','o','p','Q','W','E','R','T','Y','U','I','O','P'};
        set<char> row2 = {'a','s','d','f','g','h','j','k','l','A','S','D','F','G','H','J','K','L'};
        set<char> row3 = {'z','x','c','v','b','n','m','Z','X','C','V','B','N','M'};

        //定义set容器判断单词使用了哪几行
        set<int> flag;
        //返回的答案
        vector<string> ans;

        //字符串数组相当于二维数组，使用两层循环就可以访问到
        //每个字符串元素的每一个字符
        for (int i = 0; i < (signed)words.size(); i++)
        {
            for (auto ch : words[i])
            {
                if (row1.count(ch))
                {
                    flag.insert(1);
                }
                else if (row2.count(ch))
                {
                    flag.insert(2);
                }
                else
                {
                    flag.insert(3);
                }
            }
            //外层循环执行一次后，判断标志位数组长度是否为1，
            //并将该数组清空
            if ((signed)flag.size() == 1)
            {
                ans.push_back(words[i]);
                flag.clear();
            }
            flag.clear();
        }
        return ans;
    }
};

提示：这里对文章进行总结：

四、总结

字符串数组可以看作一个二维数组，使用两层循环就可以访问到每个字符串元素的每一个字符

        for (auto word : words)
        {
            for (auto ch : word)
            {
                cout << num;
            }
            cout << endl;
        }

因为输入的字符串数组每个元素由字符char组成，所以键盘行的时候需要 定义类型为char，这样才能比较。在定义set时，使用了{“a”}和{a}两种赋值方法，是错误的，这里踩了个坑，应该使用单引号。
如有问题，请大家指出