ZWEI (zwe)题解

题目描述

有N 个由小写拉丁字母构成的单词A1, A2, …, AN，其中任意一个单词都不是另一个单词的前缀。

注意，每个单词都是自己的前缀，所以这其中当然没有完全相同的单词。

现在希望把每个单词Ai都改成它的一个非空前缀A’i，使得在A’1,A’2, …, A’N 中仍没有一个单词

是另外一个的前缀。在此前提下，使得每个A’i 的长度最小。现在你的任务就是，对于给定的N

个单词，求出满足条件的方案。

输入

输入文件的第一行包含一个正整数N，表示单词的数目。

之后N 行，每行一个字符串，表示单词。

输出

按照输入文件的顺序输出改动之后的单词，每个单词一行。

样例输入

5
applepi
apollonius
october
octopus
zwei

样例输出

app
apo
octob
octop
z

提示

N≤10^2，单词最长不超过10^2 个字符

N≤10^3，单词最长不超过10^3 个字符

想法

• n<=10^3 于是这道题就成了乱搞题
• 先进行一遍快排然后上下比较

算法

• 将读入的字符串排序
• 对于一个字符串将它与其他字符串比较
• 复杂度O(n^2logn)

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
using namespace std;
const int Maxn=1002,Maxl=1000002,inf=999999999+208;
int n,Pn=1,En=0;
char a[Maxn][Maxn];
int fst[Maxl],next[Maxl],to[Maxl],w[Maxl],cnt[Maxl];
void addedge(int u,int v,int ch)
{
    next[++En]=fst[u];fst[u]=En;to[En]=v;w[En]=ch;
}
void build(int x)
{
    int idx=1,tail=1,len=strlen(a[x]+1);
    while (tail<=len)
    {
        bool flag=false;
        for (int k=fst[idx];k;k=next[k])
        {
            if (w[k]==a[x][tail]) {idx=to[k];flag=true;break;}
        }
        if (!flag) {addedge(idx,++Pn,a[x][tail]);idx=Pn;}
        cnt[idx]++;tail++;
    }
}
void work(int x)
{
    int idx=1,tail=1,len=strlen(a[x]+1);
    while (tail<=len)
    {
        for (int k=fst[idx];k;k=next[k])
        {
            if (w[k]==a[x][tail]) {idx=to[k];break;}
        }
        printf("%c",a[x][tail]);
        if (cnt[idx]==1) break;
        tail++;
    }
    printf("\n");
}
int main()
{
    scanf(" %d",&n);cnt[1]=inf;
    for (int i=1;i<=n;i++)
    {
        scanf(" %s",a[i]+1);
        build(i);
    }
    for (int i=1;i<=n;i++) work(i);
    return 0;
}

• 再附上一个O(n^3)的代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
using namespace std;
int n,ed[1005],str[1005],cnt;
char ch[1005][1005];
bool visit[1005];
int main()
{
    //freopen("zwe.in","r",stdin);
    //freopen("zwe.out","w",stdout);
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf(" %s",ch[i]);
        str[i]=strlen(ch[i]);
    }
    for (int i=1;i<=n;i++)
        ed[i]=str[i];
    if(n!=1)
    {
        for (int i=1;i<=n;i++)
        {
            for (int j=1;j<=n;j++)visit[j]=1;
            cnt=n-1;
            for (int j=0;j<=str[i];j++)
            {
                if(cnt==0){ed[i]=j;break;}
                for (int k=1;k<=n;k++)
                {
                    if(visit[k]==0)continue;
                    if(ch[k][j]!=ch[i][j])
                    {
                        if(visit[k]==true)cnt--;
                        visit[k]=false;
                    }
                }
            }
        }
        for (int i=1;i<=n;i++)
        {
            for (int j=0;j<ed[i];j++)
                printf("%c",ch[i][j]);
            printf("\n");
        }
    }
    else
    {
        printf("%c",ch[1][0]);
    }
    return 0;
}