20：查找兄弟单词

输入描述:
先输入字典中单词的个数，再输入n个单词作为字典单词。
输入一个单词，查找其在字典中兄弟单词的个数
再输入数字n

输出描述:
根据输入，输出查找到的兄弟单词的个数

#include <cstdio> 
#include <string> 
#include <vector> 
#include <iostream> 
#include <algorithm> 
using namespace std; 
vector<string> dict; 
bool is_friend(string a, string b) { 
if (a == b) return false; 
sort(a.begin(), a.end()); 
sort(b.begin(), b.end()); 
return a == b; 
} 

int main() { 
int n; 
while (scanf("%d", &n) == 1) { 
dict.clear(); 
for (int i = 0; i < n; i++) { 
string s; 
cin >> s; 
dict.push_back(s); 
} 
sort(dict.begin(), dict.end()); 
string ask; 
cin >> ask; 
int id; 
scanf("%d", &id); 
int cnt = 0; 
string res = ""; 
for (int i = 0; i < (int)dict.size(); i++) { 
if (is_friend(dict[i], ask)) { 
cnt++; 
if (cnt == id) { 
res = dict[i]; 
} 
} 
} 
printf("%d\n", cnt); 
if (cnt >= id) 
printf("%s\n", res.c_str()); 
} 
return 0;
}