字典树 + DP

http://www.acdream.tk/problem.php?cid=1001&pid=1 

dp[i] 表示匹配前i个字符的最大权值

View Code
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
class trie{
public :
int num;
trie* child[27];
trie()
{
num=0;
memset(child,0,sizeof(child));
}
}root;
void insert(char *s,int id)
{
class trie *cur=&root;
int len=strlen(s);
for(int i=0;i<len;i++)
{
int id=s[i]-'a';
if(!cur->child[id])
cur->child[id]=new trie;
cur=cur->child[id];
}
cur->num=id;
}
queue<int> Q;
void find(char *s)
{
while(!Q.empty()) Q.pop();
class trie *cur=&root;
for(int i=0;s[i];i++)
{
int id=s[i]-'a';
if(!cur->child[id]) return ;
cur=cur->child[id];
if(cur->num) Q.push(cur->num);
}
}
void init(trie* T)
{
int i;
for(i=0;i<26;i++)
{
if(T->child[i])
init(T->child[i]);
}
T->num=0;
for(i=0;i<26;i++)
T->child[i]=0;
}
int max(int a,int b)
{
return a>b?a:b;
}
char s[10010];
char ss[50];
int w[1010];
int dp[10100];
int length[1010];
int main()
{
int i,j,n;

while(scanf("%d%s",&n,s+1)!=EOF)
{
class trie* cur=&root;
init(cur);
int len=strlen(s+1);
for(i=1;i<=n;i++)
{
scanf("%s%d",ss,&w[i]);
length[i]=strlen(ss);
insert(ss,i);
}
memset(dp,-1,sizeof(dp));
dp[0]=0;
for(i=1;i<=len;i++)if(dp[i-1]!=-1)
{
find(s+i);
while(!Q.empty())
{
j=Q.front();Q.pop();
dp[i+length[j]-1]=max(dp[i+length[j]-1],dp[i-1]+w[j]);
}
}
printf("%d\n",dp[len]);
}
return 0;
}



posted @ 2012-03-23 21:11  Because Of You  Views(370)  Comments(0Edit  收藏  举报