【Leetcode】Implement strStr()

问题

Implement strStr().

Returns a pointer to the first occurrence of needle in haystack, or null if needle is not part of haystack.

代码

int hashCode(char* needle , int c)
{
	int hashcode = 0;
	for (int i = 0 ; i < c ; i++) {
		hashcode <<= 1;
		hashcode += needle[i];
	}
	return hashcode;
}

void putInMap(map<int , list<int>* >& map, int hashCode , int value)
{
	list<int> *list_s = map[hashCode];
	if (!list_s) {
		list_s = new list<int>;
		map[hashCode] = list_s;
	}
	list_s->push_back(value);
	
}

class Solution {
public:
    char *strStr(char *haystack, char *needle) 
    {
        if (*needle =='\0')
            return haystack;
		if (!haystack || !needle || *needle == '\0' || *haystack =='\0') {
			return NULL;
		}
		
		map<int , list<int >* > map;
		size_t size = strlen(haystack);
		size_t needle_size = strlen(needle);
		
		if(size < needle_size)
			return NULL;
		
		int tempHashCode = hashCode(haystack, needle_size);
		int myHashCode = hashCode(needle, needle_size);
		/* set int map */
		putInMap(map , tempHashCode , 0);

		int gap = 1 << (needle_size -1);
		
		for (int i = needle_size; i < size ; i++) {
			tempHashCode -= haystack[ i - needle_size] << (needle_size - 1);
			tempHashCode <<= 1;
			tempHashCode += haystack[i];
			putInMap(map , tempHashCode , (i - needle_size + 1));
		}
		
		list<int> *tempList = map[myHashCode];
		if (!tempList) {
			return NULL;
		}else{
			while (tempList->size() != 0) {
				int tempI = tempList->front();
				tempList->pop_front();
				char *a = haystack + tempI;
				char *b = needle;
				while (*b != '\0') {
					if(*b ++ != *a ++)
						break;
				}
				if (*b == '\0') {
					return haystack + tempI;
				}
			}
		}
		return NULL;
	}
};

分析

我还没有学会用kmp算法做算法查找，我选择的一个方法是使用hash先让所有的字符串散列。 然后用map查找，总的复杂度是nlogn，因为每次插入map中，需要logn的开销（map使用红黑树实现的），然后遍历的时候也用到了一个小技巧，hash值的算取。

总结

这个我觉得用kmp算法可能会更省代码和空间一些。