2017阿里C++研发工程师-校招-单词匹配

题目描述

给一个字符串， 然后给一个字典。 把字符串分解成字典里的单词组成的句子， 请输出所需空格最少的方案。并输出该方案。

样例

例如： 字符串为： str="ilikealibaba", 字典为dict= {"i", "like", "ali", "ba", "alibaba", "baba"}

输入：
ilikealibaba
6
i
like
ali
ba
alibaba
baba

输出：
i like alibaba

解题思路：

空格最少，很显然用BFS应该可以做。 不过状态转移的时候需要写个小函数， 另外由于需要打印路径，所以要记录一下路径。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <algorithm>
#include <stdexcept>
#include <cstring>
using namespace std;


struct Node
{
	string word;     // 当前匹配过的单词
	int pos;            // 记录下一个要匹配的位置 
	string path;      // 记录路径  
}; 

int matchW(const string& str, int pos, const string word)
{
	for(int i = 0; i < word.size(); i++)
	{
		if(pos >= str.size())
			return -1; 
		if(str[pos++] != word[i])
			return -1; 
	}
	
	return pos; 
}

void mincut(const string& str, const set<string>& dict)
{
	if(str.size() == 0)
	{
		printf("n/a
"); 
		return; 
	}
	
	queue<Node> qu; 
	for(auto it : dict) 
	{
		int tmp = matchW(str, 0, it); 
		if(tmp != -1)        // 该单词是否能匹配
		{
			Node tt; 
			tt.word = it; 
			tt.pos = tmp;
			tt.path += it;  
			qu.push(tt); 
		}
	}
	
	while(!qu.empty())
	{
		int len = qu.size(); 
		while(len--)
		{
			Node tt = qu.front(); 
			qu.pop(); 
			if(tt.pos == str.size())    // 已匹配完成
			{
				cout << tt.path << endl;      // 打印路径
				return; 
			}
			
			for(auto it : dict)
			{
				int tmp = matchW(str, tt.pos, it); 
				if(tmp != -1)
				{
					Node gg; 
					gg.word = it; 
					gg.pos = tmp;
					gg.path = tt.path + " " + it;   
					qu.push(gg); 
				}
			}
		} 
	}
	
	printf("n/a
"); 
}


int main(int argc, const char * argv[])
{
    string strS;
    string dictStr;
    int nDict;
    set<string> dict;
    
    cin>>strS;
    cin>>nDict;
    for (int i = 0; i < nDict; i++)
    {
        cin>>dictStr;
        dict.insert(dictStr);
    }
    mincut(strS, dict);
    
    return 0;
}