首页 > 代码库 > BZOJ 2754 SCOI2012 喵星球上的点名 fail树+set启发式合并
BZOJ 2754 SCOI2012 喵星球上的点名 fail树+set启发式合并
题目大意:给定n个目标串和m个模式串,问这m个模式串每个在多少个目标串中出现过,以及n个目标串每个以最多多少个模式串为子串
我错了……就算用fail树+set启发式合并也优化不到O(nlog^2n)……这题的数据范围相当无解啊
首先将所有名字和点名的字符串全都插进AC自动机
将每个点上开一个set记录这个点是哪些喵星人的名字的前缀
然后建立fail树 沿着fail树从下到上启发式合并
每合并完一个点 如果这个点是某次点名的字符串 那么这次点名点到的喵星人就是这个点的set中的所有元素 统计答案即可
统计答案以外的操作显然是O(nlog^2n) 但是统计答案这里还是没有避免高复杂度
最终复杂度为O(nlog^2n+Σans2) 其中ans2是第二问的答案
感觉这个算法的瓶颈就在Σans2上了……或许我们可以换一种做法统计ans2?感觉……不是很必要的样子……
#include <set> #include <map> #include <vector> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 200200 using namespace std; struct Trie{ Trie *fail; map<int,Trie*> son; set<int> s; vector<Trie*> fail_son; int ed,cnt; }*root,mempool[M],*C=mempool; int n,m,ans1[50500],ans2[20200]; int substitute[50500]; void Insert(Trie* &p,int left,int pos,bool flag) { if(!p) p=C++; if(!flag) p->s.insert(pos); if(!left) { if(flag) { if(p->ed) substitute[p->ed]=pos; p->ed=pos; p->cnt++; } return ; } int temp; scanf("%d",&temp); Insert(p->son[temp],left-1,pos,flag); } void Build_Tree() { static Trie *q[M]; static int r,h; map<int,Trie*>::iterator it; for(it=root->son.begin();it!=root->son.end();it++) { it->second->fail=root; root->fail_son.push_back(it->second); q[++r]=it->second; } while(r!=h) { Trie *p=q[++h]; for(it=p->son.begin();it!=p->son.end();it++) { Trie *temp=p->fail; while(temp!=root&&!temp->son[it->first]) temp=temp->fail; if(temp->son[it->first]) temp=temp->son[it->first]; it->second->fail=temp; temp->fail_son.push_back(it->second); q[++r]=it->second; } } } void DFS(Trie *p) { vector<Trie*>::iterator it; set<int>::iterator _it; for(it=p->fail_son.begin();it!=p->fail_son.end();it++) { DFS(*it); set<int>&s=p->s; set<int>&son_s=(*it)->s; if( s.size()<son_s.size() ) swap(s,son_s); for(_it=son_s.begin();_it!=son_s.end();_it++) s.insert(*_it); son_s.clear(); } if(p->ed) { ans1[p->ed]=p->s.size(); for(_it=p->s.begin();_it!=p->s.end();_it++) ans2[*_it]+=p->cnt; } } int main() { int i,x; cin>>n>>m; for(i=1;i<=n;i++) { scanf("%d",&x); Insert(root,x,i,0); scanf("%d",&x); Insert(root,x,i,0); } for(i=1;i<=m;i++) { scanf("%d",&x); Insert(root,x,i,1); } Build_Tree(); DFS(root); for(i=m;i;i--) if(substitute[i]) ans1[i]=ans1[substitute[i]]; for(i=1;i<=m;i++) printf("%d\n",ans1[i]); for(i=1;i<=n;i++) printf("%d%c",ans2[i],i==n?'\n':' '); return 0; }
BZOJ 2754 SCOI2012 喵星球上的点名 fail树+set启发式合并
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。