汕头市队赛 SRM10 T1模拟只会猜题意

模拟只会猜题意 SRM 10

描述

有一本n个单词的词典，求按下列方法能造出的不相同的词汇数目。 
1.词典中的单词是一个词。
2.能分为两部分的，其中前一部分是一个词典词或者其非空前缀，后一部分是一个词典词或者其非空后缀的词。

输入格式

第一行一个整数n，接下来n行每行一个字符串，表示单词

输出格式

一个整数，答案

样例输入

3
a
ab
bc

样例输出

20

数据范围与约定

对10%的数据，n=1
对另外40%的数据，1<=n<=10,单词只含字符'a'
对所有的数据，1<=n<=10,单词长度<=10，单词只含小写字母

样例解释

可以造出以下单词:
a,ab,aab,ac,abc,aa
abb,abab,abbc,aba
bb,bab,bc,bbc,ba
bcb,bcab,bcc,bcbc,bca

————————————————————————————

只是道水题却花了我将近两个钟QAQ

用string写很方便QAQ

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int M=1e7;
int read(){
    int ans=0,f=1,c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+(c-'0'); c=getchar();}
    return ans*f;
}
int n,ans,cnt[55],sum;
string s[55],c[M];
bool cmp(string a,string b){return a<b;}
int main()
{
    n=read();
    for(int i=1;i<=n;i++) cin>>s[i],cnt[i]=s[i].length();
    for(int i=1;i<=n;i++) c[++sum]=s[i];
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      for(int l=1;l<=cnt[i];l++)
       for(int r=cnt[j]-1;r>=0;r--)
            c[++sum]=s[i].substr(0,l)+s[j].substr(r,cnt[j]-r);
    sort(c+1,c+1+sum,cmp);
    ans=sum;
    for(int i=2;i<=sum;i++) if(c[i]==c[i-1]) ans--;
    printf("%d\n",ans);
    return 0;
}