POJ1035 Spell checker

题目来源：http://poj.org/problem?id=1035

题目大意：

　　写一个拼写检查程序。给定一些字典里的单词和一些待检查的词。需要检查的词可能是以下几种情况：

　　1. 是正确的词；

　　2. 删除一个字母可以成为一个正确的词；

　　3. 加入一个字母可以成为一个正确的词；

　　4. 替换一个字母可以成为正确的词。

　　写一个程序找出给定的词可能是字典中的哪个词。

输入：首先给出字典的所有单词。"#"结束。然后给出待查的词，"#"结束。

输出：首先输出给出的待查词，若词正确输出" is correct",否则，先输出一个冒号，然后输出可能对应的字典里的词，有多个时按在字典中出现的顺序输出。每个单词前有个空格隔开。

---

Sample Input

i
is
has
have
be
my
more
contest
me
too
if
award
#
me
aware
m
contest
hav
oo
or
i
fi
mre
#

Sample Output

me is correct
aware: award
m: i my me
contest is correct
hav: has have
oo: too
or:
i is correct
fi: i
mre: more me

---

遍历暴力解决之..（对于长度明显无法对应的词略去不查）。

//////////////////////////////////////////////////////////////////////////
//        POJ1035 Spell checker
//        Memory: 964K        Time: 313MS
//        Language: C++        Result: Accepted
//////////////////////////////////////////////////////////////////////////

#include <iostream>
#include <map>
#include <set>
#include <list>
#include <string>

using namespace std;

map<string, int> dictionary;
set<size_t> lens;
bool comp(string a, string b) {
    return (dictionary[a] - dictionary[b]) < 0 ? true : false;
}

int main() {
    string buff;
    int index = 0;
    while (cin >> buff && buff != "#") {
        dictionary[buff] = ++index;
        size_t length = buff.size();
        lens.insert(length);
    }
    while (cin >> buff && buff != "#") {
        size_t length = buff.size();
        list<string> queue;
        if (lens.find(length) != lens.end()) {
            if (dictionary.find(buff) != dictionary.end()) {
                cout << buff << " is correct" << endl;
                continue;
            }
            for (char c = 'a'; c <= 'z'; ++c) {
                for (int i = 0; i < buff.size(); ++i) {
                    string temp = buff;
                    temp.replace(i, 1, 1, c);
                    if (dictionary.find(temp) != dictionary.end()) {
                        queue.push_back(temp);
                    }
                }
            }
        }
        if (lens.find(length - 1) != lens.end()) {
            for (int i = 0; i < buff.size(); ++i) {
                string temp = buff;
                temp.erase(i, 1);
                if (dictionary.find(temp) != dictionary.end()) {
                    queue.push_back(temp);
                }
            }
        }
        if (lens.find(length + 1) != lens.end()) {
            for (char c = 'a'; c <= 'z'; ++c) {
                for (int i = 0; i <= buff.size(); ++i) {
                    string temp = buff;
                    temp.insert(i, 1, c);
                    if (dictionary.find(temp) != dictionary.end()) {
                        queue.push_back(temp);
                    }
                }
            }
        }
        queue.unique();
        queue.sort(comp);
        cout << buff << ":";
        for (list<string>::iterator it = queue.begin(); it != queue.end(); ++it) {
            cout << " " << *it;
        }
        cout << endl;
    }
    system("pause");
    return 0;
}