华为机试题-查找兄弟单词

这篇题解看了牛课网的讨论答案

输入描述：
先输入字典中单词的个数n，再输入n个单词作为字典单词。
再输入一个单词，查找其在字典中兄弟单词的个数m
再输入数字k
输出描述:
根据输入，输出查找到的兄弟单词的个数m 

然后输出查找到的兄弟单词的第k个单词。

输入

3 abc bca cab abc 1

输出

2
bca

 

#include<iostream>
#include<string>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
bool isbrother(string s1,string s2){
    if(s1.size()==s2.size()){
        if(s1==s2)return false;
        sort(s1.begin(),s1.end());//对字符串s1内的元素按字典序排序
        sort(s2.begin(),s2.end());
        if(s1==s2) return true;
    }
    return false;
}
int main()
{
    int n;
    string str,word,s;
    int index;
    vector<string> vs;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        cin>>str;
        vs.push_back(str);
    }
    sort(vs.begin(),vs.end());//对若干字符串按字典序排序
    cin>>word;
    cin>>index;
    int counts=0;
    for(int i=0;i<n;i++)
    {

        if(isbrother(word,vs[i]))
        {
            counts++;
            if(counts==index)
                s=vs[i];
        }
    }
    if(!vs.empty())
        cout << counts << endl;
    if(counts >= index)
        cout << s << endl;
}