hdu5790 Prefix(Trie树+主席树)
2016-08-03 19:18
316 查看
Problem Description
Alice gets N strings. Now she has Q questions to ask you. For each question, she wanna know how many different prefix strings between Lth and Rth strings. It's so easy right? So solve it!
Input
The input contains multiple test cases.
For each test case, the first line contains one integer N(1≤N≤100000).
Then next N lines contain N strings and the total length of N strings is between 1 and 100000. The next line contains one integer Q(1≤Q≤100000).
We define a specail integer Z=0. For each query, you get two integer L, R(0=<L,R<N). Then the query interval [L,R] is [min((Z+L)%N,(Z+R)%N)+1,max((Z+L)%N,(Z+R)%N)+1]. And Z change to the answer of this query.
Output
For each question, output the answer.
Sample Input
3
abc
aba
baa
3
0 2
0 1
1 1
Sample Output
7
6
3
题意:给你n个字符串,问你第L个字符串到R个字符串中不同前缀的个数,且强制在线。
思路:这题和之前d-query这题很相似,那题问的是区间内不同数的种类。这题问的是不同前缀个数,所以我们可以先把所有的字符串插入到Trie树中,然后每次插入维护每一个节点最后被遍历到的时刻,然后用主席树维护下就行了。
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<string>
#include<bitset>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef long double ldb;
#define inf 99999999
#define pi acos(-1.0)
#define Key_value ch[ch[root][1]][0]
#define maxn 100050
#define maxnode 1000050
char s[maxn];
int n;
int ch[maxnode][28];
int val[maxnode];
int sz;
#define M 1000500*30
int lson[M],rson[M],c[M],T[M];
int th;
int build(int l,int r)
{
int i,j,newroot=++th,mid;
c[newroot]=0;
if(l!=r){
mid=(l+r)/2;
lson[newroot]=build(l,mid);
rson[newroot]=build(mid+1,r);
}
return newroot;
}
int update(int root,int zhi,int value)
{
int i,j,newroot=++th;
int tmp=newroot;
int l=1,r=n,mid;
c[newroot]=c[root]+value;
while(l<r){
mid=(l+r)/2;
if(zhi<=mid){
r=mid;
lson[newroot]=++th;rson[newroot]=rson[root];
newroot=lson[newroot];root=lson[root];
}
else{
l=mid+1;
lson[newroot]=lson[root];rson[newroot]=++th;
newroot=rson[newroot];root=rson[root];
}
c[newroot]=c[root]+value;
}
return tmp;
}
int question(int root,int pos)
{
int i,j;
int sum=0;
int l=1,r=n,mid;
while(l<r){
mid=(l+r)/2;
if(pos<=mid){
r=mid;
sum+=c[rson[root] ];
root=lson[root];
}
else{
l=mid+1;
root=rson[root];
}
}
sum+=c[root];
return sum;
}
void init(){
sz=0;memset(ch[0],0,sizeof(ch[0]));
memset(val,0,sizeof(val));
}
int idx(char c){
return c-'a';
}
void charu(char *s,int tm){
int u=0,len=strlen(s),i,c;
T[tm]=T[tm-1];
for(i=0;i<len;i++){
c=idx(s[i]);
if(!ch[u][c]){
sz++;
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=tm;
T[tm]=update(T[tm],tm,1);
ch[u][c]=sz;
u=ch[u][c];
}
else if(ch[u][c]){
T[tm]=update(T[tm],val[ch[u][c] ],-1);
val[ch[u][c] ]=tm;
T[tm]=update(T[tm],tm,1);
u=ch[u][c];
}
}
}
int main()
{
int m,i,j;
while(scanf("%d",&n)!=EOF)
{
init();
th=0;
T[0]=build(1,n);
for(i=1;i<=n;i++){
scanf("%s",s);
charu(s,i);
}
scanf("%d",&m);
int l,r,z=0,t1,t2;
for(i=1;i<=m;i++){
scanf("%d%d",&l,&r);
l=(z+l)%n+1;
r=(z+r)%n+1;
if(l>r)swap(l,r);
z=question(T[r],l);
printf("%d\n",z);
}
}
return 0;
}
Alice gets N strings. Now she has Q questions to ask you. For each question, she wanna know how many different prefix strings between Lth and Rth strings. It's so easy right? So solve it!
Input
The input contains multiple test cases.
For each test case, the first line contains one integer N(1≤N≤100000).
Then next N lines contain N strings and the total length of N strings is between 1 and 100000. The next line contains one integer Q(1≤Q≤100000).
We define a specail integer Z=0. For each query, you get two integer L, R(0=<L,R<N). Then the query interval [L,R] is [min((Z+L)%N,(Z+R)%N)+1,max((Z+L)%N,(Z+R)%N)+1]. And Z change to the answer of this query.
Output
For each question, output the answer.
Sample Input
3
abc
aba
baa
3
0 2
0 1
1 1
Sample Output
7
6
3
题意:给你n个字符串,问你第L个字符串到R个字符串中不同前缀的个数,且强制在线。
思路:这题和之前d-query这题很相似,那题问的是区间内不同数的种类。这题问的是不同前缀个数,所以我们可以先把所有的字符串插入到Trie树中,然后每次插入维护每一个节点最后被遍历到的时刻,然后用主席树维护下就行了。
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<string>
#include<bitset>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef long double ldb;
#define inf 99999999
#define pi acos(-1.0)
#define Key_value ch[ch[root][1]][0]
#define maxn 100050
#define maxnode 1000050
char s[maxn];
int n;
int ch[maxnode][28];
int val[maxnode];
int sz;
#define M 1000500*30
int lson[M],rson[M],c[M],T[M];
int th;
int build(int l,int r)
{
int i,j,newroot=++th,mid;
c[newroot]=0;
if(l!=r){
mid=(l+r)/2;
lson[newroot]=build(l,mid);
rson[newroot]=build(mid+1,r);
}
return newroot;
}
int update(int root,int zhi,int value)
{
int i,j,newroot=++th;
int tmp=newroot;
int l=1,r=n,mid;
c[newroot]=c[root]+value;
while(l<r){
mid=(l+r)/2;
if(zhi<=mid){
r=mid;
lson[newroot]=++th;rson[newroot]=rson[root];
newroot=lson[newroot];root=lson[root];
}
else{
l=mid+1;
lson[newroot]=lson[root];rson[newroot]=++th;
newroot=rson[newroot];root=rson[root];
}
c[newroot]=c[root]+value;
}
return tmp;
}
int question(int root,int pos)
{
int i,j;
int sum=0;
int l=1,r=n,mid;
while(l<r){
mid=(l+r)/2;
if(pos<=mid){
r=mid;
sum+=c[rson[root] ];
root=lson[root];
}
else{
l=mid+1;
root=rson[root];
}
}
sum+=c[root];
return sum;
}
void init(){
sz=0;memset(ch[0],0,sizeof(ch[0]));
memset(val,0,sizeof(val));
}
int idx(char c){
return c-'a';
}
void charu(char *s,int tm){
int u=0,len=strlen(s),i,c;
T[tm]=T[tm-1];
for(i=0;i<len;i++){
c=idx(s[i]);
if(!ch[u][c]){
sz++;
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=tm;
T[tm]=update(T[tm],tm,1);
ch[u][c]=sz;
u=ch[u][c];
}
else if(ch[u][c]){
T[tm]=update(T[tm],val[ch[u][c] ],-1);
val[ch[u][c] ]=tm;
T[tm]=update(T[tm],tm,1);
u=ch[u][c];
}
}
}
int main()
{
int m,i,j;
while(scanf("%d",&n)!=EOF)
{
init();
th=0;
T[0]=build(1,n);
for(i=1;i<=n;i++){
scanf("%s",s);
charu(s,i);
}
scanf("%d",&m);
int l,r,z=0,t1,t2;
for(i=1;i<=m;i++){
scanf("%d%d",&l,&r);
l=(z+l)%n+1;
r=(z+r)%n+1;
if(l>r)swap(l,r);
z=question(T[r],l);
printf("%d\n",z);
}
}
return 0;
}
相关文章推荐
- HDU5790 Prefix 字典树+主席树
- 【HDU - 5790 】Prefix(主席树+Trie树)
- leetcode---Implement Trie (Prefix Tree)---Trie树
- 机房测试 字符串 【Hash+KMP】【分块+可持久化Trie树+贪心】【AC自动机+主席树】
- HDU 5790 Prefix(字典树、主席树)
- Trie树(Prefix Tree)介绍
- LeetCode Implement Trie (Prefix Tree) (实现trie树3个函数:插入,查找,前缀)
- hdu5790 Prefix(2016多校第五场1009)trie+主席树
- 【BZOJ 3439】Kpm的MC密码 主席树+trie树
- 【BZOJ】Kpm的MC密码-trie树+主席树+dfs序
- UVa 11488 Hyper Prefix Sets Trie树计算最大公共前缀长度*个数
- 【bzoj3439】Kpm的MC密码 trie树+主席树
- Shortest Proper Prefix---Trie树
- 2016多校训练Contest5: 1010 Prefix hdu5790
- hdu 5790 prefix 主席树
- 【主席树】Codechef Prefix XOR
- uva11488 Hyper Prefix Sets(Trie树)
- [BZOJ]3439: Kpm的MC密码 trie树+主席树(线段树合并)
- 求公共前缀长度与所选字符串个数的乘积的最大值 Trie树求最值 UVA 11488 Hyper Prefix Sets
- hdu 5790 prefix 主席树在线维护区间不同数的个数