HJ27 查找兄弟单词

描述

定义一个单词的“兄弟单词”为：交换该单词字母顺序（注：可以交换任意次），而不添加、删除、修改原有的字母就能生成的单词。

兄弟单词要求和原来的单词不同。例如： ab 和 ba 是兄弟单词。 ab 和 ab 则不是兄弟单词。

现在给定你 n 个单词，另外再给你一个单词 x ，让你寻找 x 的兄弟单词里，按字典序排列后的第 k 个单词是什么？

注意：字典中可能有重复单词。

数据范围：1 \le n \le 1000 \1≤n≤1000 ，输入的字符串长度满足 1 \le len(str) \le 10 \1≤len(str)≤10 ， 1 \le k < n \1≤k<n

输入描述：

输入只有一行。 先输入字典中单词的个数n，再输入n个单词作为字典单词。 然后输入一个单词x 最后后输入一个整数k

输出描述：

第一行输出查找到x的兄弟单词的个数m 第二行输出查找到的按照字典顺序排序后的第k个兄弟单词，没有符合第k个的话则不用输出。

示例1

输入：

3 abc bca cab abc 1

复制输出：

2
bca

复制

示例2

输入：

6 cab ad abcd cba abc bca abc 1

复制输出：

3
bca

复制说明：

abc的兄弟单词有cab cba bca，所以输出3
经字典序排列后，变为bca cab cba，所以第1个字典序兄弟单词为bca 

#include <iostream>
#include <map>
#include <vector>
#include <algorithm>


void brotherWord(std::vector<std::string> strVect, std::string word, int k)
{
    std::multimap<std::string, int> rtnMap;               //存放是兄弟单词的容器
    std::vector<std::string>::iterator iter = strVect.begin();
    while(iter != strVect.end())
    {
        std::string curWord = *iter;
        if(curWord != word)                     //兄弟单词不能一模一样
        {
            std::string tempword = word;
            std::sort(tempword.begin(), tempword.end());

            std::string tempcurword = curWord;
            std::sort(tempcurword.begin(), tempcurword.end());

            if(tempcurword == tempword)         //是兄弟单词
            {
                rtnMap.insert(std::pair<std::string, int>(curWord, 1));
            }
        }
        iter++;
    }
    // print(rtnMap);
    std::vector<std::string> rtnVect;
    std::multimap<std::string, int>::iterator iterMap = rtnMap.begin();
    while(iterMap != rtnMap.end())
    {
        rtnVect.push_back(iterMap->first);
        iterMap++;
    }
    std::cout << rtnVect.size() << std::endl;
    if(rtnVect.size() > 1)
        std::cout << rtnVect[k-1] << std::endl;
}

int main()
{
    int n;
    std::cin >> n;
    std::vector<std::string> strVect;
    while(n--)
    {
        std::string temp;
        std::cin >> temp;
        strVect.push_back(temp);
    }

    std::string word;
    std::cin >> word;

    int k;
    std::cin >> k;


    brotherWord(strVect, word, k);

    return 0;
}

题解：先将所有的兄弟单词放在multimap中，会进行自动排序。然后将map的key也就是所有的兄弟单词放在vector中，目的是为了直接输出第K个兄弟单词，注意这里输出是k-1,防止数组下标越界所以需要判断vector中至少有一个兄弟单词。