复合词UVa10391(STL简单应用)

一、题目

输入一系列由小写字母组成的单词。输入已按照字典序排序（这句话就是个陷阱），且不超过120000个。找出所有的复合词，即恰好由两个单词连接而成的单词。

二、解题思路

要么枚举两两拼接的情况，O(n^2),n这么大肯定会超时。要么枚举每个单词的拆分情况，当单词比较短时，O(n*m),可能可行。

我们用string类型的数组存储这些单词，map记录单词的是否出现，set自动排序并去重。

三、代码

#include<stdio.h>
#include<iostream>
#include<stdbool.h>
#include<string>
#include<set>
#include<map>
#include<algorithm>
using namespace std;

const int maxn = 120000 + 10;
string words[maxn];
map<string, bool>myhash;
set<string>ans;

int main()
{
    int cnt = 0;
    while (cin >> words[cnt])
    {
        myhash[words[cnt++]] = true;
    }
    for (int i = 0; i < cnt; i++)
    {
        for (int j = 0; j < words[i].size() - 1; j++)
        {
            string a = words[i].substr(0, j + 1);
            string b = words[i].substr(j + 1);
            if (myhash[a] && myhash[b])
            {
                ans.insert(words[i]);    //这里直接输出好像不对，可能出现重复的情况
            }
        }
    }
    for (set<string>::iterator it = ans.begin(); it != ans.end(); it++)
        cout << *it;
    return 0;
}