字典树及manacher

  • 字典树:

1.查找前缀出现次数

交的时候用 C++ 提交,G++ 爆内存 HDU1251

指针模拟

#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
using namespace std;
struct node
{
	int count;
	node *next[26];
	node()
	{
		count=0;
		memset(next,0,sizeof(next));
	}
};
node root;
void insert(char *s)
{
	node *p=&root;
	int l=strlen(s);
	for(int i=0;i<l;i++)
	{
		if(p->next[s[i]-'a']==NULL)
		{
			p->next[s[i]-'a']=new node;
		}
		p=p->next[s[i]-'a'];
		p->count++;
	}
}
int find(char *s)
{
	node *p=&root;
	int l=strlen(s);
	for(int i=0;i<l;i++)
	{
		if(p->next[s[i]-'a']==NULL)
		{
			return 0;
		}
		p=p->next[s[i]-'a'];
	}
	return p->count;
}
int main()
{
	char s[13];
	while(gets(s))
	{
		if(s[0]==NULL)  break;
		insert(s);
	}
	while(gets(s))
	{
		printf("%d\n",find(s));
	}
	return 0;
}

 数组模拟

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int trie[400001][26],len,root,tot,sum[400001];
bool p;
int n,m; 
char s[11];
void insert()
{
    len=strlen(s);
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id]) trie[root][id]=++tot;
        sum[trie[root][id]]++;//前缀后移一个位置保存 
        root=trie[root][id];
    }
}
int search()
{
    root=0;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id]) return 0;
        root=trie[root][id];
    }//root经过此循环后变成前缀最后一个字母所在位置的后一个位置 
    return sum[root];//因为前缀后移了一个保存,所以此时的sum[root]就是要求的前缀出现的次数 
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        cin>>s;
        insert();
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        cin>>s;
        printf("%d\n",search());
    }
}

 用map简略写法

#include <stdio.h>
#include <string.h>
#include <map>
#include <string>
using namespace std;
map<string,int>m1;
int main()
{
	char z[10];
	while(gets(z))
	{
		if(strlen(z)==0)
			break;
		for(int i=strlen(z);i>0;i--)
		{
			z[i]='\0';//把该位置字符变为空
			m1[z]++;//用map计数
		}
	}
	while(gets(z))
	{
		printf("%d\n",m1[z]);
	}
	return 0;
}

2.判断是否都是前缀串(删除和重建)HDU1350

#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
using namespace std;
bool v=1;
struct node
{
	bool flag;
	node *next[2];
	node()
	{
		flag=false;
		memset(next,0,sizeof(next));
	}
};
node root;
void insert(char *s)
{
	node *p=&root;
	int l=strlen(s);
	for(int i=0;i<l;i++)
	{
		if(p->next[s[i]-'0']==NULL)
		{
			p->next[s[i]-'0']=new node;
		}
		p=p->next[s[i]-'0'];
		if(p->flag)   v=0;
	}
	p->flag=true;
}
void del(node *p)
{
	for(int i=0;i<2;i++)
	{
		if(p->next[i])  del(p->next[i]);
	 } 
	 delete p;
}
int main()
{
	char s[10];
	int j=1;
	while(scanf("%s",&s)!=EOF)
	{
		if(s[0]=='9')
		{
			if(v) printf("Set %d is immediately decodable\n",j++);
			else  printf("Set %d is not immediately decodable\n",j++);
			for(int i=0;i<2;i++)
		        {
			    if(root.next[i])  del(root.next[i]);
		        }
			root.next[0]=root.next[1]=NULL;v=1;
			continue;
		}
		else if(v)
		{
		   insert(s);
		}
	}
	return 0;
}
  • manacher

求最长回文长度 HDU3068

平均访问每个字符四到五次。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
char s[110005];
char ss[220010];
int p[220010];
int i,j,len;
int manacher(int len)
{
	int id,mlen=0,mx=0;
	for(int i=1;i<len;i++)
	{
		if(i<mx)  p[i]=min(p[2*id-i],mx-i);
		else  p[i]=1;
		while(ss[i-p[i]]==ss[i+p[i]])  p[i]++;
		if(mx<i+p[i])
		{
			id=i;mx=i+p[i];
		}
		mlen=max(mlen,p[i]-1);
	}
	return mlen;
} 
int main()
{
	while(scanf("%s",&s)!=EOF)
	{
		len=strlen(s);
		ss[0]='@';ss[1]='#';
		j=2;
		for(i=0;i<len;i++)
		{
		    ss[j++]=s[i];
		    ss[j++]='#';
		}
		ss[j]='\0'; 
		printf("%d\n",manacher(j));
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/haohaoxuexilmy/article/details/82428204