传闻是FB 2012的题,真不是盖的啊。
有一个长度为n的字符串str,有非常多的关键字query(长度不超过10),需要判断每个关键字是否是str的子串。
注意:query是动态的输入进行查询的,预先并不知道所有的query。
请实现2个函数initWithString(str)
和existSubString(query)
。我们会首先调用一次 initWithString(str)
,你可以在这个函数中做一些预处理操作。然后对于每一个query,函数existSubString(query)
需要返回这个query是否为str的子串。
有一个很重要的点是注意从query长度不超过10能挖掘出什么来。
预处理肯定是要建立后缀数组的了,然后排序,之后就可以拿query串到排序数组里去找是否是某个串的前缀,如果是的话就返回true了。
这样做是可解的了,主要是注意sort时候要给一个比较函数,不然就拿char*的地址来比较了。
但是很容易发现源串的长度会非常大,比如1000000,这样建立后缀数组后,排序的时间会非常的长,我们需要从query串长度很小这里来优化一下这个排序,既然query与后缀数组中的串最多也就比较10次,那么后缀数组中的两个后缀大于10的长度之后的字符的关系其实是没有意义的,所以排序的时候,比较两个串只用看前10个的关系,这样排序的效率就提升了很多,也就不会超时了。
#include<algorithm> using namespace std; vector<char*> prefix; bool compare(char* s, char* q) { if ( !s || !q ) return !s; int n=10; while(n--&&*s!='\0'&&*q!='\0') { if( *s < *q ) return true; else if ( *s > *q ) return false; else s++,q++; } if ( n==0 ) return true; else return *s=='\0'; } int startWith(char* src,char* query); // 预处理初始化 void initWithString(char *str) { if ( !str ) return; while(*str!='\0') prefix.push_back(str++); sort(prefix.begin(),prefix.end(),compare); } // 如果query是str的字串,返回true,否则返回false bool existSubString(char *query) { if ( !query || *query=='\0' ) return true; int l =0,r=prefix.size()-1; while(l<=r) { int mid=l+((r-l)>>1); int cmp=startWith(prefix[mid],query); if ( cmp == 0 ) return true; else if (cmp<0 ) l=mid+1; else r=mid-1; } return false; } int startWith(char* src,char* query) { if ( !src || !query) { if ( !query ) return 0; else return -1; } while(*src!='\0' && *query!='\0') { if ( *src<*query) return -1; else if ( *src >*query ) return 1; else src++,query++; } if ( *query=='\0' ) return 0; else return -1; }