C++实现单词按自己的字典序排序

#include <bits/stdc++.h>
using namespace std;
int n,id;
string s[1010];
int dict[300];//此处定义字典序数组
bool cmp(string s1,string s2)
{
    int i=0;
    while(1)
    {
    if(dict[s1[i]]<dict[s2[i]]) return true;
    if(dict[s1[i]]>dict[s2[i]]) return false;
    i++;
    }
}
int main()
{
    char x;
    for(int i=1;i<=26;i++)
        {
        cin>>x;
        dict[x]=i;
        }
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>s[i];
    sort(s+1,s+n+1,cmp);
    cin>>id;//此处输入排序方式，1是正序，0是倒序
    if(id==1)
        for(int i=1;i<=n;i++)
            cout<<s[i]<<endl;
    else
        for(int i=n;i>=1;i--)
            cout<<s[i]<<endl;
    return 0;
}